Skip to content

Commit 5ca74b6

Browse files
committed
Don't need torch.cuda.is_available() to compile
1 parent e61f105 commit 5ca74b6

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

setup.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -131,10 +131,6 @@ def append_nvcc_threads(nvcc_extra_args):
131131
ext_modules = []
132132

133133

134-
if not torch.cuda.is_available():
135-
raise RuntimeError(f"CUDA/HIP not available within Pytorch. {PACKAGE_NAME} is intended for a CUDA/HIP-enabled Pytorch installation.")
136-
137-
138134
HIP_BUILD = bool(torch.version.hip)
139135

140136
if not SKIP_CUDA_BUILD:

0 commit comments

Comments
 (0)