[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
@@ -54,6 +54,7 @@ sources = [
    "csrc/speculative/eagle_utils.cu",
    "csrc/kvcacheio/transfer.cu",
    "csrc/elementwise/pos_enc.cu",
+    "csrc/sgl_diffusion/elementwise/timestep_embedding.cu",
 ]

 cxx_flags = ["-O3"]