fix bug in OpKernelKey equal method

8 years ago · 953eaadbf6
parent 6a5793baba
commit 953eaadbf6
1 changed files with 3 additions and 1 deletions
--- a/paddle/framework/operator.h
+++ b/paddle/framework/operator.h
@ -199,7 +199,9 @@ class OperatorWithKernel : public OperatorBase {
      place_ = dev_ctx.GetPlace();
    }

-    bool operator==(const OpKernelKey& o) const { return place_ == o.place_; }
+    bool operator==(const OpKernelKey& o) const {
+      return platform::places_are_same_class(place_, o.place_);
+    }
  };

  struct OpKernelHash {