Skip to content

Commit

Permalink
[conda] Do not depend on triton for 3.12 yet
Browse files Browse the repository at this point in the history
  • Loading branch information
malfet authored Nov 30, 2023
1 parent 88457a1 commit ca378c1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion conda/build_pytorch.sh
Original file line number Diff line number Diff line change
Expand Up @@ -276,7 +276,7 @@ else
if [[ "$OSTYPE" != "msys" ]]; then
# TODO: Remove me when Triton has a proper release channel
TRITON_SHORTHASH=$(cut -c1-10 $pytorch_rootdir/.github/ci_commit_pins/triton.txt)
export CONDA_TRITON_CONSTRAINT=" - torchtriton==2.1.0+${TRITON_SHORTHASH}"
export CONDA_TRITON_CONSTRAINT=" - torchtriton==2.1.0+${TRITON_SHORTHASH} # [py < 312]"
fi

build_string_suffix="cuda${CUDA_VERSION}_cudnn${CUDNN_VERSION}_${build_string_suffix}"
Expand Down

0 comments on commit ca378c1

Please sign in to comment.