Tyler Michael SmithandClaude Opus 4.6 6a2d75c993 Wrap MK test worker in set_current_vllm_config context ...

The vllm config context was only active during init_distributed_environment
and initialize_model_parallel, but not during the actual test worker
execution. DeepEP v2's maybe_make_prepare_finalize calls
get_current_vllm_config() to read enforce_eager, which fails without
the context.

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

2026-05-03 17:14:15 -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 is_moe to test VllmConfig model_config

2026-05-03 17:01:41 -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