Unverified Commit 3133c192 authored by Tres's avatar Tres Committed by GitHub
Browse files

[ROCM] Reorder arguments and rename parameters for...


[ROCM] Reorder arguments and rename parameters for rope_cached_thd_positions_2c_fwd_inplace (#29993)
Signed-off-by: default avatarTres Popp <tres.popp@amd.com>
parent 76fd458a
......@@ -1374,14 +1374,14 @@ class rocm_aiter_ops:
key_ = key[..., :rotary_dim]
positions = positions.view(*query.shape[:1])
rope_cached_thd_positions_2c_fwd_inplace(
positions,
sin,
cos,
query_,
key_,
cos,
sin,
positions,
rotate_style,
reuse_freqs_front_part=True,
is_nope_first=False,
nope_first=False,
)
query = query.view(query_shape)
key = key.view(key_shape)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment