Don't need torch.cuda.is_available() to compile

Dao-AILab · Jun 18, 2024 · 20928c2 · 20928c2
1 parent daeadc4
commit 20928c2
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/causal_conv1d/__init__.py b/causal_conv1d/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "1.3.0"
+__version__ = "1.3.0.post1"
 
 from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update
diff --git a/setup.py b/setup.py
@@ -125,10 +125,6 @@ def append_nvcc_threads(nvcc_extra_args):
 ext_modules = []
 
 
-if not torch.cuda.is_available():
-    raise RuntimeError(f"CUDA/HIP not available within Pytorch. {PACKAGE_NAME} is intended for a CUDA/HIP-enabled Pytorch installation.")
-
-
 HIP_BUILD = bool(torch.version.hip)
 
 if not SKIP_CUDA_BUILD: