pytorch

fa885baf - [ROCm] Update ROCm pin to fix triton wheel lib issue (#108137)

[ROCm] Update ROCm pin to fix triton wheel lib issue (#108137) ROCm nightly wheels are facing issue as so files required for triton wheel are not being copied to third_party after https://github.com/pytorch/pytorch/pull/107600 This PR is updating our pinned commit to bring in a revert a change to search for none numbered so's which seems to be causing the issue Before PR 107600 https://github.com/pytorch/pytorch/actions/runs/5763938339/job/15626824908 ``` 2023-08-04T15:35:34.8049152Z Copying DRM Libraries 2023-08-04T15:35:34.8239451Z + ROCM_SO=("libhsa-runtime64.so.1" "libamdhip64.so.5" "libamd_comgr.so.2") 2023-08-04T15:35:34.8239816Z + mkdir -p python/triton/third_party/rocm/lib 2023-08-04T15:35:34.8295087Z + for lib in '"${ROCM_SO[@]}"' 2023-08-04T15:35:34.8295443Z + filepath=/tmp/vanilla_extract/opt/rocm-5.4.2/lib/libamdhip64.so.5 2023-08-04T15:35:34.8295891Z + cp /tmp/vanilla_extract/opt/rocm-5.4.2/lib/libamdhip64.so.5 python/triton/third_party/rocm/lib/ 2023-08-04T15:35:34.8421258Z ++ echo libamdhip64.so.5 2023-08-04T15:35:34.8421621Z ++ sed -e 's/\.so.*/.so/g' 2023-08-04T15:35:34.8432022Z + LINKNAME=libamdhip64.so 2023-08-04T15:35:34.8432686Z + ln -sf libamdhip64.so.5 python/triton/third_party/rocm/lib/libamdhip64.so ... 2023-08-04T15:35:35.7473664Z copying triton/third_party/rocm/lib/libamdhip64.so -> build/lib.linux-x86_64-cpython-38/triton/third_party/rocm/lib 2023-08-04T15:35:35.7617341Z copying triton/third_party/rocm/lib/libamdhip64.so.5 -> build/lib.linux-x86_64-cpython-38/triton/third_party/rocm/lib ... 2023-08-04T15:40:10.5063779Z copying build/lib.linux-x86_64-cpython-38/triton/third_party/rocm/lib/libamdhip64.so -> build/bdist.linux-x86_64/wheel/triton/third_party/rocm/lib 2023-08-04T15:40:10.6144654Z copying build/lib.linux-x86_64-cpython-38/triton/third_party/rocm/lib/libamdhip64.so.5 -> build/bdist.linux-x86_64/wheel/triton/third_party/rocm/lib ... 2023-08-04T15:40:37.3571973Z adding 'triton/third_party/rocm/lib/libamdhip64.so' 2023-08-04T15:40:38.4553988Z adding 'triton/third_party/rocm/lib/libamdhip64.so.5' ... 2023-08-04T15:40:53.3747917Z Setting rpath of triton/third_party/rocm/lib/libamdhip64.so to '' 2023-08-04T15:40:53.4602326Z $ORIGIN 2023-08-04T15:40:53.4620034Z Setting rpath of triton/third_party/rocm/lib/libamdhip64.so.5 to '' 2023-08-04T15:40:53.6337419Z $ORIGIN ... 2023-08-04T15:40:53.7152828Z Copied triton/third_party/rocm/lib/libamdhip64.so to triton/third_party/rocm/lib/libamdhip64.so 2023-08-04T15:40:53.7215480Z Copied triton/third_party/rocm/lib/libamdhip64.so.5 to triton/third_party/rocm/lib/libamdhip64.so ... ``` After PR 107600 https://github.com/pytorch/pytorch/actions/runs/5967761429/job/16190110783 ``` 2023-08-24T18:59:14.3976234Z + ROCM_SO=("libhsa-runtime64.so" "libamdhip64.so" "libamd_comgr.so" "libdrm.so" "libdrm_amdgpu.so") 2023-08-24T18:59:14.3977489Z + for lib in '"${ROCM_SO[@]}"' 2023-08-24T18:59:14.4216575Z + file_path=($(find $ROCM_HOME/lib/ -name "$lib")) 2023-08-24T18:59:14.4219237Z ++ find /opt/rocm/lib/ -name libamdhip64.so 2023-08-24T18:59:14.4254361Z + [[ -z /opt/rocm/lib/libamdhip64.so ]] 2023-08-24T18:59:14.4254677Z + [[ -z /opt/rocm/lib/libamdhip64.so ]] 2023-08-24T18:59:14.4254965Z + [[ -z /opt/rocm/lib/libamdhip64.so ]] 2023-08-24T18:59:14.4255246Z + [[ -z /opt/rocm/lib/libamdhip64.so ]] 2023-08-24T18:59:14.4255538Z + cp /opt/rocm/lib/libamdhip64.so python/triton/third_party/rocm/lib 2023-08-24T18:59:14.5085155Z ++ echo libamdhip64.so 2023-08-24T18:59:14.5085818Z ++ sed -e 's/\.so.*/.so/g' 2023-08-24T18:59:14.5097094Z + LINKNAME=libamdhip64.so 2023-08-24T18:59:14.5097572Z + ln -sf libamdhip64.so python/triton/third_party/rocm/lib/libamdhip64.so ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/108137 Approved by: https://github.com/sunway513, https://github.com/huydhn