[Diffusion] Delete sgl-kernel outdated time_embedding kernel (#17278)

2026-01-28 14:18:53 +08:00
parent 67fb492c9a
commit fb74e43707
12 changed files with 0 additions and 326 deletions
@@ -219,18 +219,6 @@ TORCH_LIBRARY_EXPAND(sgl_kernel, m) {
      "                 Tensor!? key, int head_size,"
      "                 Tensor cos_sin_cache, bool is_neox) -> ()");
  m.impl("rotary_embedding", torch::kCUDA, &rotary_embedding);
-  /*
-   * From csrc/sgl_diffusion/elementwise
-   */
-  m.def(
-      "timestep_embedding(Tensor input,"
-      "Tensor output,"
-      "int dim,"
-      "bool flip_sin_to_cos,"
-      "float downscale_freq_shift,"
-      "float scale,"
-      "int max_period) -> Tensor");
-  m.impl("timestep_embedding", torch::kCUDA, &timestep_embedding);

  /*
   * From csrc/memory