Skip to content

Commit 1d444d3

Browse files
authored
Revert "Use pytorch-triton package untill triton pypi release" (#1673)
1 parent 7afbef9 commit 1d444d3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

manywheel/build_cuda.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -266,8 +266,7 @@ fi
266266
# and torch.compile doesn't work.
267267
if [[ $(uname) == "Linux" && "$DESIRED_PYTHON" != "3.12" ]]; then
268268
TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)
269-
# while triton is not released officially yet. use pytorch triton
270-
TRITON_REQUIREMENT="pytorch-triton==${TRITON_VERSION}; platform_system == 'Linux' and platform_machine == 'x86_64'"
269+
TRITON_REQUIREMENT="triton==${TRITON_VERSION}; platform_system == 'Linux' and platform_machine == 'x86_64'"
271270

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

0 commit comments

Comments
 (0)