Commit a3cd2eeb authored by Nicolas Vasilache's avatar Nicolas Vasilache Committed by Nicolas Vasilache
Browse files

[mlir][nvgpu] Add a nvgpu.rewrite_copy_as_tma transform operation.

This revision adds support for direct lowering of a linalg.copy on buffers between global and shared memory to a tma async load + synchronization operations.
This uses the recently introduced Hopper NVVM and NVGPU abstraction to connect things end to end.

Differential Revision: https://reviews.llvm.org/D157087
parent b6d994de
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment