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.

贡献者指南