frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add meta function for _upsample_bilinear2d_aa (#94982)

Browse Source 
Differential Revision: D43353000

Pull Request resolved: https://github.com/pytorch/pytorch/pull/94982
Approved by: https://github.com/ezyang
This commit is contained in:
Yanan Cao (PyTorch)
2023-02-19 07:11:18 +00:00
committed by PyTorch MergeBot
parent 17d0b7f532
commit 039b4c8809
9 changed files with 73 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_fx_experimental.py
View File

@ -1508,7 +1508,7 @@ class TestNormalizeOperators(JitTestCase):
@ops(op_db, allowed_dtypes=(torch.float,))
def test_normalize_operator_exhaustive(self, device, dtype, op):
# These ops currently don't trace in FX for various reasons (i.e. they take a list of tensors)
fx_fail = {"cat", "stack", "hstack", "vstack", "dstack", "linalg.multi_dot"}
fx_fail = {"cat", "stack", "hstack", "vstack", "dstack", "linalg.multi_dot", "_upsample_bilinear2d_aa"}
sample_inputs_itr = op.sample_inputs(device, dtype, requires_grad=False)
if isinstance(op.op, torch._ops.OpOverload):
self.skipTest("normalize operator doesn't work on torch.ops")