Remove from CUDAOpBuilder

microsoft · Nov 6, 2023 · d4e9bd9 · d4e9bd9
1 parent dec7bcc
commit d4e9bd9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -597,7 +597,12 @@ def is_compatible(self, verbose=True):
         return super().is_compatible(verbose)
 
     def builder(self):
-        self.build_for_cpu = not torch.cuda.is_available()
+        try:
+            if not self.is_rocm_pytorch():
+                assert_no_cuda_mismatch(self.name)
+            self.build_for_cpu = False
+        except BaseException:
+            self.build_for_cpu = True
 
         if self.build_for_cpu:
             from torch.utils.cpp_extension import CppExtension as ExtensionBuilder