Rohan Potdar f38f8c9742 [ROCm]: Enable customop and rope+kvcache fusion for AITER RoPE (#35180) ...

Signed-off-by: Rohan138 <rohanpotdar138@gmail.com>

2026-02-25 04:36:40 +00:00

..

__init__.py

[Models] Intern-S1-Pro (#33636)

2026-02-03 05:49:45 -08:00

base.py

[ROCm]: Enable customop and rope+kvcache fusion for AITER RoPE (#35180)

2026-02-25 04:36:40 +00:00

common.py

[torch.compile] Compile CustomOp.forward_native for SiluAndMul and QuantFP8 to avoid raw torch ops inside opaque custom ops (#32806)

2026-01-22 19:52:26 -08:00

deepseek_scaling_rope.py

[Bugfix] Support RotaryEmbedding CustomOp for gpt-oss (#33800)

2026-02-04 20:17:41 +00:00

dual_chunk_rope.py

[Doc] Add developer guide for CustomOp (#30886)

2026-01-09 16:21:11 +00:00

dynamic_ntk_alpha_rope.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

dynamic_ntk_scaling_rope.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

ernie45_vl_rope.py

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

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

fope.py

[Models] Intern-S1-Pro (#33636)

2026-02-03 05:49:45 -08:00

linear_scaling_rope.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

llama3_rope.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

llama4_vision_rope.py

[Misc][LLaMa4] Compile LLaMa Vision Encoder (#30709)

2026-01-09 22:01:38 -05:00

mrope_interleaved.py

[model] Add support for openPangu7B-VL (#32449)

2026-01-30 15:54:27 +08:00

mrope.py

[Bugfix] Fix MRotaryEmbedding missing truncate attr with YaRN scaling (#35080)

2026-02-23 16:05:52 +00:00

ntk_scaling_rope.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

phi3_long_rope_scaled_rope.py

[Bugfix] In LongRoPE, decide short vs long based on max_model_len (#27431)

2025-10-28 12:00:56 +00:00

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