diff --git a/op_builder/gds.py b/op_builder/gds.py index 727ebdf48372..d815ccea387a 100644 --- a/op_builder/gds.py +++ b/op_builder/gds.py @@ -49,6 +49,11 @@ def is_compatible(self, verbose=False): return False CUDA_HOME = torch.utils.cpp_extension.CUDA_HOME + if CUDA_HOME is None: + if verbose: + self.warning("Please install torch CUDA if trying to pre-compile GDS with CUDA") + return False + CUDA_LIB64 = os.path.join(CUDA_HOME, "lib64") gds_compatible = self.has_function(funcname="cuFileDriverOpen", libraries=("cufile", ),