Skip to content

Commit

Permalink
Revert "Use pytorch-triton package untill triton pypi release" (#1673)
Browse files Browse the repository at this point in the history
atalman authored Jan 10, 2024

Verified

This commit was signed with the committer’s verified signature.
neo1973 Markus Härer
1 parent 7afbef9 commit 1d444d3
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions manywheel/build_cuda.sh
Original file line number Diff line number Diff line change
@@ -266,8 +266,7 @@ fi
# and torch.compile doesn't work.
if [[ $(uname) == "Linux" && "$DESIRED_PYTHON" != "3.12" ]]; then
TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)
# while triton is not released officially yet. use pytorch triton
TRITON_REQUIREMENT="pytorch-triton==${TRITON_VERSION}; platform_system == 'Linux' and platform_machine == 'x86_64'"
TRITON_REQUIREMENT="triton==${TRITON_VERSION}; platform_system == 'Linux' and platform_machine == 'x86_64'"

if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${TRITON_REQUIREMENT}"

0 comments on commit 1d444d3

Please sign in to comment.