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

Revert "Dispatch torch.norm to linalg.vector_norm and linalg.matrix_norm (#81761)"

Browse Source 
This reverts commit 65beff5acb0d7c0c484bd0558bcaf8ddc9c96aab.

Reverted https://github.com/pytorch/pytorch/pull/81761 on behalf of https://github.com/mehtanirav due to Breakages in pytorch/glow
This commit is contained in:
PyTorch MergeBot
2022-09-06 22:31:14 +00:00
parent 0e49bcfd41
commit 166dec74b5
13 changed files with 59 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_decomp.py
View File

@ -157,8 +157,8 @@ def op_assert_ref(test_case, op, test_dtype, i, orig, decomp, ref, args, kwargs)
(torch.float16, torch.ops.aten.native_layer_norm.default): 1e-5,
(torch.bfloat16, torch.ops.aten.native_batch_norm.default): 1e-5,
(torch.float16, torch.ops.aten.native_batch_norm.default): 1e-5,
(torch.bfloat16, torch.ops.aten.linalg_vector_norm.default): 1e-5,
(torch.float16, torch.ops.aten.linalg_vector_norm.default): 1e-5,
(torch.bfloat16, torch.ops.aten.linalg_vector_norm.default): 1e-6,
(torch.float16, torch.ops.aten.linalg_vector_norm.default): 1e-6,
}
if ref.is_floating_point():
orig_diff = (orig - ref).abs().max()
@ -283,7 +283,6 @@ CROSS_REF_EXCLUDE_SET = {
# See https://github.com/pytorch/pytorch/issues/81669
(None, None, "nn.functional.relu6"),
(None, None, "meshgrid"),
(None, None, "norm"),
}
all_decomposed = set()