Zhanda ZhuandYubo Gao 5d5c776444 [Perf] FP8 FlashInfer Attn for ViT (#38065) ...

Signed-off-by: Zhanda Zhu <zhandazhu@gmail.com>
Co-authored-by: Yubo Gao <ybgao-nvidia@users.noreply.github.com>

2026-04-27 13:44:15 +08:00

..

test_activation.py

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

2026-03-12 07:57:47 -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_cpu_activation.py

[perf][cpu] Accelerate BF16 GELU with LUT impl on Arm CPUs (#37469)

2026-04-15 22:26:17 -07:00

test_fused_q_kv_rmsnorm.py

[Feat] DeepSeek V4 Rebased (#40860)

2026-04-26 18:31:08 -07:00

test_fused_qk_norm_rope.py

[Misc] Fix Current vLLM config is not set. warnings, assert to avoid issues in the future (#31747)

2026-01-08 15:20:49 -08:00

test_fused_quant_layernorm.py

Add platform manual_seed_all API (#38468)

2026-04-10 13:43:50 +08:00

test_fused_rms_norm_gated.py

[CustomOp] CustomOp FusedRMSNormGated (#35877)

2026-03-06 10:53:37 -08:00

test_fused_silu_mul_block_quant.py

[ROCm] Enable fused_silu_mul_block_quant on ROCm (#38817)

2026-04-08 11:23:32 -05:00

test_layernorm.py

[vLLM IR] rework gemma_rms_norm (#39014)

2026-04-07 01:37:00 -07:00

test_minimax_reduce_rms.py

[Kernel] Porting the TRTLLM minimax_allreduce_rms kernels (#37045)

2026-04-11 00:20:20 +08:00

test_mrope.py

[Bugfix] Standardize getting number of image patches/tokens (#34358)

2026-02-12 20:47:01 -08: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_rotary_embedding_mla_cache_fused.py

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

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

test_rotary_embedding.py

[CI] Bump mypy version (#34950)

2026-03-04 20:55:11 +00: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

[Perf] FP8 FlashInfer Attn for ViT (#38065)

2026-04-27 13:44:15 +08:00

test_vit_fp8_quant.py

[Perf] FP8 FlashInfer Attn for ViT (#38065)

2026-04-27 13:44:15 +08:00

test_vit_fp8_scaling.py

[Perf] FP8 FlashInfer Attn for ViT (#38065)

2026-04-27 13:44:15 +08:00