Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[CodePartition] Cleanup unused data structures #4

Open
wants to merge 10 commits into
base: ws
Choose a base branch
from

Conversation

htyu
Copy link
Contributor

@htyu htyu commented Dec 5, 2024

No description provided.

bertmaher and others added 10 commits November 18, 2024 20:47
… annotations

This commit is a squash generated by:
```
git diff --stat b62b221a...9755e293a -- . ':(exclude)python/gemmbench' ':(exclude)python/hstuBench' ':(exclude)third_party/proton'
```

Additional modifications:

- Update README.md with Warp Specialization Support
- Propagate mma layout to following elementwise operations.
… by a parent loop argument

Summary: Handling a case when slicing a forOP, the init argument value is from an outer loop arg.

Reviewed By: mren
Summary: Parent loop needs a rotating buffer id for its own loads, use the linear loop index to compute the right phase for its child loop instead, as indicated in D64357623.

Reviewed By: mren
Summary:
Allowing loops to be specialized with custom number of buffer specified by `num_stages`, e.g,

    for start_n in tl.range(0, N, BLOCK_N, num_stages=1):

Reviewed By: mren
Summary: Newly created ops with in one task region should have exactly the same task id with the region.

Reviewed By: mren
Test Plan:
python python/gemmbench/gemmbench.py
Summary:
Handle the following control flow
IfOp
ForOp
   ForOp
Also handle the case where the parentOp has different induction variable type from the forOp, in that case, we perform calculation in int64.
@htyu htyu requested review from bertmaher and manman-ren December 5, 2024 17:59
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Dec 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants