Do not hardcode triton version in builder code (#1646)

* Do not hardcode triton version in builder code * Minor tweak to use pytorch_rootdir
pytorch · Dec 18, 2023 · 22d7be5 · 22d7be5
1 parent cd257e9
commit 22d7be5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
@@ -275,8 +275,9 @@ else
     fi
     if [[ "$OSTYPE" != "msys" ]]; then
         # TODO: Remove me when Triton has a proper release channel
+        TRITON_VERSION=$(cat $pytorch_rootdir/.ci/docker/triton_version.txt)
         TRITON_SHORTHASH=$(cut -c1-10 $pytorch_rootdir/.github/ci_commit_pins/triton.txt)
-        export CONDA_TRITON_CONSTRAINT="    - torchtriton==2.1.0+${TRITON_SHORTHASH} # [py < 312]"
+        export CONDA_TRITON_CONSTRAINT="    - torchtriton==${TRITON_VERSION}+${TRITON_SHORTHASH} # [py < 312]"
     fi
 
     build_string_suffix="cuda${CUDA_VERSION}_cudnn${CUDNN_VERSION}_${build_string_suffix}"