From b96c2a7ea3bf99cc87a7f78b1e2c77ffcd441239 Mon Sep 17 00:00:00 2001 From: atalman Date: Thu, 30 May 2024 10:24:03 -0700 Subject: [PATCH] remove triton constraint for py312 --- conda/build_pytorch.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh index 56bb7654c..9f0649c50 100755 --- a/conda/build_pytorch.sh +++ b/conda/build_pytorch.sh @@ -287,9 +287,9 @@ else TRITON_VERSION=$(cat $pytorch_rootdir/.ci/docker/triton_version.txt) if [[ -n "$OVERRIDE_PACKAGE_VERSION" && "$OVERRIDE_PACKAGE_VERSION" =~ .*dev.* ]]; then TRITON_SHORTHASH=$(cut -c1-10 $pytorch_rootdir/.github/ci_commit_pins/triton.txt) - export CONDA_TRITON_CONSTRAINT=" - torchtriton==${TRITON_VERSION}+${TRITON_SHORTHASH} # [py < 312]" + export CONDA_TRITON_CONSTRAINT=" - torchtriton==${TRITON_VERSION}+${TRITON_SHORTHASH} # [py < 313]" else - export CONDA_TRITON_CONSTRAINT=" - torchtriton==${TRITON_VERSION} # [py < 312]" + export CONDA_TRITON_CONSTRAINT=" - torchtriton==${TRITON_VERSION} # [py < 313]" fi fi