Skip to content

Commit 339f91b

Browse files
fix format issue
Signed-off-by: HU Yuan2 <[email protected]>
1 parent e9cf458 commit 339f91b

File tree

2 files changed

+10
-16
lines changed

2 files changed

+10
-16
lines changed

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

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -312,15 +312,9 @@ static std::shared_ptr<ov::Model> buildROPE_VIT(const int seq_length,
312312
auto seq_length_s = static_cast<size_t>(seq_length);
313313
auto rotary_ndims_s = static_cast<size_t>(rotary_ndims);
314314
auto num_heads_s = static_cast<size_t>(num_heads);
315-
auto input = std::make_shared<ov::opset1::Parameter>(ov::element::f32,
316-
ov::Shape{seq_length_s, num_heads_s, rotary_ndims_s});
317-
auto Constant_396096 = makeConst(ov::element::f32,
318-
ov::Shape({
319-
1,
320-
1,
321-
1,
322-
}),
323-
{-1.000000f});
315+
auto input =
316+
std::make_shared<ov::opset1::Parameter>(ov::element::f32, ov::Shape{seq_length_s, num_heads_s, rotary_ndims_s});
317+
auto Constant_396096 = makeConst(ov::element::f32, ov::Shape({1, 1, 1}), {-1.000000f});
324318

325319
auto param_cos =
326320
std::make_shared<ov::opset1::Parameter>(ov::element::f32, ov::Shape{seq_length_s, 1, rotary_ndims_s});
@@ -329,12 +323,14 @@ static std::shared_ptr<ov::Model> buildROPE_VIT(const int seq_length,
329323
ov::Output<ov::Node> cat_Concat;
330324
if (is_split) {
331325
auto split = makeOP<ov::opset1::VariadicSplit>({input, {2}, {rotary_ndims / 2, rotary_ndims / 2}});
332-
auto neg_Multiply = makeOP<ov::opset1::Multiply>({split->output(1), Constant_396096}, {{"auto_broadcast", "numpy"}});
326+
auto neg_Multiply =
327+
makeOP<ov::opset1::Multiply>({split->output(1), Constant_396096}, {{"auto_broadcast", "numpy"}});
333328
cat_Concat = makeOP<ov::opset1::Concat>({neg_Multiply, split->output(0)}, {{"axis", -1}});
334329
} else {
335330
auto slice_right_part = makeOP<ov::opset8::Slice>({input, {rotary_ndims / 2}, {INT_MAX}, {1}, {2}});
336331
auto slice_left_part = makeOP<ov::opset8::Slice>({input, {0}, {rotary_ndims / 2}, {1}, {2}});
337-
auto neg_Multiply = makeOP<ov::opset1::Multiply>({slice_right_part, Constant_396096}, {{"auto_broadcast", "numpy"}});
332+
auto neg_Multiply =
333+
makeOP<ov::opset1::Multiply>({slice_right_part, Constant_396096}, {{"auto_broadcast", "numpy"}});
338334
cat_Concat = makeOP<ov::opset1::Concat>({neg_Multiply, slice_left_part}, {{"axis", -1}});
339335
}
340336
auto mul_sin_Multiply = makeOP<ov::opset1::Multiply>({cat_Concat, param_sin}, {{"auto_broadcast", "numpy"}});
@@ -760,10 +756,8 @@ TEST_P(ConvertToROPETestVIT, ConvertToROPE_qwen) {
760756
{
761757
auto input =
762758
std::make_shared<ov::opset1::Parameter>(ov::element::f32, ov::Shape{seq_len, num_heads, rotary_ndims});
763-
auto param_cos =
764-
std::make_shared<ov::opset1::Parameter>(ov::element::f32, ov::Shape{seq_len, 1, rotary_ndims});
765-
auto param_sin =
766-
std::make_shared<ov::opset1::Parameter>(ov::element::f32, ov::Shape{seq_len, 1, rotary_ndims});
759+
auto param_cos = std::make_shared<ov::opset1::Parameter>(ov::element::f32, ov::Shape{seq_len, 1, rotary_ndims});
760+
auto param_sin = std::make_shared<ov::opset1::Parameter>(ov::element::f32, ov::Shape{seq_len, 1, rotary_ndims});
767761
auto rope = makeOP<ov::op::internal::RoPE>({input, param_cos, param_sin},
768762
{{"config.slice_start", 0},
769763
{"config.slice_stop", 0},

src/plugins/intel_gpu/src/graph/impls/ocl_v2/rope_opt.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ class RopeGenerator : public KernelGenerator {
154154
auto y = extract_channel(ChannelName::Y, out_l);
155155
wgs.global = {b, f, y * cfg.rotary_ndims / 2ul / vec_size};
156156
if (cfg.support_3d_rope) {
157-
wgs.global = {b , f, cfg.rotary_ndims / 2ul / vec_size};
157+
wgs.global = {b, f, cfg.rotary_ndims / 2ul / vec_size};
158158
}
159159
}
160160

0 commit comments

Comments
 (0)