From 20928c244c31fedc87c3770f1db2711c51c783b8 Mon Sep 17 00:00:00 2001
From: Tri Dao <tridpq@gmail.com>
Date: Tue, 18 Jun 2024 03:33:50 -0700
Subject: [PATCH] Don't need torch.cuda.is_available() to compile

---
 causal_conv1d/__init__.py | 2 +-
 setup.py                  | 4 ----
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/causal_conv1d/__init__.py b/causal_conv1d/__init__.py
index d061008..8a5dca7 100644
--- 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
index 4c9a6bf..b974854 100644
--- 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: