Skip to content

Commit 02d8822

Browse files
code style
1 parent e463ae4 commit 02d8822

File tree

1 file changed

+5
-13
lines changed

1 file changed

+5
-13
lines changed

src/common/transformations/tests/common_optimizations/fuse_rotary_positional_embeddings.cpp

+5-13
Original file line numberDiff line numberDiff line change
@@ -1462,16 +1462,11 @@ TEST_F(TransformationTestsF, ConvertToROPE_chatGLM4_PagedAttention_GPU) {
14621462
auto input0 = std::make_shared<ov::opset1::Parameter>(ov::element::f16, ov::PartialShape{DYN, 1, 4608});
14631463
auto input1 = std::make_shared<ov::opset1::Parameter>(ov::element::f16, ov::PartialShape{DYN, 1, 32, 2});
14641464

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}});
14701468
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}},
14751470
{{"begin_mask", {1, 1, 1, 0}},
14761471
{"end_mask", {1, 1, 1, 0}},
14771472
{"new_axis_mask", {}},
@@ -1512,10 +1507,7 @@ TEST_F(TransformationTestsF, ConvertToROPE_chatGLM4_PagedAttention_GPU) {
15121507
auto aten_stack = makeOP<opset1::Concat>({Unsqueeze_81695, Unsqueeze_81696}, {{"axis", -1}});
15131508
auto aten_flatten_Reshape = makeOP<opset1::Reshape>({aten_stack, {0, 32, 0, 64}}, {{"special_zero", true}});
15141509
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}},
15191511
{{"begin_mask", {1, 1, 1, 0}},
15201512
{"end_mask", {1, 1, 1, 0}},
15211513
{"new_axis_mask", {}},

0 commit comments

Comments
 (0)