1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/sycl/sycl_device_context.cc b/tensorflow/core/common_runtime/sycl/sycl_device_context.cc
index b49420b1b5..a6be9195d4 100644
--- a/tensorflow/core/common_runtime/sycl/sycl_device_context.cc
+++ b/tensorflow/core/common_runtime/sycl/sycl_device_context.cc
@@ -95,6 +95,7 @@ void SYCLDeviceContext::CopyCPUTensorToDevice(const Tensor *cpu_tensor,
       assert(false && "unsupported type");
     }
   }
+  device->eigen_sycl_device()->synchronize();
   done(Status::OK());
 }
 
@@ -172,6 +173,7 @@ void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor,
       assert(false && "unsupported type");
     }
   }
+  device->eigen_sycl_device()->synchronize();
   done(Status::OK());
 }