Tyler Michael SmithandClaude Opus 4.6 7053a17886 Add barrier before mk.apply in MK test to sync ranks ...

DeepEP v2 dispatch is a collective — both ranks must enter it
together. Add barrier to prevent rank drift from setup overhead.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>

2026-05-03 20:50:09 -04:00

..

__init__.py

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

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

cli_args.py

[Feature]: Remove Chunking From FusedMoE (#34086)

2026-03-12 14:24:38 -04:00

common.py

Add barrier before mk.apply in MK test to sync ranks

2026-05-03 20:50:09 -04:00

make_feature_matrix.py

[Feature]: Remove Chunking From FusedMoE (#34086)

2026-03-12 14:24:38 -04:00

mk_objects.py

Add TrtLLM FP8 support for DeepEP v2 + modular kernel test framework

2026-05-03 16:57:03 -04:00

parallel_utils.py

Wrap MK test worker in set_current_vllm_config context

2026-05-03 17:14:15 -04:00

profile_modular_kernel.py

[Feature]: Remove Chunking From FusedMoE (#34086)

2026-03-12 14:24:38 -04:00