Skip to content

Commit 2f00d85

Browse files
ethanwee1jithunnair-amd
authored andcommitted
[release/2.4] Update libamd_comgr.so.3 in triton wheel build (#1969)
libamd_comgr.so.2 is breaking rocm6.4 and newer. Breaking this build: http://rocm-ci.amd.com/job/pytorch2.4-manylinux-wheels_rel-6.4/14/ similar changes to ROCm/triton#760 Validation http://rocm-ci.amd.com/job/pytorch2.4-manylinux-wheels_rel-6.4/18/
1 parent b72d367 commit 2f00d85

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.github/scripts/amd/package_triton_wheel.sh

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,11 +54,16 @@ done
5454
ROCM_SO=(
5555
"libamdhip64.so.6"
5656
"libhsa-runtime64.so.1"
57-
"libamd_comgr.so.2"
5857
"libdrm.so.2"
5958
"libdrm_amdgpu.so.1"
6059
)
6160

61+
if [[ $ROCM_INT -ge 60400 ]]; then
62+
ROCM_SO+=("libamd_comgr.so.3")
63+
else
64+
ROCM_SO+=("libamd_comgr.so.2")
65+
fi
66+
6267
if [[ $ROCM_INT -ge 60100 ]]; then
6368
ROCM_SO+=("librocprofiler-register.so.0")
6469
fi

0 commit comments

Comments
 (0)