[all-commits] [llvm/llvm-project] 56d607: [mlir][sparse][python] add an "exhaustive" sparse ...

Aart Bik via All-commits all-commits at lists.llvm.org
Thu Aug 12 11:13:19 PDT 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 56d607006d6baac2094818e3d2387b9c6fcf10ff
      https://github.com/llvm/llvm-project/commit/56d607006d6baac2094818e3d2387b9c6fcf10ff
  Author: Aart Bik <ajcbik at google.com>
  Date:   2021-08-12 (Thu, 12 Aug 2021)

  Changed paths:
    A mlir/test/python/dialects/sparse_tensor/test_SpMM.py

  Log Message:
  -----------
  [mlir][sparse][python] add an "exhaustive" sparse test using python

Using the python API to easily set up sparse kernels, this test
exhaustively builds, compilers, and runs SpMM for all annotations
on a sparse tensor, making sure every version generates the correct
result. This test also illustrates using the python API to set up
a sparse kernel and sparse compilation.

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D107943




More information about the All-commits mailing list