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

PEP585 update - torch/_dynamo (#145105)

Browse Source 
See #145101 for details.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145105
Approved by: https://github.com/bobrenjc93
This commit is contained in:
Aaron Orenstein
2025-01-18 08:56:06 -08:00
committed by PyTorch MergeBot
parent c95efc37ba
commit a79100ab11
69 changed files with 847 additions and 864 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/test_custom_ops.py
View File

@ -1003,7 +1003,7 @@ def _(x):
del foo
with self.assertRaisesRegex(ValueError, r"For example, typing.List\[int\]"):
with self.assertRaisesRegex(ValueError, r"For example, list\[int\]"):
# test that we propose a correct and supported type.
@torch.library.custom_op(f"{TestCustomOp.test_ns}::foo", mutates_args={})
def foo(x: Tensor, y: Tuple[int, int]) -> Tensor:
@ -3990,33 +3990,33 @@ class TestTypeConversion(TestCase):
]
def test_simple_tuple(self):
self.assertEqual(List, tuple_to_list(Tuple))
self.assertEqual(list, tuple_to_list(Tuple))
def test_supported_types(self):
for t in self.supported_base_types:
result_type = tuple_to_list(Tuple[t, t, t])
self.assertEqual(result_type, List[t])
self.assertEqual(result_type, list[t])
result_type = tuple_to_list(Tuple[t])
self.assertEqual(result_type, List[t])
self.assertEqual(result_type, list[t])
def test_optional(self):
for t in self.supported_base_types:
result_type = tuple_to_list(Tuple[t, Optional[t]])
self.assertEqual(result_type, List[Optional[t]])
self.assertEqual(result_type, list[Optional[t]])
result_type = tuple_to_list(Tuple[t, t, Optional[t]])
self.assertEqual(result_type, List[Optional[t]])
self.assertEqual(result_type, list[Optional[t]])
result_type = tuple_to_list(Tuple[t, ...])
self.assertEqual(result_type, List[t])
self.assertEqual(result_type, list[t])
def test_mixed_types(self):
result_type = tuple_to_list(Tuple[int, float])
self.assertEqual(result_type, List[typing.Union[int, float]])
self.assertEqual(result_type, list[typing.Union[int, float]])
result_type = tuple_to_list(Tuple[int, float, str])
self.assertEqual(result_type, List[typing.Union[int, float, str]])
self.assertEqual(result_type, list[typing.Union[int, float, str]])
only_for = ("cpu", "cuda")