Add ifdef guard

pytorch · Mar 22, 2023 · 9c09221 · 9c09221
1 parent c47a1ff
commit 9c09221
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/aten/src/ATen/native/sparse/SparseBlasImpl.cpp b/aten/src/ATen/native/sparse/SparseBlasImpl.cpp
@@ -218,6 +218,7 @@ Tensor& _compressed_row_strided_addmm_out(
 }
 
 namespace cpu {
+#if !AT_USE_MKL_SPARSE()
 template<typename scalar_t>
 void addmv_sparse_csr(
         const scalar_t* mat_values,
@@ -266,6 +267,8 @@ void addmv_sparse_bsr(
           }
         });
 }
+#endif // !AT_USE_MKL_SPARSE()
+
 /*
   Computes a sparse matrix-dense vector product defined as
   y <- alpha*op(A)*x + beta*y