Skip to content

Commit f94994d

Browse files
committed
test conflict
Signed-off-by: xuchen-intel <chen.xu@intel.com>
1 parent 0d255cf commit f94994d

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

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

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5450,7 +5450,8 @@ TEST_F(SDPAToPATest, SDPAToPA_Gemma3_TokenTypeIds) {
54505450
auto sinks = v0::Constant::create(element::f32, Shape{0, 0, 0, 0}, {});
54515451

54525452
auto token_type_ids_i32 = makeOP<v0::Convert>({token_type_ids_param}, {{"destination_type", "i32"}});
5453-
5453+
auto qq_bias = v0::Constant::create(element::u8, Shape{0}, {});
5454+
auto qq_bias_begins = v0::Constant::create(element::i32, Shape{0}, {});
54545455
auto PA = std::make_shared<ov::op::PagedAttentionExtension>(
54555456
OutputVector{Q_flat,
54565457
K_flat,
@@ -5477,7 +5478,9 @@ TEST_F(SDPAToPATest, SDPAToPA_Gemma3_TokenTypeIds) {
54775478
adaptive_rkv_evictable_sizes,
54785479
adaptive_rkv_diversity_block_set_indices,
54795480
adaptive_rkv_diversity_block_set_indices_begins,
5480-
token_type_ids_i32});
5481+
token_type_ids_i32,
5482+
qq_bias,
5483+
qq_bias_begins});
54815484

54825485
auto ShapeOf_v = makeOP<v3::ShapeOf>({Transpose_v2}, {{"output_type", "i64"}});
54835486
auto Gather_dim = makeOP<v8::Gather>({ShapeOf_v, -1, 0}, {{"batch_dims", 0}});

0 commit comments

Comments
 (0)