File tree 1 file changed +1
-1
lines changed
1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -78,7 +78,7 @@ TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)
78
78
if [[ " $PACKAGE_TYPE " =~ .* wheel.* && -n " ${PYTORCH_EXTRA_INSTALL_REQUIREMENTS:- } " ]]; then
79
79
# Only linux Python < 3.12 are supported wheels for triton
80
80
TRITON_CONSTRAINT=" platform_system == 'Linux' and platform_machine == 'x86_64' and python_version < '3.12'"
81
- TRITON_REQUIREMENT=" pytorch- triton==${TRITON_VERSION} ; ${TRITON_CONSTRAINT} "
81
+ TRITON_REQUIREMENT=" triton==${TRITON_VERSION} ; ${TRITON_CONSTRAINT} "
82
82
if [[ -n " $PYTORCH_BUILD_VERSION " && " $PYTORCH_BUILD_VERSION " =~ .* dev.* ]]; then
83
83
TRITON_SHORTHASH=$( cut -c1-10 $PYTORCH_ROOT /.ci/docker/ci_commit_pins/triton.txt)
84
84
TRITON_REQUIREMENT=" pytorch-triton==${TRITON_VERSION} +${TRITON_SHORTHASH} ; ${TRITON_CONSTRAINT} "
You can’t perform that action at this time.
0 commit comments