| commit | ee01d0807b924874a329be78c6ee880f556645db | [log] [tgz] |
|---|---|---|
| author | Jason Ansel <jansel@meta.com> | Sun Feb 25 12:28:08 2024 -0800 |
| committer | PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> | Mon Feb 26 17:16:31 2024 +0000 |
| tree | 5c7251cf06cef40588fc247fdfd61c1a4e582713 | |
| parent | 7eb7ac815f0247a62b621897cea95ec4ca56d52e [diff] |
[dynamo] Function => FunctionCtx for placeholder obj (#120577) Pull Request resolved: https://github.com/pytorch/pytorch/pull/120577 Approved by: https://github.com/yanboliang
diff --git a/torch/_dynamo/side_effects.py b/torch/_dynamo/side_effects.py index 67cdb8f..1e331ac 100644 --- a/torch/_dynamo/side_effects.py +++ b/torch/_dynamo/side_effects.py
@@ -242,7 +242,7 @@ options, ): if user_cls is torch.autograd.function.FunctionCtx: - obj = torch.autograd.Function() + obj = torch.autograd.function.FunctionCtx() elif issubclass(user_cls, torch.nn.Module): obj = nn_module_new(user_cls) else: