fix "multiple rules generate" error on CUDA

pytorch · Oct 18, 2023 · b4de21e · b4de21e
1 parent 134af61
commit b4de21e
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -131,11 +131,10 @@ def get_extensions():
 
     main_file = glob.glob(os.path.join(extensions_dir, "*.cpp")) + glob.glob(
         os.path.join(extensions_dir, "ops", "*.cpp")
-    )
+    ) + glob.glob(os.path.join(extensions_dir, "ops", "autocast", "*.cpp"))
     source_cpu = (
         glob.glob(os.path.join(extensions_dir, "ops", "autograd", "*.cpp"))
         + glob.glob(os.path.join(extensions_dir, "ops", "cpu", "*.cpp"))
-        + glob.glob(os.path.join(extensions_dir, "ops", "autocast", "*.cpp"))
         + glob.glob(os.path.join(extensions_dir, "ops", "quantized", "cpu", "*.cpp"))
     )
     source_mps = glob.glob(os.path.join(extensions_dir, "ops", "mps", "*.mm"))
@@ -185,8 +184,6 @@ def get_extensions():
     else:
         source_cuda = glob.glob(os.path.join(extensions_dir, "ops", "cuda", "*.cu"))
 
-    source_cuda += glob.glob(os.path.join(extensions_dir, "ops", "autocast", "*.cpp"))
-
     sources = main_file + source_cpu
     extension = CppExtension