@@ -1462,16 +1462,11 @@ TEST_F(TransformationTestsF, ConvertToROPE_chatGLM4_PagedAttention_GPU) {
1462
1462
auto input0 = std::make_shared<ov::opset1::Parameter>(ov::element::f16, ov::PartialShape{DYN, 1 , 4608 });
1463
1463
auto input1 = std::make_shared<ov::opset1::Parameter>(ov::element::f16, ov::PartialShape{DYN, 1 , 32 , 2 });
1464
1464
1465
- auto ListUnpack =
1466
- makeOP<opset1::VariadicSplit>({input0, -1 , {4096 , 256 , 256 }});
1467
- auto aten_transpose = makeOP<opset1::Reshape>(
1468
- {ListUnpack->output (0 ), {-1 , 32 , 1 , 128 }},
1469
- {{" special_zero" , false }});
1465
+ auto ListUnpack = makeOP<opset1::VariadicSplit>({input0, -1 , {4096 , 256 , 256 }});
1466
+ auto aten_transpose =
1467
+ makeOP<opset1::Reshape>({ListUnpack->output (0 ), {-1 , 32 , 1 , 128 }}, {{" special_zero" , false }});
1470
1468
auto aten_slice_Slice =
1471
- makeOP<opset1::StridedSlice>({aten_transpose,
1472
- {0 , 0 , 0 , 0 },
1473
- {0 , 0 , 0 , 64 },
1474
- {1 , 1 , 1 , 1 }},
1469
+ makeOP<opset1::StridedSlice>({aten_transpose, {0 , 0 , 0 , 0 }, {0 , 0 , 0 , 64 }, {1 , 1 , 1 , 1 }},
1475
1470
{{" begin_mask" , {1 , 1 , 1 , 0 }},
1476
1471
{" end_mask" , {1 , 1 , 1 , 0 }},
1477
1472
{" new_axis_mask" , {}},
@@ -1512,10 +1507,7 @@ TEST_F(TransformationTestsF, ConvertToROPE_chatGLM4_PagedAttention_GPU) {
1512
1507
auto aten_stack = makeOP<opset1::Concat>({Unsqueeze_81695, Unsqueeze_81696}, {{" axis" , -1 }});
1513
1508
auto aten_flatten_Reshape = makeOP<opset1::Reshape>({aten_stack, {0 , 32 , 0 , 64 }}, {{" special_zero" , true }});
1514
1509
auto aten_slice_Slice_3 =
1515
- makeOP<opset1::StridedSlice>({aten_transpose,
1516
- {0 , 0 , 0 , 64 },
1517
- {0 , 0 , 0 , INT_MAX},
1518
- {1 , 1 , 1 , 1 }},
1510
+ makeOP<opset1::StridedSlice>({aten_transpose, {0 , 0 , 0 , 64 }, {0 , 0 , 0 , INT_MAX}, {1 , 1 , 1 , 1 }},
1519
1511
{{" begin_mask" , {1 , 1 , 1 , 0 }},
1520
1512
{" end_mask" , {1 , 1 , 1 , 0 }},
1521
1513
{" new_axis_mask" , {}},
0 commit comments