llvm/llvm-project
Voir sur GitHubImplement more Python DSL-based "exhaustive" sparse kernel generators (easy starter)
Open
#51 657 ouverte le 26 oct. 2021
bugzillagood first issuemlir:llvmmlir:sparsetensor
Métriques du dépôt
- Stars
- (26 378 stars)
- Métriques de merge PR
- (Merge moyen 1j 2h) (1 000 PRs mergées en 30 j)
Description
| Bugzilla Link | 52315 |
| Version | unspecified |
| OS | Linux |
| CC | @joker-eph |
Extended Description
Currently, we have an interesting "exhaustive" test generator that constructs, runs, and verifies many versions of SpMM (see https://github.com/llvm/llvm-project/blob/main/mlir/test/python/dialects/sparse_tensor/test_SpMM.py) [note that we don't actually exhaustively generate all versions to keep testing time down, but the code can in principle generate and test all versions).
We need more interesting kernels like this to be "exhaustively" tested(or, for practical reasons, pick a good subset of versions). This entry encapsulates adding a number of interesting kernels to this directory using the same style of DSL based testing.