Skip to content

Make TP running but we have numerical error

fa9cf1e
Select commit
Loading
Failed to load commit list.
Closed

Integrate CUDA Graph 4D in test_megatron_e2e_pipeline.py #121

Make TP running but we have numerical error
fa9cf1e
Select commit
Loading
Failed to load commit list.
Cursor / Cursor Bugbot completed Dec 14, 2025 in 4m 49s

Bugbot Review

Bugbot Analysis Progress (4m 51s elapsed)

✅ Gathered PR context (2s)
✅ Analyzed code changes (1s)
✅ Completed bug detection — 5 potential bugs found (4m 36s)
✅ Validation and filtering completed (0s)
✅ Posted analysis results — 5 bugs reported (11s)
✅ Analysis completed successfully (0s)

Final Result: Bugbot completed review and found 5 potential issues

Request ID: serverGenReqId_4d9190f2-685c-429e-ac58-77e53506ed7e

Details

This PR is being reviewed by Cursor Bugbot

Details

You are on the Bugbot Free tier. On this plan, Bugbot will review limited PRs each billing cycle.

To receive Bugbot reviews on all of your PRs, visit the Cursor dashboard to activate Pro and start your 14-day free trial.


Bug: Commented import breaks function that still uses module

The import wlbllm.registry statement is commented out, but wlbllm.registry is still used in wlb_swap_next_forward_metadata() and wlb_swap_next_backward_metadata() functions. When these functions are called (when WLBLLM_MODE=1), they will raise a NameError because the module was never imported.

d2/runtime/megatron/forward_backward_func.py#L92-L101

https://github.com/GindaChen/d2/blob/fa9cf1ef0a1c35fb8639ece84530fcf66feb2c85/d2/runtime/megatron/forward_backward_func.py#L92-L101

Fix in Cursor Fix in Web