Skip to content

Commit 68bacb6

Browse files
authored
Updates to support rocm5.3 wheel builds (#6)
* Changes to support ROCm 5.3 * Updated as per comments
1 parent b086528 commit 68bacb6

File tree

1 file changed

+74
-1
lines changed

1 file changed

+74
-1
lines changed

manywheel/build_rocm.sh

Lines changed: 74 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,80 @@ else
8989
fi
9090
ROCM_INT=$(($ROCM_VERSION_MAJOR * 10000 + $ROCM_VERSION_MINOR * 100 + $ROCM_VERSION_PATCH))
9191

92-
if [[ $ROCM_INT -ge 50200 ]]; then
92+
if [[ $ROCM_INT -ge 50300 ]]; then
93+
DEPS_LIST=(
94+
"/opt/rocm/lib/libMIOpen.so.1"
95+
"/opt/rocm/lib/libamdhip64.so.5"
96+
"/opt/rocm/lib/libhipblas.so.0"
97+
"/opt/rocm/lib/libhipfft.so"
98+
"/opt/rocm/lib/libhiprand.so.1"
99+
"/opt/rocm/lib/libhipsparse.so.0"
100+
"/opt/rocm/lib/libhsa-runtime64.so.1"
101+
"/opt/rocm/lib/libamd_comgr.so.2"
102+
"/opt/rocm/magma/lib/libmagma.so"
103+
"/opt/rocm/lib/librccl.so.1"
104+
"/opt/rocm/lib/librocblas.so.0"
105+
"/opt/rocm/lib/librocfft-device-0.so.0"
106+
"/opt/rocm/lib/librocfft-device-1.so.0"
107+
"/opt/rocm/lib/librocfft-device-2.so.0"
108+
"/opt/rocm/lib/librocfft-device-3.so.0"
109+
"/opt/rocm/lib/librocfft.so.0"
110+
"/opt/rocm/lib/librocm_smi64.so.5"
111+
"/opt/rocm/lib/librocrand.so.1"
112+
"/opt/rocm/lib/librocsolver.so.0"
113+
"/opt/rocm/lib/librocsparse.so.0"
114+
"/opt/rocm/lib/libroctracer64.so.4"
115+
"/opt/rocm/lib/libroctx64.so.4"
116+
"$LIBGOMP_PATH"
117+
"$LIBNUMA_PATH"
118+
"$LIBELF_PATH"
119+
"$LIBTINFO_PATH"
120+
"$LIBDRM_PATH"
121+
"$LIBDRM_AMDGPU_PATH"
122+
)
123+
124+
DEPS_SONAME=(
125+
"libMIOpen.so.1"
126+
"libamdhip64.so.5"
127+
"libhipblas.so.0"
128+
"libhipfft.so"
129+
"libhiprand.so.1"
130+
"libhipsparse.so.0"
131+
"libhsa-runtime64.so.1"
132+
"libamd_comgr.so.2"
133+
"libmagma.so"
134+
"librccl.so.1"
135+
"librocblas.so.0"
136+
"librocfft-device-0.so.0"
137+
"librocfft-device-1.so.0"
138+
"librocfft-device-2.so.0"
139+
"librocfft-device-3.so.0"
140+
"librocfft.so.0"
141+
"librocm_smi64.so.5"
142+
"librocrand.so.1"
143+
"librocsolver.so.0"
144+
"librocsparse.so.0"
145+
"libroctracer64.so.4"
146+
"libroctx64.so.4"
147+
"libgomp.so.1"
148+
"libnuma.so.1"
149+
"libelf.so.1"
150+
"libtinfo.so.5"
151+
"libdrm.so.2"
152+
"libdrm_amdgpu.so.1"
153+
)
154+
155+
DEPS_AUX_SRCLIST=(
156+
"/opt/rocm/lib/rocblas/library/*"
157+
"/opt/amdgpu/share/libdrm/amdgpu.ids"
158+
)
159+
160+
DEPS_AUX_DSTLIST=(
161+
"lib/rocblas/library/."
162+
"share/libdrm/amdgpu.ids"
163+
)
164+
165+
elif [[ $ROCM_INT -ge 50200 ]]; then
93166
DEPS_LIST=(
94167
"/opt/rocm/lib/libMIOpen.so.1"
95168
"/opt/rocm/lib/libamdhip64.so.5"

0 commit comments

Comments
 (0)