Shangdi Yu

Results 32 comments of Shangdi Yu

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.

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.