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

Added support for symbolic is_contiguous (#84829)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84829
Approved by: https://github.com/ezyang
This commit is contained in:
Horace He
2022-09-16 02:29:13 +00:00
committed by PyTorch MergeBot
parent 5652ab22f6
commit 4bdc0af53d
10 changed files with 99 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_decomp.py
View File

@ -277,6 +277,8 @@ CROSS_REF_EXCLUDE_SET = {
("cuda", torch.float64, "nn.functional.dropout"),
("cuda", torch.float32, "nn.functional.dropout"),
(None, None, "new_empty"),
(None, None, "empty_like"),
(None, None, "empty"),
# decomp has problem even with opmath
# doesn't work
("cuda", torch.bfloat16, "nn.functional.embedding"),
@ -398,6 +400,8 @@ class TestDecomp(TestCase):
if func not in decomposition_table or func in [
torch.ops.aten.detach.default,
# non-deterministic ops
torch.ops.aten.empty.memory_format,
torch.ops.aten.empty_like.default,
torch.ops.aten.new_empty.default
] or any_unsupported(args, kwargs):
return func(*args, **kwargs)