tile-ai / tile-ai/tilelang

[Feature request] `finalize_reducer` support vectorization and more op type

Open
#1,016 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
7.4k
Forks
745
Avg merge
1d 1h
Merged PRs (30d)
104

Description

Now `finalize_reducer` only support op in ("sum", "max", "min"). (See `tilelang/language/allocate.py` line 146)

We have `reduce_abssum` and `reduce_absmax`, which can use in `finalize_reducer`.

What's more, `finalize_reducer` now don't support vectorization. This can lead to inefficient kernel performance in certain cases.

Contributor guide

Open the contributing guide

Research direction

Start at tilelang/language/allocate.py around line 146 and trace how finalize_reducer handles the current sum, max, and min operations. Compare that path with reduce_abssum and reduce_absmax, then identify how vectorization is represented in the surrounding reducer logic. Done means the additional operations and vectorized finalize_reducer use cases are supported without the current performance limitation.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
compilers, performance
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
45/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.