[dynamo] use lazy disable dynamo for manual seed (#125196)

Pull Request resolved: #125196 Approved by: https://github.com/fegin, https://github.com/yanboliang
pytorch · May 3, 2024 · cf525f6 · cf525f6
1 parent 9dcd72f
commit cf525f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_dynamo/__init__.py b/torch/_dynamo/__init__.py
@@ -61,7 +61,7 @@
 
     # Wrap manual_seed with the disable decorator.
     # Can't do it at its implementation due to dependency issues.
-    torch.manual_seed = disable(torch.manual_seed)
+    torch.manual_seed = torch._disable_dynamo(torch.manual_seed)
     # Add the new manual_seed to the builtin registry.
     torch.jit._builtins._register_builtin(torch.manual_seed, "aten::manual_seed")