pytorch
808bdbab - Fix try/except flow where DataDependentOutputException is getting wrapped in a RuntimeError (#89314)

Commit
1 year ago
Fix try/except flow where DataDependentOutputException is getting wrapped in a RuntimeError (#89314) Repro fixed ``` def fn(a): return a.repeat_interleave(14, dim=0).repeat_interleave(14, dim=1) x = torch.ones(14, 14).to(dtype=torch.int64) opt_fn = torch._dynamo.optimize("eager")(fn) opt_fn(x) ``` Fixes [#1886](https://github.com/pytorch/torchdynamo/issues/1886) Pull Request resolved: https://github.com/pytorch/pytorch/pull/89314 Approved by: https://github.com/anijain2305, https://github.com/eellison
Author
Committer
Parents
Loading