We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 87f72eb commit 677cb0dCopy full SHA for 677cb0d
torchrec/distributed/train_pipeline/train_pipelines.py
@@ -1624,7 +1624,7 @@ def get_compiled_autograd_ctx(
1624
self.initialized = True
1625
return contextlib.nullcontext()
1626
1627
- return torch._dynamo.compiled_autograd.enable(
+ return torch._dynamo.compiled_autograd._enable(
1628
# pyre-ignore
1629
torch.compile(**self.compiled_autograd_options)
1630
)
0 commit comments