-
Notifications
You must be signed in to change notification settings - Fork 2.8k
[CPU]Support RoPE for GLM4 #32682
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
[CPU]Support RoPE for GLM4 #32682
Conversation
b442d44 to
0e8211a
Compare
Signed-off-by: Zhang Yi <[email protected]>
0e8211a to
318f64c
Compare
|
|
||
| auto qkv_proj = pattern::wrap_type<v1::VariadicSplit>({reshape, 3, {"ndims", "ndims"}}); | ||
| qkv_proj->set_output_size(2); | ||
| auto slice_1 = NewGenSlice(reshape, 0, "ndims", 1, 3) | qkv_proj->output(0); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
it works with the default inputs only, if output ID is really important for this transformation, please use the output_index_matches() predicate, see https://github.com/openvinotoolkit/openvino/pull/31647/files
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed in cc8c9df
src/plugins/intel_cpu/tests/functional/shared_tests_instances/subgraph_tests/rotary_pos_emb.cpp
Show resolved
Hide resolved
cc8c9df to
502cd17
Compare
|
Hi @maxnick could you please take a review? Thanks! |
Details:
Tickets: