Skip to content

Commit add07dd

Browse files
committed
fix
1 parent bf95ce1 commit add07dd

File tree

1 file changed

+13
-11
lines changed

1 file changed

+13
-11
lines changed

onnxruntime/test/optimizer/transpose_optimizer_test.cc

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3827,17 +3827,19 @@ TEST(TransposeOptimizerTests, TestQLinearSoftmax) {
38273827
EXPECT_EQ(transpose_cost, 0);
38283828
};
38293829

3830-
TransformerTester(build_test_case_1,
3831-
check_optimized_graph_1,
3832-
TransformerLevel::Level2,
3833-
TransformerLevel::Level3,
3834-
/*opset_version*/ {13, 23},
3835-
/*per_sample_tolerance*/ 0.0,
3836-
/*relative_per_sample_tolerance*/ 0.0,
3837-
/*transformer*/ nullptr,
3838-
/*add_session_options*/ {},
3839-
/*disabled_optimizers*/ {},
3840-
/*ep*/ DefaultCpuExecutionProvider());
3830+
for (int opset_version : {13, 23}) {
3831+
TransformerTester(build_test_case_1,
3832+
check_optimized_graph_1,
3833+
TransformerLevel::Level2,
3834+
TransformerLevel::Level3,
3835+
/*opset_version*/ opset_version,
3836+
/*per_sample_tolerance*/ 0.0,
3837+
/*relative_per_sample_tolerance*/ 0.0,
3838+
/*transformer*/ nullptr,
3839+
/*add_session_options*/ {},
3840+
/*disabled_optimizers*/ {},
3841+
/*ep*/ DefaultCpuExecutionProvider());
3842+
}
38413843
}
38423844

38433845
TEST(TransposeOptimizerTests, TestBroadcastReusedInputs) {

0 commit comments

Comments
 (0)