Skip to content
This repository was archived by the owner on Aug 15, 2025. It is now read-only.

Do not hardcode triton version in builder code #1646

Merged
atalman merged 2 commits into from
Dec 18, 2023
Merged

Do not hardcode triton version in builder code #1646

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a61b566
Do not hardcode triton version in builder code
huydhn Dec 18, 2023
2d64e57
Minor tweak to use pytorch_rootdir
huydhn Dec 18, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion conda/build_pytorch.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -275,8 +275,9 @@ else
fi
if [[ "$OSTYPE" != "msys" ]]; then
# TODO: Remove me when Triton has a proper release channel
TRITON_VERSION=$(cat $PYTORCH_ROOT/.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}"
Expand Down