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

[Mosaic GPU] Add a bank-conflict checker to tiled transfer + transfer planner #25149

Closed
wants to merge 0 commits into from

Conversation

copybara-service[bot]
Copy link

[Mosaic GPU] Add a bank-conflict checker to tiled transfer + transfer planner

Instead of only allowing a fixed set of layouts that we've hand verified as
bank-conflict free, we now simulate the transactions performed within each
warp and verify that no bank conflicts happen. If we detect that the simple
schedule does not work out, we attempt to partition the threads in a warp
into two groups and stagger the transfers in a way that lets us avoid conflicts.

This allows us to match the hand-designed transfer schedule I wrote for 32-bit
types, and even generalizes it to more cases automatically (e.g. swizzle=32).

@copybara-service copybara-service bot force-pushed the test_700692053 branch 2 times, most recently from a34392b to e7f3ed3 Compare December 2, 2024 12:14
@copybara-service copybara-service bot closed this Dec 2, 2024
@copybara-service copybara-service bot deleted the test_700692053 branch December 2, 2024 12:27
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.

1 participant