Skip to content

Commit 6e635e5

Browse files
committed
clang
1 parent 63d1013 commit 6e635e5

File tree

1 file changed

+16
-3
lines changed

1 file changed

+16
-3
lines changed

src/common/transformations/tests/op_conversions/scaled_dot_product_decomposition_test.cpp

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -403,8 +403,13 @@ TEST_F(TransformationTestsF, ScaledDotProductAttentionDecomposition_PreScaledQue
403403

404404
{
405405
// Expected: scale applied to K^T (Q is pre-scaled)
406-
auto ref =
407-
scaled_dot_product_attention_decomposition(query_prescaled, key, value, attention_mask, sdpa_scale, casual, true);
406+
auto ref = scaled_dot_product_attention_decomposition(query_prescaled,
407+
key,
408+
value,
409+
attention_mask,
410+
sdpa_scale,
411+
casual,
412+
true);
408413
model_ref =
409414
std::make_shared<ov::Model>(OutputVector{ref}, ParameterVector{raw_query, key, value, attention_mask});
410415
}
@@ -436,7 +441,15 @@ TEST_F(TransformationTestsF, ScaledDotProductAttentionDecomposition_Sinks) {
436441
}
437442

438443
{
439-
auto ref = scaled_dot_product_attention_decomposition(query, key, value, attention_mask, scale, casual, false, false, sinks);
444+
auto ref = scaled_dot_product_attention_decomposition(query,
445+
key,
446+
value,
447+
attention_mask,
448+
scale,
449+
casual,
450+
false,
451+
false,
452+
sinks);
440453
model_ref = std::make_shared<ov::Model>(OutputVector{ref},
441454
ParameterVector{query, key, value, attention_mask, scale, sinks});
442455
}

0 commit comments

Comments
 (0)