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

[MemoryBanking] Support memory banking forGetGlobalOp #8047

Merged
merged 12 commits into from
Jan 11, 2025

Conversation

jiahanxie353
Copy link
Contributor

This patch supports memory banking for multi-dimensional GetGlobalOps and restructure the data in their corresponding GlobalOps by the banking scheme (such as the banking factor and the dimension).

Copy link
Member

@cgyurgyik cgyurgyik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice job! Mostly style nits and a few smaller comments.

lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
test/Transforms/memory_banking_multi_dim.mlir Show resolved Hide resolved
lib/Transforms/MemoryBanking.cpp Show resolved Hide resolved
@jiahanxie353
Copy link
Contributor Author

Thanks @cgyurgyik for the review and suggestions! I've finished modifying the code based on the request. And it's ready for another round of review!

Copy link
Member

@cgyurgyik cgyurgyik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

lib/Transforms/MemoryBanking.cpp Outdated Show resolved Hide resolved
test/Transforms/memory_banking_multi_dim.mlir Show resolved Hide resolved
@jiahanxie353 jiahanxie353 merged commit bf9f0dc into llvm:main Jan 11, 2025
4 checks passed
@jiahanxie353 jiahanxie353 deleted the getglobal-membank branch January 11, 2025 03:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants