Corey LowmanandMichael Goin d747c2ef18 [Perf] Fix jit compiles at runtime of fla gated delta rule (#25432) ...

Co-authored-by: Michael Goin <mgoin64@gmail.com>

2025-09-24 11:16:13 +08:00

..

__init__.py

[Kernels] Add Flash Linear Attention Kernels (#24518)

2025-09-10 00:04:41 +08:00

chunk_delta_h.py

Add the support for the qwen3 next model (a hybrid attention model). (#24526)

2025-09-11 15:32:09 +08:00

chunk_o.py

Add the support for the qwen3 next model (a hybrid attention model). (#24526)

2025-09-11 15:32:09 +08:00

chunk_scaled_dot_kkt.py

Add the support for the qwen3 next model (a hybrid attention model). (#24526)

2025-09-11 15:32:09 +08:00

chunk.py

[Kernels] Add Flash Linear Attention Kernels (#24518)

2025-09-10 00:04:41 +08:00

cumsum.py

fix some typos (#24616)

2025-09-11 10:48:46 -07:00

fused_recurrent.py

[Perf] Fix jit compiles at runtime of fla gated delta rule (#25432)

2025-09-24 11:16:13 +08:00

index.py

[Kernels] Add Flash Linear Attention Kernels (#24518)

2025-09-10 00:04:41 +08:00

l2norm.py

Add the support for the qwen3 next model (a hybrid attention model). (#24526)

2025-09-11 15:32:09 +08:00

layernorm_guard.py

[Kernels] Add Flash Linear Attention Kernels (#24518)

2025-09-10 00:04:41 +08:00

op.py

Add the support for the qwen3 next model (a hybrid attention model). (#24526)

2025-09-11 15:32:09 +08:00

solve_tril.py

[Kernels] Add Flash Linear Attention Kernels (#24518)

2025-09-10 00:04:41 +08:00

utils.py

[Kernels] Add Flash Linear Attention Kernels (#24518)

2025-09-10 00:04:41 +08:00

wy_fast.py

[Kernels] Add Flash Linear Attention Kernels (#24518)

2025-09-10 00:04:41 +08:00