maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir][sparse][bufferization] fix a few memory leaks 

Fixed some new memory leaks after migration to new
bufferization. One is expected, the other may need
some more careful analysis.

Reviewed By: jpienaar

Differential Revision: https://reviews.llvm.org/D129805
This commit is contained in:
Aart Bik 2022-07-14 14:55:43 -07:00
parent 2e29b0138c
commit f30822d29c
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matmul.mlir
View File

@ -239,6 +239,13 @@ module {
sparse_tensor.release %7 : tensor<4x4xf64, #CSR>
sparse_tensor.release %8 : tensor<4x4xf64, #DCSR>
// TODO(springerm): needed?
%m0 = bufferization.to_memref %0 : memref<4x4xf64>
memref.dealloc %m0 : memref<4x4xf64>
%m3 = bufferization.to_memref %3 : memref<4x4xf64>
memref.dealloc %m3 : memref<4x4xf64>
%m6 = bufferization.to_memref %6 : memref<4x4xf64>
memref.dealloc %m6 : memref<4x4xf64>
return
}
}

1
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_transpose.mlir
View File

@ -113,6 +113,7 @@ module {
// Release resources.
sparse_tensor.release %a : tensor<3x4xf64, #DCSR>
sparse_tensor.release %0 : tensor<4x3xf64, #DCSR>
sparse_tensor.release %1 : tensor<4x3xf64, #DCSR>
return
}