Jee Jee Li e2c8eeac40 [kernel] Fused embedding kernel (#53677) ...

Signed-off-by: Jee Jee Li <jeejeelee@inferact.ai>

2026-08-31 11:34:14 +00:00

..

test_activation.py

[Perf][DSv4] Use native CUDA SwiGLU clamp kernel for Humming MoE (throughput +1.40%) (#53685)

2026-08-27 13:52:45 -07:00

test_apply_rotary_emb.py

[CustomOp] Extract ApplyRotaryEmb as CustomOp and unify the dispatch logic (#29873)

2025-12-15 19:08:16 -08:00

test_batched_weight_rms_norm.py

[DFlash] Fuse precompute kv per-layer rmsnorms (#46761)

2026-06-25 19:32:07 -07:00

test_cpu_activation.py

[CPU][Perf]Added tanh AOR for faster gelu activations. (#44639)

2026-06-30 23:24:40 -07:00

test_fused_allreduce_gemma_rms_norm.py

[Model] Add MiniMax M3 support (#45381)

2026-06-16 01:01:25 +08:00

test_fused_embed_norm.py

Replicated embedding and norm fusion for DSV3 flat model (#48484)

2026-08-18 14:53:21 +08:00

test_fused_q_kv_rmsnorm.py

[Bugfix] Allocate packed outputs in fused_q_kv_rmsnorm so q_b_proj keeps its low-latency GEMM path at decode (#53109)

2026-08-28 02:08:27 +00:00

test_fused_qk_norm_rope.py

[MyPy][1/N] Fix mypy errors in some tests/ directories and enforce follow-imports=silent (#49570)

2026-07-30 05:02:51 -07:00

test_fused_quant_layernorm.py

[Bugfix] Remove race in fused groupwise RMSNorm quantization (#54111)

2026-08-27 19:01:22 -07:00

test_fused_rms_norm_gated.py

[Misc] Add and enable Triton kernel unit tests on XPU (#45694)

2026-08-07 02:53:03 -07:00

test_fused_silu_mul_block_quant.py

[Bugfix] Fix int32 token offset overflow in fused SiLU block quant (#53409)

2026-08-28 16:23:36 -04:00

test_layernorm.py

[Test][ROCm] Account for gfx950 FP8 RMSNorm rounding (#49839)

2026-07-30 18:46:37 +08:00

test_minimax_reduce_rms.py

[Model]Fix MiniMaxM2ForCausalLM perf regression (#45935)

2026-06-22 00:28:52 +08:00

test_mrope.py

[Bugfix] Fix .../mrope.py::apply_interleaved_rope() when torch.compile is used in torch==2.13 (#52005)

2026-08-13 22:08:40 +00:00

test_opcheck.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_permute_cols.py

[CI][AMD] Skip test_permute_cols since the kernel is not used and not built for ROCm (#32444)

2026-01-16 16:22:53 +08:00

test_pos_encoding.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_rocm_aiter_ops.py

[ROCm][Perf] Fuse SWA q/kv RMSNorm and q FP8 group quant for DeepSeek-V4 (#53540)

2026-08-27 09:37:54 -07:00

test_rotary_embedding_mla_cache_fused.py

[ROCm] Relax MLA rope+cache test tolerances for bf16 (#51083)

2026-08-05 17:50:52 -05:00

test_rotary_embedding.py

[CI] fix test_rotary_embedding_opcheck format error (#41202)

2026-04-29 00:32:37 -07:00

test_uva.py

[Hardware] Replace torch.cuda.device_count/current_device/set_device API (#36145)

2026-03-12 07:57:47 -07:00

test_vit_bilinear_pos_embed.py

[Perf] triton bilinear_pos_embed kernel for ViT (#37948)

2026-04-01 01:52:02 -07:00

test_vit_fp8_attn.py

[MyPy][1/N] Fix mypy errors in some tests/ directories and enforce follow-imports=silent (#49570)

2026-07-30 05:02:51 -07:00

test_vit_fp8_quant.py

[ROCm][CI] fix fp8 range in vit_fp8_quant (#46410)

2026-06-23 11:34:21 -05:00

test_vit_fp8_scaling.py

[MyPy][1/N] Fix mypy errors in some tests/ directories and enforce follow-imports=silent (#49570)

2026-07-30 05:02:51 -07:00

test_vocab_parallel_embedding.py

[kernel] Fused embedding kernel (#53677)

2026-08-31 11:34:14 +00:00