[Diffusion] Move diffusion time embedding to jit kernel (#16879)

2026-01-17 12:21:22 +08:00
parent a7b5f75d88
commit 2cdd4370bc
10 changed files with 228 additions and 206 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -32,7 +32,6 @@ 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,