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.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Repro
import torch import torch.fx x = torch.randn(4, 5) mask = torch.randn(4, 5) > 0.5 def f(x, mask): # return x.masked_fill_(mask, 1.0) # PASSES return x.masked_fill_(mask, float("-inf")) print(f(x, mask)) # Only fails when symbolic_trace fx_mod = torch.fx.symbolic_trace(f) scripted_f = torch.jit.script(fx_mod) print(scripted_f(x, mask))
@eellison
The text was updated successfully, but these errors were encountered:
[JIT] Add support for closed over inf
00d11de
Fixes pytorch/torchdynamo#124 Pull Request resolved: #75439 Approved by: https://github.com/anijain2305, https://github.com/davidberard98
[JIT] Add support for closed over inf (#75439)
a393904
Summary: Fixes pytorch/torchdynamo#124 Pull Request resolved: #75439 Approved by: https://github.com/anijain2305, https://github.com/davidberard98 Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/00d11de564313ec4e59c40674da3f9f400de7089 Reviewed By: b0noI Differential Revision: D35485701 Pulled By: eellison fbshipit-source-id: a67310d07c5b344645f5584ac1ac1caa23c7f0b9
Successfully merging a pull request may close this issue.
Repro
@eellison
The text was updated successfully, but these errors were encountered: