From 096166912d4239c491cf899b983b631a46e3df8e Mon Sep 17 00:00:00 2001 From: amancini-N Date: Fri, 6 Dec 2024 11:44:14 +0000 Subject: [PATCH] Ensure test is executed on cpu and cuda EPs --- onnxruntime/test/contrib_ops/beam_search_test.cc | 5 +++-- ...s.onnx => dummy_t5_with_sequence_input_ids.onnx} | Bin 2 files changed, 3 insertions(+), 2 deletions(-) rename onnxruntime/test/testdata/{tiny_t5_with_sequence_input_ids.onnx => dummy_t5_with_sequence_input_ids.onnx} (100%) diff --git a/onnxruntime/test/contrib_ops/beam_search_test.cc b/onnxruntime/test/contrib_ops/beam_search_test.cc index cb2abd09bd418..9f0dd909c18be 100644 --- 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/onnxruntime/test/testdata/tiny_t5_with_sequence_input_ids.onnx b/onnxruntime/test/testdata/dummy_t5_with_sequence_input_ids.onnx similarity index 100% rename from onnxruntime/test/testdata/tiny_t5_with_sequence_input_ids.onnx rename to onnxruntime/test/testdata/dummy_t5_with_sequence_input_ids.onnx