Fix crash when attempting to copy string tensor from CPU to DML (#126)

microsoft · Dec 16, 2020 · 33d39bf · 33d39bf
1 parent c77b197
commit 33d39bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/core/common_runtime/eager/tensor_handle.cc b/tensorflow/core/common_runtime/eager/tensor_handle.cc
@@ -495,8 +495,8 @@ Status TensorHandle::CopyToDevice(EagerContext* ctx, tensorflow::Device* dstd,
     *output = *src;
     return Status::OK();
   }
-  if (!dst_device_context && (src->dtype() != tensorflow::DT_VARIANT &&
-                   !tensorflow::DataTypeCanUseMemcpy(src->dtype()))) {
+  if (dst_device_context && (src->dtype() != tensorflow::DT_VARIANT &&
+                             !tensorflow::DataTypeCanUseMemcpy(src->dtype()))) {
     return tensorflow::errors::InvalidArgument(
         "Can't copy Tensor with type ",
         tensorflow::DataTypeString(src->dtype()), " to device ", dstd->name(),