diff --git a/paddle/fluid/framework/fleet/heter_ps/hashtable_kernel.cu b/paddle/fluid/framework/fleet/heter_ps/hashtable_kernel.cu index 92df8d8581a869df3469882e6bb9a9d9e7ef7ca3..bb9998249048e1628d267cdb52987bbb07692b2c 100644 --- 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::HashTable(size_t capacity) { template HashTable::~HashTable() { delete container_; + cudaFree(device_optimizer_config_); } template