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

2026-01-17 05:29:22 -08:00
parent 9c2530642c
commit 53609e5e5b
10 changed files with 206 additions and 228 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -32,6 +32,7 @@ from sgl_kernel.elementwise import (
    rmsnorm,
    rotary_embedding,
    silu_and_mul,
+    timestep_embedding,
 )
 from sgl_kernel.expert_specialization import (
    es_fp8_blockwise_scaled_grouped_mm,