llvm/llvm-project

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

Open

#51,657 建立於 2021年10月26日

在 GitHub 查看
 (7 留言) (0 反應) (1 負責人)C++ (10,782 fork)batch import
bugzillagood first issuemlir:llvmmlir:sparsetensor

倉庫指標

Star
 (26,378 star)
PR 合併指標
 (平均合併 1天 2小時) (30 天內合併 1,000 個 PR)

描述

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.

貢獻者指南