Peiming Liu
94e27c265a
[mlir][sparse] reuse tensor.insert operation to insert elements into … ( #84987 )
...
…a sparse tensor.
2024-03-12 16:59:17 -07:00
Peiming Liu
fc9f1d49aa
[mlir][sparse] use a consistent order between [dis]assembleOp and sto… ( #84079 )
...
…rage layout.
2024-03-06 09:57:41 -08:00
Peiming Liu
07bf1ddb4e
[mlir][sparse] support non-id map for [Dis]assembleOp ( #80355 )
2024-02-01 15:11:33 -08:00
Peiming Liu
573c4db947
[mlir][sparse] refine reinterpret_map test cases ( #72684 )
2023-11-17 10:04:56 -08:00
Yinying Li
c5a67e16b6
[mlir][sparse] Use variable instead of inlining sparse encoding ( #72561 )
...
Example:
#CSR = #sparse_tensor.encoding<{
map = (d0, d1) -> (d0 : dense, d1 : compressed),
}>
// CHECK: #[[$CSR.*]] = #sparse_tensor.encoding<{ map = (d0, d1) -> (d0
: dense, d1 : compressed) }>
// CHECK-LABEL: func private @sparse_csr(
// CHECK-SAME: tensor<?x?xf32, **#[[$CSR]]**>)
func.func private @sparse_csr(tensor<?x?xf32, #CSR>)
2023-11-16 19:30:21 -05:00
Peiming Liu
c99951d491
[mlir][sparse] end-to-end matmul between Dense and BSR tensors ( #71448 )
2023-11-08 11:28:00 -08:00
Peiming Liu
c0d78c4232
[mlir][sparse] Implement rewriters to reinterpret maps on alloc_tenso… ( #70993 )
...
…r operation
2023-11-01 18:15:11 -07:00
Peiming Liu
3426d330a7
[mlir][sparse] Implement rewriters to reinterpret maps on foreach ( #70868 )
2023-11-01 12:11:47 -07:00
Aart Bik
e599978760
[mlir][sparse] first proof-of-concept non-permutation rewriter ( #70863 )
...
Rather than extending sparsifier codegen with higher order
non-permutations, we follow the path of rewriting linalg generic ops
into higher order operations. That way, code generation will simply work
out of the box. This is a very first proof-of-concept rewriting of that
idea.
2023-10-31 16:19:27 -07:00
Aart Bik
7cfac1bedd
[mlir][sparse] add boilterplate code for a new reintepret map pass ( #70393 )
...
The interesting stuff is of course still coming ;-)
2023-10-26 17:57:46 -07:00
