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
Aart Bik
5f32bcfbae
[mlir][sparse][gpu] re-enable all GPU libgen tests ( #72185 )
...
Previous change no longer properly used the GPU libgen pass (even though
most tests still passed falling back to CPU). This revision puts the
proper pass order into place. Also bit of a cleanup of CPU codegen vs.
libgen setup.
2023-11-14 09:06:15 -08:00
Aart Bik
a4eadd7fb6
[mlir][sparse][gpu] add GPU BSR SDDMM check test ( #71491 )
...
also minor edits in other GPU check tests
2023-11-06 22:36:25 -08:00
Peiming Liu
6ca47eb49d
[mlir][sparse] rename sparse_tensor.(un)pack to sparse_tensor.(dis)as… ( #67717 )
...
…semble
Pack/Unpack are overridden in many other places, rename the operations
to avoid confusion.
2023-09-28 11:01:10 -07:00
Aart Bik
3e4a8c2c7d
[mlir][sparse] remove most bufferization.alloc_tensor ops from sparse ( #66847 )
...
The only ones left need actual deprecation in bufferization module.
2023-09-20 09:51:08 -07:00
Aart Bik
619a888dd8
[mlir][sparse][gpu] free all buffers allocated for spGEMM ( #66813 )
...
Yup, a bit of an oversight ;-)
2023-09-19 14:33:12 -07:00
Yinying Li
e2e429d994
[mlir][sparse] Migrate more tests to new syntax ( #66309 )
...
CSR:
`lvlTypes = [ "dense", "compressed" ]` to `map = (d0, d1) -> (d0 :
dense, d1 : compressed)`
CSC:
`lvlTypes = [ "dense", "compressed" ], dimToLvl = affine_map<(d0, d1) ->
(d1, d0)>` to `map = (d0, d1) -> (d1 : dense, d0 : compressed)`
This is an ongoing effort: #66146
2023-09-14 12:21:13 -04:00
Aart Bik
289f7231f9
[mlir][sparse][gpu] minor code cleanup for sparse gpu ops
...
Consistent order of ops and related methods.
Also, renamed SpGEMMGetSizeOp to SpMatGetSizeOp
since this is a general utility for sparse matrices,
not specific to GEMM ops only.
Reviewed By: Peiming
Differential Revision: https://reviews.llvm.org/D157922
2023-08-14 15:08:57 -07:00
Aart Bik
7a8dcf2ec7
[mlir][sparse][gpu] add CHECK test to spGEMM libgen
...
Reviewed By: K-Wu
Differential Revision: https://reviews.llvm.org/D157675
2023-08-11 09:56:13 -07:00
