From 0c3102b4f9a6cbdd440ee243bbb2fe844a88c1e4 Mon Sep 17 00:00:00 2001 From: Joseph Groenenboom Date: Thu, 8 Jun 2023 21:25:56 +0000 Subject: [PATCH] Skip MultiHeadAttentionTest for ROCmEP The rest of the test suite is skipped for ROCm. Adding skips for new tests. --- onnxruntime/test/contrib_ops/multihead_attention_op_test.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/onnxruntime/test/contrib_ops/multihead_attention_op_test.cc b/onnxruntime/test/contrib_ops/multihead_attention_op_test.cc index 9ca8e72b58ad6..cd6fefc871aa2 100644 --- a/onnxruntime/test/contrib_ops/multihead_attention_op_test.cc +++ b/onnxruntime/test/contrib_ops/multihead_attention_op_test.cc @@ -476,6 +476,7 @@ TEST(MultiHeadAttentionTest, CrossAttention_Batch1_HeadSize32_LeftSidePadding_Ma } TEST(MultiHeadAttentionTest, CrossAttention_Batch2_HeadSize32_NoBias_NoMask_PackedKV) { + ROCM_GTEST_SKIP("ROCm MHA does not support mask"); MIGX_GTEST_SKIP("MIGX MHA does not support Packed KV"); AttentionTestData data; GetCrossAttentionData_Batch2_HeadSize32_NoBias_NoMask_PackedKV(data); @@ -483,6 +484,7 @@ TEST(MultiHeadAttentionTest, CrossAttention_Batch2_HeadSize32_NoBias_NoMask_Pack } TEST(MultiHeadAttentionTest, SelfAttention_Batch2_HeadSize32_NoBias_NoMask_PackedQKV) { + ROCM_GTEST_SKIP("ROCm MHA does not support mask"); MIGX_GTEST_SKIP("MIGX MHA does not support Packed QKV"); AttentionTestData data; GetSelfAttentionData_Batch2_HeadSize32_NoBias_NoMask_PackedQKV(data);