diff --git a/3rdparty/amd/sgl-kernel/CMakeLists_rocm.txt b/3rdparty/amd/sgl-kernel/CMakeLists_rocm.txt
index 5d225d872..e4d29ae73 100644
--- a/3rdparty/amd/sgl-kernel/CMakeLists_rocm.txt
+++ b/3rdparty/amd/sgl-kernel/CMakeLists_rocm.txt
@@ -102,14 +102,15 @@ ${PROJ_ROOT}/csrc/allreduce/deterministic_all_reduce.hip
 ${PROJ_ROOT}/csrc/allreduce/quick_all_reduce.hip
 ${PROJ_ROOT}/csrc/common_extension_rocm.cc
 ${PROJ_ROOT}/csrc/elementwise/activation.hip
+${PROJ_ROOT}/csrc/elementwise/pos_enc.hip
 ${PROJ_ROOT}/csrc/elementwise/topk.hip
 ${PROJ_ROOT}/csrc/grammar/apply_token_bitmask_inplace_hip.hip
+${PROJ_ROOT}/csrc/kvcacheio/transfer.hip
 ${PROJ_ROOT}/csrc/moe/moe_align_kernel.hip
 ${PROJ_ROOT}/csrc/moe/moe_topk_softmax_kernels.hip
 ${PROJ_ROOT}/csrc/moe/moe_topk_sigmoid_kernels.hip
+${PROJ_ROOT}/csrc/sgl_diffusion/elementwise/timestep_embedding.hip
 ${PROJ_ROOT}/csrc/speculative/eagle_utils.hip
-${PROJ_ROOT}/csrc/kvcacheio/transfer.hip
-${PROJ_ROOT}/csrc/elementwise/pos_enc.hip
 )
 set_source_files_properties(
   ${SOURCES}
diff --git a/3rdparty/amd/sgl-kernel/rocm_hipify.py b/3rdparty/amd/sgl-kernel/rocm_hipify.py
index be83dd4b0..c758fe0f7 100644
--- a/3rdparty/amd/sgl-kernel/rocm_hipify.py
+++ b/3rdparty/amd/sgl-kernel/rocm_hipify.py
@@ -17,14 +17,15 @@ sources = [
     "csrc/allreduce/quick_all_reduce.cu",
     "csrc/common_extension_rocm.cc",
     "csrc/elementwise/activation.cu",
+    "csrc/elementwise/pos_enc.cu",
     "csrc/elementwise/topk.cu",
     "csrc/grammar/apply_token_bitmask_inplace_cuda.cu",
+    "csrc/kvcacheio/transfer.cu",
     "csrc/moe/moe_align_kernel.cu",
     "csrc/moe/moe_topk_softmax_kernels.cu",
     "csrc/moe/moe_topk_sigmoid_kernels.cu",
+    "csrc/sgl_diffusion/elementwise/timestep_embedding.cu",
     "csrc/speculative/eagle_utils.cu",
-    "csrc/kvcacheio/transfer.cu",
-    "csrc/elementwise/pos_enc.cu",
 ]
 
 libraries = ["hiprtc", "amdhip64", "c10", "torch", "torch_python"]