HipGraph: Add a default catch for unknown node types (#2794)

ROCm · Dec 19, 2024 · cf4d244 · cf4d244
1 parent 3c18b05
commit cf4d244
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/third_party/xla/xla/stream_executor/rocm/rocm_driver.cc b/third_party/xla/xla/stream_executor/rocm/rocm_driver.cc
@@ -687,6 +687,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");