llvm-project.git/mlir/test/Integration/Dialect/SparseTensor/python/test_SpMM.py, branch main

[mlir][sparse] fix bug with all-dense assembler (#108615)

2024-09-14T00:24:48+00:00

When only all-dense "sparse" tensors occur in a function prototype, the
assembler would skip the method conversion purely based on input/output
counts. It should rewrite based on the presence of any annotation,
however.

[mlir][sparse][pybind][CAPI] remove LevelType enum from CAPI, constru… (#81682)

2024-02-14T00:45:22+00:00

…ct LevelType from LevelFormat and properties instead.

**Rationale**
We used to explicitly declare every possible combination between
`LevelFormat` and `LevelProperties`, and it now becomes difficult to
scale as more properties/level formats are going to be introduced.

[mlir][sparse] rename DimLevelType to LevelType (#73561)

2023-11-27T22:27:52+00:00

The "Dim" prefix is a legacy left-over that no longer makes sense, since
we have a very strict "Dimension" vs. "Level" definition for sparse
tensor types and their storage.

Changed all code and comments that used the phrase "sparse compiler" to instead use "sparsifier" (#71875)

2023-11-15T20:12:35+00:00

The changes in this p.r. mostly center around the tests that use the
flag sparse_compiler (also: sparse-compiler).

[mlir][sparse] add COO to python tests (#70090)

2023-10-24T19:48:46+00:00

also typo fix

[mlir][sparse] extend sparse output test (#69986)

2023-10-24T04:33:54+00:00

This adds COO and loose compressed to output testing. Also prepares BSR
for output testing, but needs the conversion to work first. Cleanup of
stale TODOs

[mlir][sparse] Populate lvlToDim (#68937)

2023-10-17T20:09:39+00:00

Updates:
1. Infer lvlToDim from dimToLvl
2. Add more tests for block sparsity
3. Finish TODOs related to lvlToDim, including adding lvlToDim to python
binding

Verification of lvlToDim that user provides will be implemented in the
next PR.

[mlir][sparse] Combining `dimOrdering`+`higherOrdering` fields into `dimToLvl`

2023-05-30T22:19:50+00:00

This is a major step along the way towards the new STEA design.  While a great deal of this patch is simple renaming, there are several significant changes as well.  I've done my best to ensure that this patch retains the previous behavior and error-conditions, even though those are at odds with the eventual intended semantics of the `dimToLvl` mapping.  Since the majority of the compiler does not yet support non-permutations, I've also added explicit assertions in places that previously had implicitly assumed it was dealing with permutations.

Reviewed By: aartbik

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

[NFC][Py Reformat] Reformat python files in mlir subdir

2023-05-26T06:05:40+00:00

This is an ongoing series of commits that are reformatting our
Python code.

Reformatting is done with `black`.

If you end up having problems merging this commit because you
have made changes to a python file, the best way to handle that
is to run git checkout --ours  and then reformat it
with black.

If you run into any problems, post to discourse about it and
we will try to help.

RFC Thread below:

https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style

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

Reland "[mlir] Make the vast majority of intgration and runner tests work on Windows"

2023-02-15T18:14:43+00:00

This reverts commit 5561e174117ff395d65b6978d04b62c1a1275138

The logic was moved from cmake into lit fixing the issue that lead to the revert and potentially others with multi-config cmake generators

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