google / google/heir

rotate-and-reduce: apply for sums of contiguous regions exceeding a threshold

Open
#521 0 comments 0 reactions 0 assignees View on GitHub
dialect: tensor_ext newcomer project
Dominant language
MLIR
Stars
906
Forks
171
Avg merge
4d 12h
Merged PRs (30d)
32

Description

Add a pass option describing some cost function for which the rotate-and-reduce optimization is worth applying to a partial contiguous region of a tensor.

As @asraa mentioned in https://github.com/google/heir/pull/575, we could also try more exotic patterns like summing reductions that are not contiguous but well-strided.

Cf. https://github.com/google/heir/pull/517#discussion_r1520698063

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.