Fix IS_SPMM_AVAILABLE macro definition

malfet · malfet · commit 1096a4f54e2c · 2020-08-05T16:35:40.000-07:00
This should fix CUDA-11 on Windows build issue
diff --git a/aten/src/ATen/native/sparse/cuda/SparseCUDABlas.cu b/aten/src/ATen/native/sparse/cuda/SparseCUDABlas.cu
@@ -15,9 +15,11 @@
 // Using these APIs in any other systems will result in compile-time or run-time failures.
 // Their support will be extended in the next releases.
 
-#define IS_SPMM_AVAILABLE() (defined(__CUDACC__) &&                              \
-                             (CUSPARSE_VERSION >= 11000 ||                       \
-                              (!defined(_MSC_VER) && CUSPARSE_VERSION >= 10301)))
+#if defined(__CUDACC__) && (CUSPARSE_VERSION >= 11000 || (!defined(_MSC_VER) && CUSPARSE_VERSION >= 10301))
+#define IS_SPMM_AVAILABLE() 1
+#else
+#define IS_SPMM_AVAILABLE() 0
+#endif
 
 #if IS_SPMM_AVAILABLE()
 #include <library_types.h>
