Revert "Fix torch check (#6402)"

This reverts commit 55b4cae.
microsoft · Aug 27, 2024 · 761c342 · 761c342
1 parent 32c1b21
commit 761c342
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/op_builder/gds.py b/op_builder/gds.py
@@ -36,13 +36,7 @@ def extra_ldflags(self):
         return super().extra_ldflags() + ['-lcufile']
 
     def is_compatible(self, verbose=False):
-        try:
-            import torch.utils.cpp_extension
-        except ImportError:
-            if verbose:
-                self.warning("Please install torch if trying to pre-compile GDS")
-            return False
-
+        import torch.utils.cpp_extension
         CUDA_HOME = torch.utils.cpp_extension.CUDA_HOME
         CUDA_LIB64 = os.path.join(CUDA_HOME, "lib64")
         gds_compatible = self.has_function(funcname="cuFileDriverOpen",