Skip to content

Commit

Permalink
[dynamo] use lazy disable dynamo for manual seed (#125196)
Browse files Browse the repository at this point in the history 
Pull Request resolved: #125196
Approved by: https://github.com/fegin, https://github.com/yanboliang
  • Loading branch information
@anijain2305 @pytorchmergebot
anijain2305 authored and pytorchmergebot committed Apr 30, 2024
1 parent e784644 commit 8320b77
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/_dynamo/__init__.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand Down

1 comment on commit 8320b77

@pytorchmergebot
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Reverted #125196 on behalf of https://github.com/kit1980 due to breaking internal builds (comment)

Please sign in to comment.