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 decomposition for reflection_pad{1,2,3}d_backward (#130299)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/130299
Approved by: https://github.com/lezcano
ghstack dependencies: #130130
This commit is contained in:
Isuru Fernando
2024-07-16 17:16:45 +00:00
committed by PyTorch MergeBot
parent 0eb43ed189
commit 43a6d20883
5 changed files with 101 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_decomp.py
View File

@ -213,6 +213,12 @@ def op_assert_ref(test_case, op, test_dtype, i, orig, decomp, ref, args, kwargs)
(torch.bfloat16, torch.ops.aten.multi_margin_loss.default): 5e-2,
(torch.float16, torch.ops.aten.multilabel_margin_loss_forward.default): 3e-2,
(torch.bfloat16, torch.ops.aten.multilabel_margin_loss_forward.default): 3e-2,
(torch.float16, torch.ops.aten.reflection_pad1d_backward.default): 5e-3,
(torch.bfloat16, torch.ops.aten.reflection_pad1d_backward.default): 5e-3,
(torch.float16, torch.ops.aten.reflection_pad2d_backward.default): 5e-3,
(torch.bfloat16, torch.ops.aten.reflection_pad2d_backward.default): 5e-3,
(torch.float16, torch.ops.aten.reflection_pad3d_backward.default): 5e-3,
(torch.bfloat16, torch.ops.aten.reflection_pad3d_backward.default): 5e-2,
# see https://github.com/pytorch/pytorch/pull/96264
(torch.float16, torch.ops.aten.mv.default): 1e-5,
(torch.bfloat16, torch.ops.aten.mv.default): 1e-5,