Prepare matmul schedulers for 2d grid traversal pattern

[rdspring1]

This PR prepares hopper matmul scheduler to use 2d grid traversal pattern.

• Rename grid_swizzle_factor to grid_traversal_factor in matmul schedulers.
• Changes grid_swizzle_factor from an int to std::pair<int, int>
• If grid_traversal_factor.second == 1, thengrid_traversal_factor.first == grid_swizzle_factor.
• Rename swizzleBlockTiles function to reorderBlockTileTraversal.

[github-actions]

Description

• Rename grid_swizzle_factor to grid_traversal_factor

• Change grid_traversal_factor type to std::pair<int, int>

• Rename swizzleBlockTiles to reorderBlockTileTraversal

---

Changes walkthrough 📝

Relevant files

Enhancement

9 files python_bindings.cpp Rename `grid_swizzle_factor` to `grid_traversal_factor`    +1/-1      ampere_multi_matmul.cpp Rename swizzleBlockTiles to reorderBlockTileTraversal and update grid_traversal_factor +8/-4      hopper_multi_matmul.cpp Rename swizzleBlockTiles to reorderBlockTileTraversal and update grid_traversal_factor +8/-4      matmul_heuristic_plugin.cpp Update `grid_traversal_factor` in config and params            +2/-2      matmul_utils.cpp Update `grid_traversal_factor` calculation                              +5/-6      test_matmul.cpp Update `grid_traversal_factor` in tests                                    +2/-2      ampere_multi_matmul.h Rename `swizzleBlockTiles` to `reorderBlockTileTraversal` +3/-2      hopper_multi_matmul.h Rename `swizzleBlockTiles` to `reorderBlockTileTraversal` +3/-2      matmul_heuristic.h Change grid_swizzle_factor to std::pair and update related methods +7/-6

---

PR Reviewer Guide 🔍

Here are some key observations to aid the review process:

🧪 PR contains tests

⚡ Recommended focus areas for review Grid Traversal Factor The code sets grid_traversal_factor to {grid_traversal_factor, 1} and disables 2D grid traversal. Ensure that this change does not inadvertently disable a feature that was previously enabled with grid_swizzle_factor. mparams->grid_traversal_factor = {grid_traversal_factor, 1}; Grid Swizzle Factor The code copies grid_traversal_factor.first to grid_swizzle_factor. Verify that this change is intentional and that the second element of grid_traversal_factor is not needed in the KernelConfig. config->grid_swizzle_factor = mparams->grid_traversal_factor.first; config->cta_order = Grid Swizzle Factor The code copies config->grid_swizzle_factor to grid_traversal_factor.first. Ensure that this change is intentional and that the second element of grid_traversal_factor is correctly handled. mparams->grid_traversal_factor.first = config->grid_swizzle_factor; switch (config->cta_order) {

[rdspring1]

!test

[jacobhinkle]

This looks good to me and is a step in the right direction. Maybe we could generalize it a bit further in the future by defining the operation recursively and holding a vector of ints instead of a tuple.