Peiming Liu
f607102a0d
[mlir][sparse] partially support lowering sparse coiteration loops to scf.while/for. ( #105565 )
2024-08-23 10:47:44 -07:00
Peiming Liu
c44202574f
[mlir][sparse] support sparsification to coiterate operations. ( #102546 )
2024-08-20 11:13:38 -07:00
Peiming Liu
951a363097
[mlir][sparse] implement sparse_tensor.extract_value operation. ( #101220 )
2024-07-31 14:47:29 -07:00
Peiming Liu
a02010b3e9
[mlir][sparse] support sparsifying sparse kernels to sparse-iterator-based loop ( #95858 )
2024-06-17 16:50:12 -07:00
Peiming Liu
7e2eeb5753
[mlir][sparse] use ValueRange instead of std::pair for iterator position. ( #90243 )
...
`ValueRange` is more easy to be extended (e.g., for padded iterator).
2024-04-29 10:47:07 -07:00
Peiming Liu
a454d92c5a
[mlir][sparse] rename files and unifies APIs ( #88162 )
2024-04-09 10:59:15 -07:00
Peiming Liu
52b69aa32f
[mlir][sparse] support sparsifying batch levels ( #83898 )
2024-03-04 14:39:06 -08:00
Peiming Liu
35fae044c5
[mlir][sparse] using non-static field to avoid data races. ( #81165 )
2024-02-08 10:12:24 -08:00
Peiming Liu
4a653b4df5
[mlir][sparse] Support pretty print to debug sparse iteration. ( #80207 )
2024-02-01 15:28:36 -08:00
Peiming Liu
298412b578
[mlir][sparse] setup SparseIterator to help generating code to traverse a sparse tensor level. ( #78345 )
2024-01-24 11:33:06 -08:00
Peiming Liu
cf4dd91165
[mlir][sparse] initialize slice-driven loop-related fields in one place ( #76099 )
2023-12-20 14:20:57 -08:00
Peiming Liu
21edad7d07
[mlir][sparse] set up the skeleton for SparseTensorLevel abstraction. ( #75645 )
...
Note that at the current moment, the newly-introduced
`SparseTensorLevel` classes are far from complete, we plan to migrate
code generation related to accessing sparse tensor levels to these
classes in the near future to simplify `LoopEmitter`.
2023-12-15 13:34:34 -08:00
Aart Bik
e52c941921
[mlir][sparse] minor cleanup of transform/utils ( #75396 )
...
Consistent include macro naming
Modified and added comments
2023-12-13 15:18:35 -08:00
Aart Bik
365777ecbe
[mlir][sparse] refactor utilities into transform/utils dir ( #75250 )
...
Separates actual transformation files from supporting utility files in
the transforms directory. Includes a bazel overlay fix for the build (as
well as a bit of cleanup of that file to be less verbose and more
flexible).
2023-12-12 15:34:31 -08:00
