Michael 53609e5e5b Revert "[Diffusion] Move diffusion time embedding to jit kernel" (#17257)

2026-01-17 21:29:22 +08:00

..

quantization

…

testing

…

__init__.py

Revert "[Diffusion] Move diffusion time embedding to jit kernel" (#17257)

2026-01-17 21:29:22 +08:00

_fa4_interface.py

[NVIDIA] upstream FA4 (#15182)

2026-01-11 15:31:28 +08:00

allreduce.py

…

attention.py

…

cutlass_moe.py

…

elementwise.py

Revert "[Diffusion] Move diffusion time embedding to jit kernel" (#17257)

2026-01-17 21:29:22 +08:00

expert_specialization.py

…

flash_attn.py

[NVIDIA] upstream FA4 (#15182)

2026-01-11 15:31:28 +08:00

flash_mla.py

…

fused_moe.py

…

gemm.py

…

grammar.py

…

hadamard.py

…

kvcacheio.py

…

load_utils.py

…

mamba.py

…

marlin.py

…

memory.py

…

moe.py

…

sampling.py

…

scalar_type.py

…

sparse_flash_attn.py

…

spatial.py

…

speculative.py

…

test_utils.py

…

top_k.py

…

utils.py

…

version.py

chore: bump sgl-kernel version to 0.3.21 (#16888)

2026-01-14 13:27:49 +08:00