From dce79e80e1b684a5641dc59657a0c04621a0782b Mon Sep 17 00:00:00 2001 From: Pruthvi Madugundu Date: Mon, 29 Aug 2022 17:00:40 -0700 Subject: [PATCH 1/2] Changes to support ROCm 5.3 --- manywheel/build_rocm.sh | 121 +++++++++++++++++++++++++++++++++++++++- 1 file changed, 120 insertions(+), 1 deletion(-) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 9b4d36348..7961ffb4c 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -89,7 +89,126 @@ else fi ROCM_INT=$(($ROCM_VERSION_MAJOR * 10000 + $ROCM_VERSION_MINOR * 100 + $ROCM_VERSION_PATCH)) -if [[ $ROCM_INT -ge 50200 ]]; then +if [[ $ROCM_INT -ge 50300 ]]; then +DEPS_LIST=( + "/opt/rocm/lib/libMIOpen.so.1" + "/opt/rocm/lib/libamdhip64.so.5" + "/opt/rocm/lib/libhipblas.so.0" + "/opt/rocm/lib/libhipfft.so" + "/opt/rocm/lib/libhiprand.so.1" + "/opt/rocm/lib/libhipsparse.so.0" + "/opt/rocm/lib/libhsa-runtime64.so.1" + "/opt/rocm/lib/libamd_comgr.so.2" + "/opt/rocm/magma/lib/libmagma.so" + "/opt/rocm/lib/librccl.so.1" + "/opt/rocm/lib/librocblas.so.0" + "/opt/rocm/lib/librocfft-device-0.so.0" + "/opt/rocm/lib/librocfft-device-1.so.0" + "/opt/rocm/lib/librocfft-device-2.so.0" + "/opt/rocm/lib/librocfft-device-3.so.0" + "/opt/rocm/lib/librocfft.so.0" + "/opt/rocm/lib/librocm_smi64.so.5" + "/opt/rocm/lib/librocrand.so.1" + "/opt/rocm/lib/librocsolver.so.0" + "/opt/rocm/lib/librocsparse.so.0" + "/opt/rocm/lib/libroctracer64.so.4" + "/opt/rocm/lib/libroctx64.so.4" + "$LIBGOMP_PATH" + "$LIBNUMA_PATH" + "$LIBELF_PATH" + "$LIBTINFO_PATH" + "$LIBDRM_PATH" + "$LIBDRM_AMDGPU_PATH" +) + +DEPS_SONAME=( + "libMIOpen.so.1" + "libamdhip64.so.5" + "libhipblas.so.0" + "libhipfft.so" + "libhiprand.so.1" + "libhipsparse.so.0" + "libhsa-runtime64.so.1" + "libamd_comgr.so.2" + "libmagma.so" + "librccl.so.1" + "librocblas.so.0" + "librocfft-device-0.so.0" + "librocfft-device-1.so.0" + "librocfft-device-2.so.0" + "librocfft-device-3.so.0" + "librocfft.so.0" + "librocm_smi64.so.5" + "librocrand.so.1" + "librocsolver.so.0" + "librocsparse.so.0" + "libroctracer64.so.4" + "libroctx64.so.4" + "libgomp.so.1" + "libnuma.so.1" + "libelf.so.1" + "libtinfo.so.5" + "libdrm.so.2" + "libdrm_amdgpu.so.1" +) + +DEPS_AUX_SRCLIST=( + "/opt/rocm/lib/rocblas/library/Kernels.so-000-*.hsaco" + + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx803.co" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx900.co" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx906.co" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx908.co" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx90a.co" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx1030.co" + + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx803.dat" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx900.dat" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx906.dat" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx908.dat" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx90a.dat" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx1030.dat" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_fallback.dat" + + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx803.hsaco" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx900.hsaco" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx906-xnack*.hsaco" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx908-xnack*.hsaco" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx90a-xnack*.hsaco" + "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx1030.hsaco" + + "/opt/amdgpu/share/libdrm/amdgpu.ids" +) + +DEPS_AUX_DSTLIST=( + "lib/rocblas/library/." + + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + "lib/rocblas/library/." + + "share/libdrm/amdgpu.ids" +) + +elif [[ $ROCM_INT -ge 50200 ]]; then DEPS_LIST=( "/opt/rocm/lib/libMIOpen.so.1" "/opt/rocm/lib/libamdhip64.so.5" From d5f7fc9f08d4d1aecbd60c3e2bf7297dc4f2b1c4 Mon Sep 17 00:00:00 2001 From: Pruthvi Madugundu Date: Tue, 30 Aug 2022 10:30:20 -0700 Subject: [PATCH 2/2] Updated as per comments --- manywheel/build_rocm.sh | 48 +---------------------------------------- 1 file changed, 1 insertion(+), 47 deletions(-) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 7961ffb4c..50d3680dd 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -153,58 +153,12 @@ DEPS_SONAME=( ) DEPS_AUX_SRCLIST=( - "/opt/rocm/lib/rocblas/library/Kernels.so-000-*.hsaco" - - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx803.co" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx900.co" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx906.co" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx908.co" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx90a.co" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx1030.co" - - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx803.dat" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx900.dat" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx906.dat" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx908.dat" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx90a.dat" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx1030.dat" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_fallback.dat" - - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx803.hsaco" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx900.hsaco" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx906-xnack*.hsaco" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx908-xnack*.hsaco" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx90a-xnack*.hsaco" - "/opt/rocm/lib/rocblas/library/TensileLibrary_*_gfx1030.hsaco" - + "/opt/rocm/lib/rocblas/library/*" "/opt/amdgpu/share/libdrm/amdgpu.ids" ) DEPS_AUX_DSTLIST=( "lib/rocblas/library/." - - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "lib/rocblas/library/." - "share/libdrm/amdgpu.ids" )