commit 90cf2c3315cf9c86d221718f0c44c767c1a9eb96
author Guray Ozen <guray.ozen@gmail.com> Thu Apr 18 09:58:24 2024 +0200
committer Copybara-Service <copybara-worker@google.com> Thu Apr 18 01:15:17 2024 -0700
tree 4bb8bfca4b4651511c59a1f398223310350e1549
parent 544bc5b201e62654c5f01fef95ab2f73da890e5d

[mlir][nvgpu] Simplify TMA IR generation (#87153)

This PR add `TmaDescriptorBuilder`
- class simplifies TMA generation.
- Makes the code ready to support various Tma configurations
   - removes strings and use the enums from `mlir.nvgpu.ENUMs`.
- Example "swizzle = swizzle_128b, l2promo=none, oob=zero,
interleave=none" to enums in `mlir.nvgpu` dialects.
- Enums have string equivalent that are used during the IR writing and
generation (see `TmaDescriptorBuilder::tensormap_descriptor_ty`).
- Improves readability and abstracts out TMA descriptor builders in
reusable component.

---------

Co-authored-by: Manish Gupta <manigupta@google.com>
GitOrigin-RevId: c82f45f9deb98aa383b0be37f8edc32f9e48c4ec