llvm/llvm-project

Implement more Python DSL-based "exhaustive" sparse kernel generators (easy starter)

Open

#51.657 aberto em 26 de out. de 2021

Ver no GitHub
 (7 comments) (0 reactions) (1 assignee)C++ (10.782 forks)batch import
bugzillagood first issuemlir:llvmmlir:sparsetensor

Métricas do repositório

Stars
 (26.378 stars)
Métricas de merge de PR
 (Mesclagem média 1d 2h) (1.000 fundiu PRs em 30d)

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.

Guia do colaborador