build: triton only has wheels for linux

brycedrennan · Jan 18, 2024 · 279dc28 · 279dc28
1 parent 3906072
commit 279dc28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -105,7 +105,7 @@ def get_git_revision_hash() -> str:
         "torchdiffeq>=0.2.0",
         "torchvision>=0.13.1",
         "transformers>=4.19.2",
-        "triton>=2.0.0; sys_platform!='darwin' and platform_machine!='aarch64'",
+        "triton>=2.0.0; sys_platform!='darwin' and platform_machine!='aarch64' and sys_platform == 'linux'",
         "kornia>=0.6",
         "uvicorn>=0.16.0",
         # "xformers>=0.0.22; sys_platform!='darwin' and platform_machine!='aarch64'",