Commit 57203d2
committed
example(eagle3): disable torch.compile for Qwen3-0.6B CI train
The recipe compiles the eagle forward with torch.compile(mode="max-autotune")
(hf_eagle.py), whose exhaustive Triton bmm autotuning logs benign
'out of resource: triton_bmm' errors on the L40 (it recovers and picks a working
config, task SUCCEEDs) and adds compile overhead. Disable it via
eagle.eagle_use_torch_compile=false for the CI smoke runs — clean logs, faster.
Coverage note: this stops exercising the compiled path; revisit once the recipe
lets the compile mode be tuned down instead of forced to max-autotune.
Signed-off-by: Chenhan Yu <chenhany@nvidia.com>
(cherry picked from commit d4b28a1)1 parent 5895b60 commit 57203d2
2 files changed
Lines changed: 8 additions & 0 deletions
Lines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
75 | 75 | | |
76 | 76 | | |
77 | 77 | | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
78 | 82 | | |
79 | 83 | | |
80 | 84 | | |
| |||
Lines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
44 | 44 | | |
45 | 45 | | |
46 | 46 | | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
47 | 51 | | |
48 | 52 | | |
49 | 53 | | |
| |||
0 commit comments