@@ -382,7 +382,7 @@ TEST(StreamKTilePartitioner_PersistentConstructor, SKOnly)
382382 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
383383
384384 StreamKTilePartitionerV2PersistentExpected expected_values{0 , 0 , 3 };
385- validate_streamk_v2_persistent <Config::GemmShape>(expected_values, tile_partitioner);
385+ validate_streamk_persistent <Config::GemmShape>(expected_values, tile_partitioner);
386386}
387387
388388TEST (StreamKTilePartitioner_PersistentConstructor, DPOnly)
@@ -395,7 +395,7 @@ TEST(StreamKTilePartitioner_PersistentConstructor, DPOnly)
395395 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
396396
397397 StreamKTilePartitionerV2PersistentExpected expected_values{2 , 0 , 3 };
398- validate_streamk_v2_persistent <Config::GemmShape>(expected_values, tile_partitioner);
398+ validate_streamk_persistent <Config::GemmShape>(expected_values, tile_partitioner);
399399}
400400
401401TEST (StreamKTilePartitioner_PersistentConstructor, DP2TileSK)
@@ -408,7 +408,7 @@ TEST(StreamKTilePartitioner_PersistentConstructor, DP2TileSK)
408408 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
409409
410410 StreamKTilePartitionerV2PersistentExpected expected_values{1 , 0 , 3 };
411- validate_streamk_v2_persistent <Config::GemmShape>(expected_values, tile_partitioner);
411+ validate_streamk_persistent <Config::GemmShape>(expected_values, tile_partitioner);
412412}
413413
414414TEST (StreamKTilePartitioner_PersistentConstructor, EdgeCase)
@@ -421,7 +421,7 @@ TEST(StreamKTilePartitioner_PersistentConstructor, EdgeCase)
421421 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
422422
423423 StreamKTilePartitionerV2PersistentExpected expected_values{0 , 1 , 4 };
424- validate_streamk_v2_persistent <Config::GemmShape>(expected_values, tile_partitioner);
424+ validate_streamk_persistent <Config::GemmShape>(expected_values, tile_partitioner);
425425}
426426
427427TEST (StreamKTilePartitioner_GridSize_Persistent, SKOnly)
@@ -460,7 +460,7 @@ TEST(StreamKTilePartitioner_NonPersistentConstructor, SKOnly)
460460 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
461461
462462 StreamKTilePartitionerV2NonPersistentExpected expected_values{0 , 0 , 0 , 3 };
463- validate_streamk_v2_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
463+ validate_streamk_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
464464}
465465
466466TEST (StreamKTilePartitioner_NonPersistentConstructor, DPOnly)
@@ -473,7 +473,7 @@ TEST(StreamKTilePartitioner_NonPersistentConstructor, DPOnly)
473473 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
474474
475475 StreamKTilePartitionerV2NonPersistentExpected expected_values{6 , 0 , 6 , 3 };
476- validate_streamk_v2_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
476+ validate_streamk_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
477477}
478478
479479TEST (StreamKTilePartitioner_NonPersistentConstructor, DP2TileSK)
@@ -486,7 +486,7 @@ TEST(StreamKTilePartitioner_NonPersistentConstructor, DP2TileSK)
486486 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
487487
488488 StreamKTilePartitionerV2NonPersistentExpected expected_values{3 , 0 , 3 , 3 };
489- validate_streamk_v2_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
489+ validate_streamk_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
490490}
491491
492492TEST (StreamKTilePartitioner_NonPersistentConstructor, EdgeCase)
@@ -499,7 +499,7 @@ TEST(StreamKTilePartitioner_NonPersistentConstructor, EdgeCase)
499499 tile_partitioner{Config::M, Config::N, Config::K, Config::GRID};
500500
501501 StreamKTilePartitionerV2NonPersistentExpected expected_values{1 , 0 , 1 , 4 };
502- validate_streamk_v2_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
502+ validate_streamk_nonpersistent <Config::GemmShape>(expected_values, tile_partitioner);
503503}
504504
505505TEST (StreamKTilePartitioner_GridSize_NonPersistent, DP2TileSK)
0 commit comments