We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7afbef9 commit 1d444d3Copy full SHA for 1d444d3
manywheel/build_cuda.sh
@@ -266,8 +266,7 @@ fi
266
# and torch.compile doesn't work.
267
if [[ $(uname) == "Linux" && "$DESIRED_PYTHON" != "3.12" ]]; then
268
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'"
+ TRITON_REQUIREMENT="triton==${TRITON_VERSION}; platform_system == 'Linux' and platform_machine == 'x86_64'"
271
272
if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
273
export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${TRITON_REQUIREMENT}"
0 commit comments