TJian 42172ad18f

[FEAT] [Performance] Add triton mrope to replace the torch code path (#22375) ...

Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>

2025-08-09 11:50:03 -07:00

..

deepgemm

[Refactor] Remove Duplicate per_block_cast_to_fp8, Remove Dependencies of DeepGEMM (#21787)

2025-08-01 01:13:27 +00:00

bench_fp8_gemm.py

[Misc] Add SPDX-FileCopyrightText (#20428)

2025-07-04 07:40:42 +00:00

bench_int8_gemm.py

[Benchmark] Refactor benchmark script for fp8 & int8 (#19627)

2025-06-15 15:15:37 +08:00

bench_nvfp4_gemm.py

[Bench] Add NVFP4 GEMM benchmark script (#20578)

2025-07-09 13:23:48 -04:00

bench_per_token_quant_fp8.py

[Perf][fp8] Use CustomOp abstraction for fp8 quant for better perf (#19830)

2025-07-11 04:56:28 +00:00

benchmark_aqlm.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_bitblas.py

[Bugfix] Add proper comparison for package versions (#22314)

2025-08-06 20:31:03 -07:00

benchmark_cutlass_fp4_moe.py

[Hardware][NVIDIA] FP4 MoE kernel optimization (#19110)

2025-06-05 09:48:26 -07:00

benchmark_grouped_gemm_cutlass.py

Revert "[Performance] Performance improvements in non-blockwise fp8 CUTLASS MoE (#20762) (#21334)

2025-07-21 21:49:01 -07:00

benchmark_layernorm.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_lora.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_machete.py

Enable ZP Support for Machete (#20268)

2025-07-01 07:12:20 +00:00

benchmark_marlin.py

[Bugfix][Benchmark] Fix Marlin benchmark (#19929)

2025-06-24 07:25:12 +09:00

benchmark_moe_align_block_size.py

[Refactor] Remove moe_align_block_size_triton (#21335)

2025-07-26 07:08:29 -07:00

benchmark_moe_permute_unpermute.py

Fix CUDA permute/unpermute for use with DeepGemm Moe (#17934)

2025-07-27 07:08:00 -07:00

benchmark_moe.py

[Misc] Minor enhancement of benchmark_moe (#22068)

2025-08-02 01:35:30 +08:00

benchmark_mrope.py

[FEAT] [Performance] Add triton mrope to replace the torch code path (#22375)

2025-08-09 11:50:03 -07:00

benchmark_paged_attention.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_per_token_group_quant.py

[Test] Add Benchmark and Unit Test for per_token_group_quant (#21860)

2025-07-30 07:15:02 -07:00

benchmark_quant.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_reshape_and_cache_flash.py

[Perf] Optimize reshape_and_cache_flash CUDA Kernel (#22036)

2025-08-01 19:18:51 -04:00

benchmark_rmsnorm.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_rope.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_shapes.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

benchmark_trtllm_decode_attention.py

[NVIDIA] Support Flashinfer TRT-LLM Prefill Attention Kernel (#22095)

2025-08-05 02:45:34 -07:00

benchmark_trtllm_prefill_attention.py

[NVIDIA] Support Flashinfer TRT-LLM Prefill Attention Kernel (#22095)

2025-08-05 02:45:34 -07:00

benchmark_w8a8_block_fp8.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

graph_machete_bench.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

requirements.txt

[Kernel] (2/N) Machete - Integrate into CompressedTensorsWNA16 and GPTQMarlin (#7701)

2024-09-23 13:46:26 -04:00

utils.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

weight_shapes.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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