use only gpu-compatible build

diagnocat · Aug 16, 2021 · fb3872d · fb3872d
1 parent 89818eb
commit fb3872d
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -29,12 +29,7 @@
         [
          'sparseconvnet/SCN/cuda.cu', 'sparseconvnet/SCN/sparseconvnet_cuda.cpp', 'sparseconvnet/SCN/pybind.cpp'],
         include_dirs=[this_dir+'/sparseconvnet/SCN/'],
-        extra_compile_args=extra)
-      if torch.cuda.is_available()  else
-      CppExtension('sparseconvnet.SCN',
-        ['sparseconvnet/SCN/pybind.cpp', 'sparseconvnet/SCN/sparseconvnet_cpu.cpp'],
-        include_dirs=[this_dir+'/sparseconvnet/SCN/'],
-        extra_compile_args=extra['cxx'])],
+        extra_compile_args=extra)],
     cmdclass={'build_ext': BuildExtension},
     zip_safe=False,
 )