Force incarnation_id of 1 in multi_device_iterator

ROCm · Aug 15, 2023 · a5072f7 · a5072f7
1 parent 2d52de6
commit a5072f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/core/kernels/data/multi_device_iterator_ops.cc b/tensorflow/core/kernels/data/multi_device_iterator_ops.cc
@@ -142,7 +142,7 @@ class MultiDeviceIterator : public ResourceBase {
     params.thread_pool = &unbounded_thread_pool_;
     params.cancellation_manager = ctx->cancellation_manager();
     IteratorContext iter_ctx(std::move(params));
-    multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, incarnation_id,
+    multi_device_buffer_->GetNextFromShard(&iter_ctx, shard_num, 1/*incarnation_id*/,
                                            std::move(callback));
     return OkStatus();
   }