From c7429ac1fd1786c95844e65c3a80ab357605cbc1 Mon Sep 17 00:00:00 2001 From: Alex Morehead Date: Wed, 4 Sep 2024 09:55:21 -0500 Subject: [PATCH] Update gds.py --- op_builder/gds.py | 5 +++++ 1 file changed, 5 insertions(+) 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", ),