Skip to content

fix(megatron_training_lib): compare verify_network_errors as the string it is - #334

Open
speriaswamy-amd wants to merge 1 commit into
mainfrom
speriaswamy/cvsman-01-verify-network-errors
Open

fix(megatron_training_lib): compare verify_network_errors as the string it is#334
speriaswamy-amd wants to merge 1 commit into
mainfrom
speriaswamy/cvsman-01-verify-network-errors

Conversation

@speriaswamy-amd

Copy link
Copy Markdown
Contributor

Part 1 of 14 in a stack for AIMVT-276.

Why

self.verify_network_errors is always the string 'True'/'False' parsed from JSON config, never a Python bool. if self.verify_network_errors is True: can therefore never be true, so the post-run RDMA/ethtool error-counter comparison silently never runs for distributed training jobs, regardless of what the config says.

What changed

  • cvs/lib/megatron_training_lib.py — compare with == 'True' instead of is True, matching the working pattern already used for the same field in torchtitan_training_lib.py.

…ng it is

self.verify_network_errors is always the string 'True'/'False' parsed from
JSON, so `is True` can never be true and the post-run RDMA/ethtool
error-counter comparison silently never runs. Matches the working
`== 'True'` pattern already used for the same field elsewhere.

AIMVT-276.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant