fix compile w/o cuda

NVlabs · Sep 16, 2024 · fa2d3ac · fa2d3ac
1 parent 8196668
commit fa2d3ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -80,7 +80,7 @@ def get_compiler():
             extra_compile_args={'nvcc': ['-O2']},
         ),
     )
-except ImportError:
+except (ImportError, OSError):
     warnings.warn("Cuda extensions for torch not found, skipping cuda healpix padding module")