Commit a726781
committed
test: adapt real-quant assertions and refit diagnostics to vllm 0.25
- the real-quant test scripts assert on the engine log's quantization= tag,
which on vLLM 0.25 prints the registered NeMo config name
(e.g. nemo_modelopt_w4a16_nvfp4) instead of modelopt; accept both (the
w4a16-real GB200 run trained and produced healthy metrics but failed only
this grep)
- log the full traceback when an IPC weight batch load fails: the refit
manifest only records the exception message, which for bare assertions
(e.g. 'AssertionError: ') leaves nothing to diagnose in CI logs
Signed-off-by: Terry Kong <terryk@nvidia.com>1 parent 3b667dd commit a726781
5 files changed
Lines changed: 8 additions & 4 deletions
File tree
- nemo_rl/models/generation/vllm
- tests/test_suites/llm
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
656 | 656 | | |
657 | 657 | | |
658 | 658 | | |
| 659 | + | |
| 660 | + | |
| 661 | + | |
| 662 | + | |
659 | 663 | | |
660 | 664 | | |
661 | 665 | | |
| |||
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
38 | 38 | | |
39 | 39 | | |
40 | 40 | | |
41 | | - | |
| 41 | + | |
42 | 42 | | |
43 | 43 | | |
44 | 44 | | |
| |||
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
40 | 40 | | |
41 | 41 | | |
42 | 42 | | |
43 | | - | |
| 43 | + | |
44 | 44 | | |
45 | 45 | | |
46 | 46 | | |
| |||
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
40 | 40 | | |
41 | 41 | | |
42 | 42 | | |
43 | | - | |
| 43 | + | |
44 | 44 | | |
45 | 45 | | |
46 | 46 | | |
| |||
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
40 | 40 | | |
41 | 41 | | |
42 | 42 | | |
43 | | - | |
| 43 | + | |
44 | 44 | | |
45 | 45 | | |
46 | 46 | | |
| |||
0 commit comments