Xiaoyu Zhang c1fe5de69c [Diffusion] Clean up diffusion Triton kernels and modernize custom op registration (#21122) ...

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

2026-03-22 22:38:57 +08:00

..

bench_awq_dequantize.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_clamp_position.py

Replace clamp_position with JIT kernel + platform dispatch (#20999)

2026-03-21 21:26:26 -07:00

bench_concat_mla.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_custom_all_reduce.py

[JIT Kernel][Feature] Support JIT custom all reduce (rewrite as v2) (#19880)

2026-03-20 18:24:07 +08:00

bench_fused_add_rmsnorm.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_fused_norm_scale_shift.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_hadamard.py

[JIT Kernel] Add hadamard kernel test and benchmark (#20030)

2026-03-17 23:16:35 -07:00

bench_hicache.py

[3/n jit_kernel restruct] Clean up benchmark naming and benchmarking helpers (#20250)

2026-03-10 16:39:03 +08:00

bench_norm_impls.py

[Diffusion] Fix torch.compile RMSNorm fallback for Z-Image (#20962)

2026-03-22 15:38:22 +08:00

bench_norm.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_nvfp4_blockwise_moe.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

bench_nvfp4_quant.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

bench_nvfp4_scaled_mm.py

[4/n jit_kernel restruct] speed up CI tests and add benchmark workflow (#20268)

2026-03-10 21:37:41 +08:00

bench_per_tensor_quant_fp8.py

add hicache jit test (#17847)

2026-02-06 16:54:33 +08:00

bench_per_token_group_quant_8bit.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_qknorm_across_heads.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_qknorm.py

[3/n jit_kernel restruct] Clean up benchmark naming and benchmarking helpers (#20250)

2026-03-10 16:39:03 +08:00

bench_qwen_image_modulation.py

[Diffusion] Clean up diffusion Triton kernels and modernize custom op registration (#21122)

2026-03-22 22:38:57 +08:00

bench_renorm.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00

bench_resolve_future_token_ids.py

Replace _resolve_future_token_ids with JIT kernel + platform dispatch (#20976)

2026-03-20 01:47:03 -07:00

bench_rmsnorm.py

[3/n jit_kernel restruct] Clean up benchmark naming and benchmarking helpers (#20250)

2026-03-10 16:39:03 +08:00

bench_rope.py

[kernel slimming] Clean many useless sgl-kernel deprecated kernels (#20277)

2026-03-14 16:45:54 +08:00

bench_store_cache.py

[kernel slimming] Clean many useless sgl-kernel deprecated kernels (#20277)

2026-03-14 16:45:54 +08:00

utils.py

refactor: consolidate is_in_ci (jit_kernel, sgl-kernel benchmarks, tests) (#21009)

2026-03-20 05:55:36 -07:00