[mlir][sparse] Moving `delete coo` into codegen instead of runtime library
authorwren romano <2998727+wrengr@users.noreply.github.com>
Wed, 30 Mar 2022 19:32:33 +0000 (12:32 -0700)
committerwren romano <2998727+wrengr@users.noreply.github.com>
Fri, 1 Apr 2022 18:08:52 +0000 (11:08 -0700)
commit63bdcaf92a5ed5dc8c9bdc0ea2d1a13b8ddb3c68
treecd39ea0430c89180323ad92f5f4a7792b1d104f3
parentec0b332cd8b57815eb3f905c0cca69864e90f3d2
[mlir][sparse] Moving `delete coo` into codegen instead of runtime library

Prior to this change there were a number of places where the allocation and deallocation of SparseTensorCOO objects were not cleanly paired, leading to inconsistencies regarding whether each function released its tensor/coo arguments or not, as well as making it easy to run afoul of memory leaks, use-after-free, or double-free errors.  This change cleans up the codegen vs runtime boundary to resolve those issues.  Now, the only time the runtime library frees an object is either (a) because it's a function explicitly designed to do so, or (b) because the allocated object is entirely local to the function and would be a memory leak if not released.  Thus, now the codegen takes complete responsibility for releasing any objects it caused to be allocated.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D122435
mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
mlir/lib/ExecutionEngine/SparseTensorUtils.cpp
mlir/test/Dialect/SparseTensor/conversion.mlir