@@ -37,7 +37,7 @@ using namespace ov::op;
3737namespace ov {
3838namespace test {
3939using InputShapes = std::vector<InputShape>;
40- using PagedAttnTestParams = std::tuple<ElementType, InputShapes, bool , bool , int32_t , ov::AnyMap>;
40+ using PagedAttnTestParams = std::tuple<ElementType, InputShapes, bool , bool , bool , int32_t , ov::AnyMap>;
4141
4242class PagedAttnTestBase : public testing ::WithParamInterface<PagedAttnTestParams>,
4343 virtual public ov::test::SubgraphBaseTest,
@@ -137,7 +137,7 @@ class PagedAttnTestBase : public testing::WithParamInterface<PagedAttnTestParams
137137 auto xattention_block_size =
138138 std::make_shared<ov::op::v0::Constant>(ov::element::i32 , Shape{}, std::vector<int32_t >{64 });
139139 auto xattention_stride =
140- std::make_shared<ov::op::v0::Constant>(ov::element::i32 , Shape{}, std::vector<int32_t >{0 });
140+ std::make_shared<ov::op::v0::Constant>(ov::element::i32 , Shape{}, std::vector<int32_t >{8 });
141141 // Create sink_input parameter for testing - shape [1, num_heads, 1, 1] as per PagedAttentionExecutor::ID_SINKS
142142 // PagedAttentionExtension always expects 21 inputs, so we must always include sinks parameter
143143 auto sinks = make_param (PartialShape{1 , head_num, 1 , 1 }, data_type, " sinks" );
@@ -656,6 +656,7 @@ INSTANTIATE_TEST_SUITE_P(smoke_PagedAttnVSSDPATest,
656656 ::testing::ValuesIn(inputShapeAndReorders),
657657 ::testing::Values(true , false ),
658658 ::testing::Values(true , false ),
659+ ::testing::Values(true , false ),
659660 ::testing::Values(0 ), // sliding_window = 0
660661 ::testing::ValuesIn(additional_configs)),
661662 PagedAttnTestBase::getTestCaseName);
@@ -666,6 +667,7 @@ INSTANTIATE_TEST_SUITE_P(smoke_PagedAttnVSSDPATest_WithSlidingWindow,
666667 ::testing::Combine (::testing::Values(ElementType::f32 ),
667668 ::testing::ValuesIn(inputShapeAndReorders),
668669 ::testing::Values(false ), // extendBlockIndices
670+ ::testing::Values(false ), // enableXattn
669671 ::testing::Values(true ), // sinkInput
670672 ::testing::Values(8 ), // sliding_window = 8
671673 ::testing::Values(ov::AnyMap{
@@ -909,7 +911,8 @@ INSTANTIATE_TEST_SUITE_P(smoke_PagedAttnVSMatmulTest,
909911 ::testing::ValuesIn(inputShapes),
910912 ::testing::Values(true , false ),
911913 ::testing::Values(true , false ),
912- ::testing::Values(0 ), // sliding_window = 0
914+ ::testing::Values(false ), // sinkInput = false
915+ ::testing::Values(0 ), // sliding_window = 0
913916 ::testing::ValuesIn(additional_configs)),
914917 PagedAttnTestBase::getTestCaseName);
915918} // namespace
0 commit comments