From fac005be27e29d0036d0c91c8c5894fce3aefabf Mon Sep 17 00:00:00 2001 From: LeoBorcherding Date: Mon, 11 May 2026 03:22:40 -0500 Subject: [PATCH] chore: trim c10d stub comment --- studio/backend/core/training/worker.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py index e7f8b7c79e..4593873383 100644 --- a/studio/backend/core/training/worker.py +++ b/studio/backend/core/training/worker.py @@ -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)