Commit 6a18a12
tests: add MoE subgraph tests with require_full_compilation enforcement
Add tests/py/dynamo/hlo/test_moe.py covering all 7 MoE routing and
dispatch variants from popular open-source LLMs (Mixtral, Qwen2, Qwen3,
Llama4, DeepSeek-V2, DeepSeek-V3/R1, Nemotron-H). 50 parameterized
test cases verify TRT numerical correctness against PyTorch reference.
Extend DispatchTestCase.run_test() with require_full_compilation=True
support: calls TRTInterpreter.validate_conversion() before building the
engine and fails immediately if any op lacks a TRT converter. All MoE
tests pass this check, confirming zero PyTorch fallback for every
routing/dispatch/expert-MLP combination tested.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>1 parent 2e26bfa commit 6a18a12
3 files changed
Lines changed: 1048 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
422 | 422 | | |
423 | 423 | | |
424 | 424 | | |
| 425 | + | |
425 | 426 | | |
426 | 427 | | |
427 | 428 | | |
| |||
434 | 435 | | |
435 | 436 | | |
436 | 437 | | |
| 438 | + | |
437 | 439 | | |
438 | 440 | | |
439 | 441 | | |
| |||
499 | 501 | | |
500 | 502 | | |
501 | 503 | | |
| 504 | + | |
| 505 | + | |
| 506 | + | |
| 507 | + | |
| 508 | + | |
| 509 | + | |
| 510 | + | |
502 | 511 | | |
503 | 512 | | |
504 | 513 | | |
| |||
Whitespace-only changes.
0 commit comments