Merge pull request #25 from 7SK/XLA_fix

fix xla
ROCm · Mar 28, 2018 · e6b0c7a · e6b0c7a
2 parents ed6bf17 + 7aecbf2
commit e6b0c7a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tensorflow/compiler/jit/kernels/xla_local_launch_op.cc b/tensorflow/compiler/jit/kernels/xla_local_launch_op.cc
@@ -151,7 +151,9 @@ XlaLocalLaunchOp::XlaLocalLaunchOp(OpKernelConstruction* ctx)
   if (device_type_ == DeviceType(DEVICE_CPU)) {
     platform_id_ = gpu::host::kHostPlatformId;
   } else if (device_type_ == DeviceType(DEVICE_GPU)) {
-    platform_id_ = gpu::cuda::kCudaPlatformId;
+    // XXX FIXME devise a way to cope with multiple platforms
+    //platform_id_ = gpu::cuda::kCudaPlatformId;
+    platform_id_ = gpu::rocm::kROCmPlatformId;
   } else {
     ctx->SetStatus(
         errors::InvalidArgument("Unknown device type for local _XlaLaunch"));