[mlir][sparse] refactored codegen environment into its own file
authorAart Bik <ajcbik@google.com>
Tue, 20 Dec 2022 23:42:11 +0000 (15:42 -0800)
committerAart Bik <ajcbik@google.com>
Wed, 21 Dec 2022 00:58:59 +0000 (16:58 -0800)
commitfbe611309edf9ab8d9c26dcc82bc1ef38e99e637
tree1dc36f08461a21f37abfa865aec483db41bb226e
parent52aab0e4b4a5d6728f4bb9c7b3340133f585fc15
[mlir][sparse] refactored codegen environment into its own file

Also, as a proof of concept, all functionality related to reductions
has been refactored into private fields and a clean public API. As a
result, some dead code was found as well. This approach also simplifies
asserting on a proper environment state for each call.

NOTE: making all other fields private and migrating more methods into
      this new class is still TBD in yes another next revision!

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D140443
mlir/lib/Dialect/SparseTensor/Transforms/CMakeLists.txt
mlir/lib/Dialect/SparseTensor/Transforms/CodegenEnv.cpp [new file with mode: 0644]
mlir/lib/Dialect/SparseTensor/Transforms/CodegenEnv.h [new file with mode: 0644]
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp