fix device optimizer config (#44282)

bcf57274 · zmxdream · GitHub · abc2cc57 · bcf57274
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/framework/fleet/heter_ps/hashtable_kernel.cu paddle/fluid/framework/fleet/heter_ps/hashtable_kernel.cu +1 -0

未找到文件。
--- a/paddle/fluid/framework/fleet/heter_ps/hashtable_kernel.cu
+++ b/paddle/fluid/framework/fleet/heter_ps/hashtable_kernel.cu
@@ -169,6 +169,7 @@ HashTable<KeyType, ValType>::HashTable(size_t capacity) {
 template <typename KeyType, typename ValType>
 HashTable<KeyType, ValType>::~HashTable() {
  delete container_;
+  cudaFree(device_optimizer_config_);
 }

 template <typename KeyType, typename ValType>