fix deallocate bug

test=develop

fix deallocate bug
test=develop
90076522 · sneaxiy · 439af8d5 · 90076522
隐藏空白更改
内联并排

Showing with 1 addition and 3 deletion

paddle/fluid/platform/device_context.cc paddle/fluid/platform/device_context.cc +1 -3

未找到文件。
--- a/paddle/fluid/platform/device_context.cc
+++ b/paddle/fluid/platform/device_context.cc
@@ -127,9 +127,7 @@ class EigenCudaStreamDevice : public Eigen::StreamInterface {
    return retv;
  }

-  void deallocate(void* buffer) const override {
-    allocations_.erase(allocations_.find(buffer));
-  }
+  void deallocate(void* buffer) const override { allocations_.erase(buffer); }

  void* scratchpad() const override {
    if (scratch_ == NULL) {