Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit c322420

Browse files
Reactivate MLP3 for autotuning runs
1 parent 7b291e3 commit c322420

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

examples/example_MLP_model.cc

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -874,6 +874,23 @@ TEST_F(ProductionModel, C2MLP1Reference) {
874874
[&]() { return true; }, [&](bool flag) { reference->RunReference(); });
875875
}
876876

877+
TEST_F(ProductionModel, MLP3) {
878+
auto B = FLAGS_B;
879+
auto N = FLAGS_N;
880+
auto O = FLAGS_O;
881+
auto P = FLAGS_P;
882+
auto Q = FLAGS_Q;
883+
auto options = tc::MappingOptions::makeNaiveMappingOptions()
884+
.fixParametersBeforeScheduling(true)
885+
.tile({16, 16, 128})
886+
.mapToThreads({16, 16})
887+
.mapToBlocks({32, 32})
888+
.useSharedMemory(true)
889+
.usePrivateMemory(true)
890+
.unroll(1);
891+
runMLP3(B, N, O, P, Q, options, true);
892+
}
893+
877894
TEST_F(ProductionModel, MLP3_P100_autotuned_B_128_N_128_O_64_P_32_Q_2) {
878895
auto B = 128;
879896
auto N = 128;

0 commit comments

Comments
 (0)