From ebcf86980a07dd5c24e763f7c1e2c8bdafe291b1 Mon Sep 17 00:00:00 2001
From: Jithun Nair <37884920+jithunnair-amd@users.noreply.github.com>
Date: Wed, 17 Apr 2024 11:32:34 -0500
Subject: [PATCH] Update setup_rocm_libs.sh

---
 scripts/amd/setup_rocm_libs.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/amd/setup_rocm_libs.sh b/scripts/amd/setup_rocm_libs.sh
index d4d8e0b0af1d..21425691ea87 100755
--- a/scripts/amd/setup_rocm_libs.sh
+++ b/scripts/amd/setup_rocm_libs.sh
@@ -91,7 +91,7 @@ do
 done
 
 # Copy Include Files
-cp -r $ROCM_HOME/include $TRITON_ROCM_DIR/
+cp -r $ROCM_HOME/include/hip $TRITON_ROCM_DIR/include
 
 # Copy linker
 mkdir -p $TRITON_ROCM_DIR/llvm/bin