From fd6b0a4356c66f5f30cedbc62b24f18d9e32806f Mon Sep 17 00:00:00 2001
From: pramenku <7664080+pramenku@users.noreply.github.com>
Date: Thu, 4 Jul 2024 21:02:49 +0530
Subject: [PATCH]  [r2.16] Update llvm path after compiler change (#2575)

Added the logic to add new path from ROCm >=6.3.0
---
 third_party/gpus/rocm_configure.bzl | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/third_party/gpus/rocm_configure.bzl b/third_party/gpus/rocm_configure.bzl
index 6eb17486514105..3ba4c356eb5e31 100644
--- a/third_party/gpus/rocm_configure.bzl
+++ b/third_party/gpus/rocm_configure.bzl
@@ -207,7 +207,11 @@ def _rocm_include_path(repository_ctx, rocm_config, bash_bin):
     inc_dirs.append(rocm_toolkit_path + "/llvm/lib/clang/17/include")
     inc_dirs.append(rocm_toolkit_path + "/llvm/lib/clang/18/include")
     inc_dirs.append(rocm_toolkit_path + "/llvm/lib/clang/19/include")
-
+    if int(rocm_config.rocm_version_number) >= 60300:
+        inc_dirs.append(rocm_toolkit_path + "/lib/llvm/lib/clang/17/include")
+        inc_dirs.append(rocm_toolkit_path + "/lib/llvm/lib/clang/18/include")
+        inc_dirs.append(rocm_toolkit_path + "/lib/llvm/lib/clang/19/include")
+    
     # Support hcc based off clang 10.0.0 (for ROCm 3.3)
     inc_dirs.append(rocm_toolkit_path + "/hcc/compiler/lib/clang/10.0.0/include/")
     inc_dirs.append(rocm_toolkit_path + "/hcc/lib/clang/10.0.0/include")