[Bugfix] fixed leak in SpMMCreateBlocks (#5210)

* fixed leak in SpMMCreateBlocks * clang format

[Bugfix] fixed leak in SpMMCreateBlocks (#5210)
* fixed leak in SpMMCreateBlocks * clang format
99937422 · Kacper Pietkun · GitHub · 45153fc0 · 99937422
Unverified Commit 99937422 authored Feb 23, 2023 by Kacper Pietkun Committed by GitHub Feb 23, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 4 deletions

src/array/cpu/spmm_blocking_libxsmm.h src/array/cpu/spmm_blocking_libxsmm.h +10 -4

No files found.
--- a/src/array/cpu/spmm_blocking_libxsmm.h
+++ b/src/array/cpu/spmm_blocking_libxsmm.h
@@ -80,10 +80,16 @@ inline void SpMMCreateBlocks(
  if (num_K_blocks > 1) {
    IdType *indptr_block_buf = reinterpret_cast<IdType *>(aligned_alloc(
        64, (M_block_size + 1) * num_M_blocks * num_K_blocks * sizeof(IdType)));
-    IdType *indices_block_buf = reinterpret_cast<IdType *>(
-        aligned_alloc(64, indptr[M] * sizeof(IdType)));
-    IdType *edges_block_buf = reinterpret_cast<IdType *>(
-        aligned_alloc(64, indptr[M] * sizeof(IdType)));
+    IdType *indices_block_buf = nullptr;
+    if (use_lhs) {
+      indices_block_buf = reinterpret_cast<IdType *>(
+          aligned_alloc(64, indptr[M] * sizeof(IdType)));
+    }
+    IdType *edges_block_buf = nullptr;
+    if (use_rhs) {
+      edges_block_buf = reinterpret_cast<IdType *>(
+          aligned_alloc(64, indptr[M] * sizeof(IdType)));
+    }

 #pragma omp parallel
    {