Add ifdef guard

pytorch · Mar 22, 2023 · 6d663f7 · 6d663f7
1 parent c47a1ff
commit 6d663f7
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
@@ -217,6 +217,7 @@ Tensor& _compressed_row_strided_addmm_out(
   return result;
 }
 
+#if !AT_USE_MKL_SPARSE()
 namespace cpu {
 template<typename scalar_t>
 void addmv_sparse_csr(
@@ -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