Adding a fix for mkl_eager_op_rewrite_test from PR tensorflow#60436

Intel-tensorflow · Jul 5, 2023 · 105b99f · 105b99f
1 parent 3d85617
commit 105b99f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite_test.cc b/tensorflow/core/common_runtime/eager/mkl_eager_op_rewrite_test.cc
@@ -38,12 +38,13 @@ class EagerOpRewriteTest : public ::testing::Test {
         std::make_unique<StaticDeviceMgr>(DeviceFactory::NewDevice(
             "CPU", {}, "/job:localhost/replica:0/task:0/device:CPU:0"));
     bool async = false;
-    tensorflow::Rendezvous* rendezvous =
-        new tensorflow::IntraProcessRendezvous(device_mgr.get());
+    auto rendezvous =
+        tsl::core::RefCountPtr<tensorflow::IntraProcessRendezvous>(
+            new tensorflow::IntraProcessRendezvous(device_mgr.get()));
     eager_ctx_ = new tensorflow::EagerContext(
         SessionOptions(),
         tensorflow::ContextDevicePlacementPolicy::DEVICE_PLACEMENT_SILENT,
-        async, device_mgr.get(), false, rendezvous, nullptr, nullptr,
+        async, device_mgr.get(), false, std::move(rendezvous), nullptr, nullptr,
         /*run_eager_op_as_function=*/true);
 
     EagerExecutor executor_(false);