From c1d69f9822f95fc8cc09d059ccc6996262a0927a Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Wed, 19 Jul 2023 04:35:13 +0000 Subject: [PATCH 1/2] Add pytorch-triton-rocm as an install dependency for ROCm --- manywheel/build_rocm.sh | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 0c1650f9b..e2b9ae66e 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -212,6 +212,17 @@ elif [[ $ROCM_INT -ge 50600 ]]; then DEPS_AUX_DSTLIST+=(${RCCL_SHARE_FILES[@]/#/$RCCL_SHARE_DST/}) fi +# Add triton install dependency +if [[ $(uname) == "Linux" ]]; then + TRITON_SHORTHASH=$(cut -c1-10 $PYTORCH_ROOT/.ci/docker/ci_commit_pins/triton-rocm.txt) + + if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then + export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="pytorch-triton-rocm==2.1.0+${TRITON_SHORTHASH}" + else + export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${PYTORCH_EXTRA_INSTALL_REQUIREMENTS} | pytorch-triton-rocm==2.1.0+${TRITON_SHORTHASH}" + fi +fi + echo "PYTORCH_ROCM_ARCH: ${PYTORCH_ROCM_ARCH}" From 6fa6ddcbf8997ebcd7175d3ec983600a20a8d4ad Mon Sep 17 00:00:00 2001 From: Jithun Nair <37884920+jithunnair-amd@users.noreply.github.com> Date: Thu, 27 Jul 2023 18:01:05 -0500 Subject: [PATCH 2/2] Update build_rocm.sh --- manywheel/build_rocm.sh | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index e2b9ae66e..1dee874d2 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -215,11 +215,12 @@ fi # Add triton install dependency if [[ $(uname) == "Linux" ]]; then TRITON_SHORTHASH=$(cut -c1-10 $PYTORCH_ROOT/.ci/docker/ci_commit_pins/triton-rocm.txt) + TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt) if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then - export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="pytorch-triton-rocm==2.1.0+${TRITON_SHORTHASH}" + export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="pytorch-triton-rocm==${TRITON_VERSION}+${TRITON_SHORTHASH}" else - export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${PYTORCH_EXTRA_INSTALL_REQUIREMENTS} | pytorch-triton-rocm==2.1.0+${TRITON_SHORTHASH}" + export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${PYTORCH_EXTRA_INSTALL_REQUIREMENTS} | pytorch-triton-rocm==${TRITON_VERSION}+${TRITON_SHORTHASH}" fi fi