disable amd for win32

triton-lang · Feb 10, 2024 · 50bbd49 · 50bbd49
1 parent 2465e8d
commit 50bbd49
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/setup.py b/python/setup.py
@@ -392,7 +392,10 @@ def build_extension(self, ext):
         url_func=lambda system, arch, version:
         f"https://anaconda.org/nvidia/cuda-nvdisasm/{version}/download/{system}-{arch}/cuda-nvdisasm-{version}-0.tar.bz2",
     )
-backends = _copy_backends(["nvidia", "amd"])
+backends = ["nvidia", "amd"]
+if os.name == "nt":
+    backends = ["nvidia"]
+backends = _copy_backends(backends)
 
 
 def add_link_to_backends():