const away const qualifier

ROCm · Aug 14, 2023 · cc39f03 · cc39f03
1 parent 14bc7f5
commit cc39f03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/compiler/xla/stream_executor/gpu/gpu_graph.cc b/tensorflow/compiler/xla/stream_executor/gpu/gpu_graph.cc
@@ -132,7 +132,7 @@ tsl::StatusOr<GpuGraphNodeHandle> AddKernelNode(
 }
 
 static GpuDevicePtr AsDevicePtr(const DeviceMemoryBase& mem) {
-  return reinterpret_cast<GpuDevicePtr>(mem.opaque());
+  return const_cast<GpuDevicePtr>(reinterpret_cast<const GpuDevicePtr>(mem.opaque()));
 }
 
 tsl::StatusOr<GpuGraphNodeHandle> AddMemcpyD2DNode(