TensorRT-LLMs/tensorrt_llm/_torch/modules/fla
bhsueh_NV 38d6e4e60b
[None][feat] Support Qwen3 next (#7892)
Signed-off-by: mengw <12670782+wm2012011492@users.noreply.github.com>
Signed-off-by: bhsueh <11360707+byshiue@users.noreply.github.com>
Signed-off-by: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com>
Co-authored-by: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com>
2025-09-29 21:16:07 +08:00
..
__init__.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
chunk_delta_h.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
chunk_o.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
chunk_scaled_dot_kkt.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
chunk.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
cumsum.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
fused_recurrent.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
fused_sigmoid_gating_recurrent.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
index.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
l2norm.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
layernorm_gated.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
op.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
solve_tril.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
utils.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00
wy_fast.py [None][feat] Support Qwen3 next (#7892) 2025-09-29 21:16:07 +08:00