Fix type consistency in SingleTileWithTMALoadStoreStMatrix for fusion…

….manage. (#3613) This was causing errors in one of my development branches. We need to be consistent with types that we manage with fusion when storing and getting. Co-authored-by: Protonu <[email protected]>
NVIDIA · Dec 18, 2024 · a5f5e63 · a5f5e63
1 parent c31b919
commit a5f5e63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/cpp/test_mma.cpp b/tests/cpp/test_mma.cpp
@@ -434,8 +434,8 @@ TEST_P(HopperRSStmatrix, SingleTileWithTMALoadStoreStMatrix) {
   auto shapes = matmulAtInputShape3DHopperRS(
       getM(macro), getN(macro), getK(macro), layout);
 
-  auto tile_m = tile_sizes.at(0);
-  auto tile_n = tile_sizes.at(1);
+  int64_t tile_m = tile_sizes.at(0);
+  int64_t tile_n = tile_sizes.at(1);
 
   if (getM(macro) % tile_m || getN(macro) % tile_n) {
     GTEST_SKIP() << "skipping test as output is not divisible by tile size";