Cyrus Leung 3d9a1d2de5

[V1] Support LLM.apply_model (#18465) ...

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>

2025-09-20 07:14:35 +00:00

..

modular_kernel_tools

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

__init__.py

[Kernel] DeepEP dispatch-combine kernel integration (#18434)

2025-06-03 12:30:02 -07:00

parallel_utils.py

[Misc] Add unit tests for MoE ModularKernel combinations + Profiling utility (#20449)

2025-07-11 07:51:46 -07:00

test_batched_deepgemm.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_batched_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_block_fp8.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_block_int8.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_count_expert_num_tokens.py

[CI Perf] Prune tests in tests/kernels/moe/ (#22939)

2025-08-14 21:33:42 -06:00

test_cutlass_grouped_gemm.py

[Kernels] Clean up FusedMoeMethodBase and modular kernel setup. Remove extra arguments from modular kernel methods. (#22035)

2025-08-15 14:46:00 -04:00

test_cutlass_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_deepep_deepgemm_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_deepep_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_deepgemm.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_flashinfer_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_flashinfer.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_gpt_oss_triton_kernels.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_grouped_topk.py

[Kernel] Add fused grouped_topk kernel for MoE (#23274)

2025-08-25 11:47:52 -07:00

test_modular_kernel_combinations.py

Enable modelopt gemma3 nvfp4/fp8, make workflow more robust (#22771)

2025-09-19 22:40:33 +00:00

test_moe_align_block_size.py

[CI Perf] Prune tests in tests/kernels/moe/ (#22939)

2025-08-14 21:33:42 -06:00

test_moe_permute_unpermute.py

[Kernel] CUTLASS MoE FP8: Integrate cuda moe permute/unpermute (#23045)

2025-08-20 10:35:26 -04:00

test_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_mxfp4_moe.py

[V1] Support LLM.apply_model (#18465)

2025-09-20 07:14:35 +00:00

test_nvfp4_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_pplx_cutlass_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_pplx_moe.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

test_rocm_aiter_topk.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_silu_mul_fp8_quant_deep_gemm.py

[Kernels][DP/EP] Optimize Silu Kernel for R1 (#24054)

2025-09-13 00:17:27 -07:00

test_triton_moe_ptpc_fp8.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00

utils.py

[Kernel] Delegate construction of FusedMoEQuantConfig to FusedMoEMethodBase subclasses (#22537)

2025-09-17 17:43:31 -06:00