[Manywheel] Restrict pytorch-triton to x86-64 Linux

Partially addresses pytorch/pytorch#114042
atalman · Nov 22, 2023 · 2a750eb · 2a750eb
1 parent fb8aae9
commit 2a750eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/manywheel/build_cuda.sh b/manywheel/build_cuda.sh
@@ -268,7 +268,7 @@ fi
 if [[ $(uname) == "Linux" && "$DESIRED_PYTHON" != "3.12" ]]; then
     TRITON_SHORTHASH=$(cut -c1-10 $PYTORCH_ROOT/.github/ci_commit_pins/triton.txt)
     TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)
-    TRITON_REQUIREMENT="pytorch-triton==${TRITON_VERSION}+${TRITON_SHORTHASH}"
+    TRITON_REQUIREMENT="pytorch-triton==${TRITON_VERSION}+${TRITON_SHORTHASH}; platform_system == 'Linux' and platform_machine == 'x86_64'"
 
     if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
         export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${TRITON_REQUIREMENT}"