Skip to content

Conversation

@adstraw
Copy link
Contributor

@adstraw adstraw commented Aug 24, 2023

Adds codegen support only for CUDA async copy with barrier synchronization. TVM currently uses group based synchronization for async copies through the InjectSoftwarePipeline pass but new "bulk" async copy instructions for Hopper require barrier synchronization. This PR adds codegen for barrier synchronization as a first step towards adding support for Hopper "bulk" async copy support.

@tvm-bot
Copy link
Collaborator

tvm-bot commented Aug 24, 2023

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

  • No users to tag found in teams: hopper tma See #10317 for details

Generated by tvm-bot

@adstraw adstraw changed the title CUDA codegen for async copy with barrier synchronization [Hopper TMA] CUDA codegen for async copy with barrier synchronization Aug 24, 2023
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.

3 participants