EnzymeAD / EnzymeAD/Enzyme-JAX
Structured Matrix / Tensors Detection
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
Listing out structured tensor that we can detect (ideally put these in a separate cc file and not with the HLO patterns):
- [ ] UpperTriangular
- [ ] UnitUpperTriangular
- [ ] LowerTriangular
- [ ] UnitLowerTriangular
- [ ] Banded
- [ ] Diagonal
- [ ] Bidiagonal
- [ ] Tridiagonal
- [ ] Symmetric
- [ ] Hermitian
- [ ] BlockSparse
- (usually, BlockDiagonal. ideally, block sparsity pattern is known at comptime)
- [ ] OneHot
(Feel free to add more here)
These enable us to lower to faster factorization routines, matmul, etc.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.