[AMD][Diffusion] support timestep embedding kernel for AMD GPUs (#16766)

2026-01-12 22:17:07 -08:00
parent ff3ddb9d9b
commit 8716589826
9 changed files with 40 additions and 19 deletions
@@ -146,6 +146,7 @@ jobs:
          docker exec -w /sglang-checkout/sgl-kernel/tests ci_sglang python3 -m pytest test_activation.py
          docker exec -w /sglang-checkout/sgl-kernel/tests ci_sglang python3 -m pytest test_topk.py
          docker exec -w /sglang-checkout/sgl-kernel/tests ci_sglang python3 -m pytest test_kvcacheio.py
+          docker exec -w /sglang-checkout/sgl-kernel/tests/sgl_diffusion ci_sglang python3 -m pytest test_timestep_embedding.py

  # =============================================== primary ====================================================