From caefc797c27c55b6ed1a432517b811c0b29077af Mon Sep 17 00:00:00 2001
From: Rakesh Roy <rakesh.roy@amd.com>
Date: Tue, 17 Dec 2024 16:45:55 +0530
Subject: [PATCH] [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.
---
 xla/stream_executor/rocm/rocm_driver.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/xla/stream_executor/rocm/rocm_driver.cc b/xla/stream_executor/rocm/rocm_driver.cc
index 465dbbe84b2a0..55fc33aa01131 100644
--- 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");