[ROCM] Reorder arguments and rename parameters for rope_cached_thd_positions_2c_fwd_inplace (#29993)

Signed-off-by: Tres Popp <tres.popp@amd.com>
This commit is contained in:
Tres
2026-01-05 08:37:57 +01:00
committed by GitHub
parent 76fd458aa7
commit 3133c192a3

View File

@@ -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)