mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
Re-enable some functionalize tests (#82363)
As far as I can tell, #81145 got merged ( https://github.com/pytorch/pytorch/pull/81145 was reverted, but the revert was reverted https://github.com/pytorch/pytorch/pull/82189) These tests were originally disabled to coordinate a merge across pytorch/pytorch and pytorch/functorch repos, since the PR got merged we can reenable these tests. It is a little fishy that nothing had to change, but all the tests seem to pass. Fixes https://github.com/pytorch/functorch/issues/974 Pull Request resolved: https://github.com/pytorch/pytorch/pull/82363 Approved by: https://github.com/ezyang
This commit is contained in:
parent
c6c0366999
commit
cd28b3cd5c
1 changed files with 0 additions and 4 deletions
|
|
@ -3044,7 +3044,6 @@ def forward(self, x_1, indices_1) -> torch.Tensor:
|
|||
out2 = vmap(functionalize(jvp_wrapper))(x, t)
|
||||
self.assertEqual(out1, out2)
|
||||
|
||||
@unittest.skip("Disabling to land #81145")
|
||||
def test_functionalize_fx_simple(self, device):
|
||||
|
||||
def f(x: torch.Tensor) -> torch.Tensor:
|
||||
|
|
@ -3084,7 +3083,6 @@ def forward(self, x_1) -> torch.Tensor:
|
|||
return transpose_copy_int
|
||||
""")
|
||||
|
||||
@unittest.skip("Disabling to land #81145")
|
||||
def test_functionalize_fx_out_op(self, device):
|
||||
|
||||
def f(inpt: torch.Tensor) -> torch.Tensor:
|
||||
|
|
@ -3109,7 +3107,6 @@ def forward(self, inpt_1) -> torch.Tensor:
|
|||
return view_copy_default_2
|
||||
""")
|
||||
|
||||
@unittest.skip("Disabling to land #81145")
|
||||
def test_functionalize_fx_multi_out_op(self, device):
|
||||
|
||||
def f(inpt: torch.Tensor) -> torch.Tensor:
|
||||
|
|
@ -3135,7 +3132,6 @@ def forward(self, inpt_1) -> torch.Tensor:
|
|||
return (view_copy_default_1, getitem)
|
||||
""")
|
||||
|
||||
@unittest.skip("Disabling to land #81145")
|
||||
def test_functionalize_fx_reapply_views_simple(self, device):
|
||||
|
||||
def f(x: torch.Tensor) -> torch.Tensor:
|
||||
|
|
|
|||
Loading…
Reference in a new issue