Ensure test is executed on cpu and cuda EPs

microsoft · Dec 6, 2024 · 0961669 · 0961669
1 parent f38c9d3
commit 0961669
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/onnxruntime/test/contrib_ops/beam_search_test.cc b/onnxruntime/test/contrib_ops/beam_search_test.cc
@@ -396,8 +396,9 @@ TEST(BeamSearchTest, GptBeamSearchFp16_VocabPadded) {
   }
 }
 
-TEST(BeamSearchTest, T5WithSequenceInputIds) {
-  ModelTester tester(CurrentTestName(), ORT_TSTR("testdata/tiny_t5_with_sequence_input_ids.onnx"));
+TEST(BeamSearchTest, DummyT5WithSequenceInputIds) {
+  ModelTester tester(CurrentTestName(), ORT_TSTR("testdata/dummy_t5_with_sequence_input_ids.onnx"));
+  tester.ConfigEp(DefaultCpuExecutionProvider());
   tester.AddInput("encoder_input_ids", {1, 5}, {16, 17, 1, 0, 8});
   tester.AddOutput("sequences", {1, 3, 10}, {2, 19, 18, 3, 8, 8, 8, 8, 8, 8, 2, 19, 18, 3, 10, 19, 18, 3, 8, 8, 2, 19, 18, 15, 13, 13, 13, 13, 13, 13});
 #ifdef USE_CUDA

diff --git a/...data/tiny_t5_with_sequence_input_ids.onnx → ...ata/dummy_t5_with_sequence_input_ids.onnx b/...data/tiny_t5_with_sequence_input_ids.onnx → ...ata/dummy_t5_with_sequence_input_ids.onnx