Skip to content

Commit 3187ee6

Browse files
committed
Remove v2 from tile partitioner validation function names
1 parent ce8704d commit 3187ee6

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

test/ck_tile/gemm_streamk/test_streamk_tile_partitioner.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

388388
TEST(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

401401
TEST(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

414414
TEST(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

427427
TEST(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

466466
TEST(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

479479
TEST(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

492492
TEST(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

505505
TEST(StreamKTilePartitioner_GridSize_NonPersistent, DP2TileSK)

test/ck_tile/gemm_streamk/test_streamk_tile_partitioner_common.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -332,7 +332,7 @@ struct StreamKTilePartitionerV2NonPersistentExpected
332332

333333
// Persistent
334334
template <typename GemmShape>
335-
void validate_streamk_v2_persistent(
335+
void validate_streamk_persistent(
336336
StreamKTilePartitionerV2PersistentExpected& expected_values,
337337
ck_tile::StreamKTilePartitioner<GemmShape, ck_tile::StreamKReductionStrategy::Atomic, true>&
338338
tile_partitioner)
@@ -344,7 +344,7 @@ void validate_streamk_v2_persistent(
344344

345345
// Non-Persistent
346346
template <typename GemmShape>
347-
void validate_streamk_v2_nonpersistent(
347+
void validate_streamk_nonpersistent(
348348
StreamKTilePartitionerV2NonPersistentExpected& expected_values,
349349
ck_tile::StreamKTilePartitioner<GemmShape, ck_tile::StreamKReductionStrategy::Atomic, false>&
350350
tile_partitioner)

0 commit comments

Comments
 (0)