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

…

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

…

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

…