[ROCm] SWDEV-504746 - Fix build error

HIP adds new enum hipGraphNodeTypeBatchMemOp in hipGraphNodeType. The switch case doesn't handle hipGraphNodeTypeBatchMemOp causing build failure in TensorFlow.
ROCm · Dec 17, 2024 · caefc79 · caefc79
1 parent 8d53a6c
commit caefc79
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xla/stream_executor/rocm/rocm_driver.cc b/xla/stream_executor/rocm/rocm_driver.cc
@@ -701,6 +701,8 @@ GpuDriver::GraphNodeGetType(hipGraphNode_t node) {
       return GraphNodeType::kMemAlloc;
     case hipGraphNodeTypeMemFree:
       return GraphNodeType::kMemFree;
+    default:
+      return absl::InternalError("Invalid HIP graph node type");
   }
 
   return absl::InternalError("Invalid HIP graph node type");