bnellnm 015fab8c2f

[Kernels][Bugfix] Use torch op for all kernels in FusedMoE forward. Add additional testing for cudagraphs. (#19717) ...

Signed-off-by: Bill Nell <bnell@redhat.com>

2025-06-24 23:22:58 -07:00

..

__init__.py

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

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

deepep_utils.py

[Kernel] Integrate batched/masked deepgemm kernel (#19111)

2025-06-04 21:59:18 +00:00

test_batched_moe.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_cutlass_moe.py

[Kernels][Bugfix] Use torch op for all kernels in FusedMoE forward. Add additional testing for cudagraphs. (#19717)

2025-06-24 23:22:58 -07:00

test_deepep_deepgemm_moe.py

Revert "[Feature] Integrate new deepgemm (#19820)" (#20049)

2025-06-24 19:45:22 -07:00

test_deepep_moe.py

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

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

test_moe_align_block_size.py

[Perf] Optimize moe_align_block_size CUDA kernel (#19572)

2025-06-17 11:49:26 -07:00

test_moe_permute_unpermute.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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

test_moe.py

[Kernels][Bugfix] Use torch op for all kernels in FusedMoE forward. Add additional testing for cudagraphs. (#19717)

2025-06-24 23:22:58 -07:00

test_nvfp4_moe.py

[Kernels][Bugfix] Use torch op for all kernels in FusedMoE forward. Add additional testing for cudagraphs. (#19717)

2025-06-24 23:22:58 -07:00

test_pplx_cutlass_moe.py

[Kernels][Bugfix] Use torch op for all kernels in FusedMoE forward. Add additional testing for cudagraphs. (#19717)

2025-06-24 23:22:58 -07:00

test_pplx_moe.py

[Kernels][Bugfix] Use torch op for all kernels in FusedMoE forward. Add additional testing for cudagraphs. (#19717)

2025-06-24 23:22:58 -07: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

[EP+DP] Optimize the little operations in the DeepGEMM + DeepEP low latency case (#19885)

2025-06-23 11:07:47 -07:00

test_triton_moe_ptpc_fp8.py

[Misc] Add SPDX-FileCopyrightText (#19100)

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