ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: trim c10d stub comment

Browse source
This commit is contained in:
LeoBorcherding 2026-05-11 03:22:40 -05:00
commit fac005be27
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
studio/backend/core/training/worker.py
View file

@ -1104,12 +1104,8 @@ def run_training_process(
if _c10d_key not in sys.modules: # guard: never overwrite real NVIDIA impl
_c10d_stub = _types.ModuleType(_c10d_key)
# ROCm Windows wheels omit the _distributed_c10d C extension.
# torch._dynamo imports torch.distributed.fsdp at load time which
# pulls in FakeProcessGroup (and potentially other symbols) from
# this module. PEP-562 module __getattr__ auto-stubs any missing
# symbol so every `from torch._C._distributed_c10d import X`
# succeeds; each stub is a plain class cached after first access.
# ROCm Windows wheels omit this C extension; auto-stub every
# missing symbol so torch._dynamo's fsdp imports don't crash.
def _c10d_stub_getattr(_attr):
if _attr.startswith("__"):
raise AttributeError(_attr)