Shangdi Yu
Shangdi Yu
encountering the same issue
With the changes in #131914, the code below can be exported (we convert autocast to HOO as discussed above). `with torch.device("meta")` still doesn't work (see #130067). ```python import torch from...
> @angelayi Hi, > > I have implemented the PR, and I am wondering if I need to add a unit test for it. If so, could you please advise...
lgtm. Let's wait for the CI checks to finish.
@pytorchbot merge
@pytorchbot merge
@pytorchbot merge -i
Got normal output like below in main: ``` tensor([[0.2307, 0.5860, 1.0000, 0.8029], [0.4501, 0.3431, 0.3892, 0.0574], [0.9867, 5.0000, 0.0976, 0.8971]]) ``` Feel free to re-open if issue persists.
@pytorchbot merge -i
There are some test failures because a call to `capture_pre_autograd_graph` is recently added in https://github.com/pytorch/pytorch/pull/139054. I'd have to wait until it's imported internally to overwrite it.