Shanshan Shen 23a1946e3b

[CustomOp][Refactor] Extract common methods for ApplyRotaryEmb CustomOp (#31021) ...

Signed-off-by: shen-shanshan <467638484@qq.com>

2025-12-19 22:16:09 +08:00

..

__init__.py

Standardise get_rope to use rope_parameters["partial_rotary_factor"], not rotary_dim (#30389)

2025-12-11 20:45:23 +00:00

base.py

[Perf] enable flashinfer rotary_embedding custom ops in DeepSeek rotary (#30729)

2025-12-18 14:31:18 -05:00

common.py

[CustomOp][Refactor] Extract common methods for ApplyRotaryEmb CustomOp (#31021)

2025-12-19 22:16:09 +08:00

deepseek_scaling_rope.py

[Perf] enable flashinfer rotary_embedding custom ops in DeepSeek rotary (#30729)

2025-12-18 14:31:18 -05:00

dual_chunk_rope.py

…

dynamic_ntk_alpha_rope.py

…

dynamic_ntk_scaling_rope.py

…

ernie45_vl_rope.py

[CustomOp] Extract ApplyRotaryEmb as CustomOp and unify the dispatch logic (#29873)

2025-12-15 19:08:16 -08:00

linear_scaling_rope.py

…

llama3_rope.py

…

llama4_vision_rope.py

…

mrope.py

[CustomOp] Extract ApplyRotaryEmb as CustomOp and unify the dispatch logic (#29873)

2025-12-15 19:08:16 -08:00

ntk_scaling_rope.py

…

phi3_long_rope_scaled_rope.py

…

xdrope.py

[CustomOp] Extract ApplyRotaryEmb as CustomOp and unify the dispatch logic (#29873)

2025-12-15 19:08:16 -08:00

yarn_scaling_rope.py

Add truncate arg to yarn to match openai implementation of gpt-oss (#28244)

2025-11-20 18:53:50 +08:00