define AT_USE_HIPSPARSE_XXX when USE_ROCM is false

To prevent compilation errors on non-ROCm platforms.
pytorch · Oct 4, 2023 · f15f5b3 · f15f5b3
1 parent 6b97905
commit f15f5b3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/aten/src/ATen/cuda/CUDASparse.h b/aten/src/ATen/cuda/CUDASparse.h
@@ -54,6 +54,10 @@
 #endif
 
 #endif // HIPSPARSE_VERSION >= 200400
+#else // USE_ROCM
+#define AT_USE_HIPSPARSE_CONST_DESCRIPTORS() 0
+#define AT_USE_HIPSPARSE_GENERIC_52_API() 0
+#define AT_USE_HIPSPARSE_GENERIC_API() 0
 #endif // USE_ROCM
 
 // cuSparse Generic API spsv function was added in CUDA 11.3.0