Skip to content

Commit 77f61e6

Browse files
committed
Revert "Fix torch check (deepspeedai#6402)"
This reverts commit 55b4cae.
1 parent 1bfa341 commit 77f61e6

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

op_builder/gds.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -36,13 +36,7 @@ def extra_ldflags(self):
3636
return super().extra_ldflags() + ['-lcufile']
3737

3838
def is_compatible(self, verbose=False):
39-
try:
40-
import torch.utils.cpp_extension
41-
except ImportError:
42-
if verbose:
43-
self.warning("Please install torch if trying to pre-compile GDS")
44-
return False
45-
39+
import torch.utils.cpp_extension
4640
CUDA_HOME = torch.utils.cpp_extension.CUDA_HOME
4741
CUDA_LIB64 = os.path.join(CUDA_HOME, "lib64")
4842
gds_compatible = self.has_function(funcname="cuFileDriverOpen",

0 commit comments

Comments
 (0)