From a9d5f7d40023ee4c508dba395b1f9d7131b27f84 Mon Sep 17 00:00:00 2001
From: amancini-N <Andrea.Mancini@nuance.com>
Date: Mon, 9 Dec 2024 10:34:44 +0000
Subject: [PATCH] Skip tests on DML

---
 onnxruntime/test/contrib_ops/beam_search_test.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/onnxruntime/test/contrib_ops/beam_search_test.cc b/onnxruntime/test/contrib_ops/beam_search_test.cc
index 9f0dd909c18be..76c02c565f632 100644
--- a/onnxruntime/test/contrib_ops/beam_search_test.cc
+++ b/onnxruntime/test/contrib_ops/beam_search_test.cc
@@ -397,6 +397,9 @@ TEST(BeamSearchTest, GptBeamSearchFp16_VocabPadded) {
 }
 
 TEST(BeamSearchTest, DummyT5WithSequenceInputIds) {
+#if defined(USE_CUDA) && defined(USE_DML)
+  SKIP_CUDA_TEST_WITH_DML;
+#endif
   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});