Skip to content

Commit e0af233

Browse files
add constant to pattern
1 parent 4bb60fb commit e0af233

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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

+2-1
Original file line numberDiff line numberDiff line change
@@ -635,7 +635,8 @@ ov::pass::RoPEFusionChatGLM::RoPEFusionChatGLM(int split_output_id, const bool s
635635
auto ScatterUpdate = makePattern<opset3::ScatterUpdate>({{0, 0}, {1}, seq_length, {0}}, {});
636636
auto slice_Slice_449_1d = makePattern<ov::opset8::Slice>({cos_sin_cache, {0}, seq_length, {1}, {1}});
637637
auto slice_Slice_449_2d = makePattern<ov::opset8::Slice>({cos_sin_cache, {0, 0}, ScatterUpdate, {1, 1}, {0}});
638-
auto slice_StridedSlice_449 = GenStridedSlice(cos_sin_cache, {0, 0}, ScatterUpdate, {1, 1}, 1);
638+
auto ss_stop = makePattern<opset1::Constant>({}, {});
639+
auto slice_StridedSlice_449 = GenStridedSlice(cos_sin_cache, {0, 0}, ss_stop | ScatterUpdate, {1, 1}, 1);
639640

640641
// [batch, 1, seq_length, half_rotary_dims, 2]
641642
view_Reshape_460 = makePattern<opset1::Reshape>(

0 commit comments

Comments
 (0)