Skip to content

Commit

Permalink
[Release-Only] Use triton pypi package for 2.1.0 release (#1533)
Browse files Browse the repository at this point in the history
* [Release-Only] Use triton pypi package for 2.1.0 release

* Clean up for release
  • Loading branch information
huydhn committed Sep 14, 2023
1 parent b4ac18a commit 922b32e
Showing 1 changed file with 0 additions and 11 deletions.
11 changes: 0 additions & 11 deletions manywheel/build_cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -257,17 +257,6 @@ else
exit 1
fi

# Prepare for 2.1.0 release
if [[ $(uname) == "Linux" ]]; then
TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)

if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="pytorch-triton==${TRITON_VERSION}"
else
export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${PYTORCH_EXTRA_INSTALL_REQUIREMENTS} | pytorch-triton==${TRITON_VERSION}"
fi
fi

# builder/test.sh requires DESIRED_CUDA to know what tests to exclude
export DESIRED_CUDA="$cuda_version_nodot"

Expand Down

0 comments on commit 922b32e

Please sign in to comment.