From 4cf67c80cf964ea028d1571f57fd769bc2156b5a Mon Sep 17 00:00:00 2001
From: Evgeny Kotov <evgeny.kotov@intel.com>
Date: Thu, 17 Jul 2025 13:17:21 +0200
Subject: [PATCH] extend unit test

---
 .../tests/common_optimizations/sdpa_fusion_test.cpp         | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/common/transformations/tests/common_optimizations/sdpa_fusion_test.cpp b/src/common/transformations/tests/common_optimizations/sdpa_fusion_test.cpp
index 650ed8769c15ae..da8ee9631f0554 100644
--- a/src/common/transformations/tests/common_optimizations/sdpa_fusion_test.cpp
+++ b/src/common/transformations/tests/common_optimizations/sdpa_fusion_test.cpp
@@ -273,7 +273,7 @@ implicit_transpose_3d(int B, int S_q, int S_kv, int E, int Ev, std::vector<Dimen
 
 INSTANTIATE_TEST_SUITE_P(SDPAFusion,
                          SDPAFusionImplicitTranspose,
-                         Combine(Values(f32, f16),                  // Types
+                         Combine(Values(f32, f16, bf16, f64),       // Types
                                  Values(true, false),               // Use attention_mask
                                  Values(true, false),               // Use scale
                                  Values(implicit_transpose_4d(1,    // B (batch)
@@ -369,7 +369,7 @@ TEST_P(SDPAFusionTransposeInMatmul, SDPAFusionTest_transpose_in_matmul) {
 
 INSTANTIATE_TEST_SUITE_P(SDPAFusion,
                          SDPAFusionTransposeInMatmul,
-                         Combine(Values(f32, f16),                  // Types
+                         Combine(Values(f32, f16, bf16, f64),       // Types
                                  Values(true, false),               // Use attention_mask
                                  Values(true, false),               // Use scale
                                  Values(explicit_transpose_4d(1,    // B (batch)
@@ -572,7 +572,7 @@ TEST_P(SDPAFusionExplicitTranspose, SDPAFusionTest_explicit_transpose) {
 
 INSTANTIATE_TEST_SUITE_P(SDPAFusion,
                          SDPAFusionExplicitTranspose,
-                         Combine(Values(f32, f16),                  // Types
+                         Combine(Values(f32, f16, bf16, f64),       // Types
                                  Values(true, false),               // Use attention_mask
                                  Values(true, false),               // Use scale
                                  Values(explicit_transpose_4d(1,    // B (batch)