Update frozenseg/modeling/pixel_decoder/ops/setup.py

frozenseg/modeling/pixel_decoder/ops/setup.py

@@ -47,11 +47,11 @@ def get_extensions():
             "-D__CUDA_NO_HALF_CONVERSIONS__",
             "-D__CUDA_NO_HALF2_OPERATORS__",
         ]
-    # else:
-    #     if CUDA_HOME is None:
-    #         raise NotImplementedError('CUDA_HOME is None. Please set environment variable CUDA_HOME.')
-    #     else:
-    #         raise NotImplementedError('No CUDA runtime is found. Please set FORCE_CUDA=1 or test it by running torch.cuda.is_available().')
+    else:
+        if CUDA_HOME is None:
+            raise NotImplementedError('CUDA_HOME is None. Please set environment variable CUDA_HOME.')
+        else:
+            raise NotImplementedError('No CUDA runtime is found. Please set FORCE_CUDA=1 or test it by running torch.cuda.is_available().')
 
     sources = [os.path.join(extensions_dir, s) for s in sources]
     include_dirs = [extensions_dir]