pytorch/torch/_library
Arsh Zahed 110ae0f433 Custom Op handle 1-element tuples (#155447)
Fixes #150472

Modification of [PR 151408](https://github.com/pytorch/pytorch/pull/151408). This PR modifies the return parsing in `infer_schema` to handle the case of a Tuple with a single element.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/155447
Approved by: https://github.com/bdhirsh, https://github.com/zou3519
2025-06-11 03:43:40 +00:00
..
__init__.py fix typo in doc and import for torch._library.triton (#144882) 2025-01-17 17:32:12 +00:00
autograd.py [custom_ops][perf] Move expensive pytree traversals of tensors to C++ (#148555) 2025-04-01 18:45:48 +00:00
custom_ops.py Inductor respects exact strides on custom ops by default (#150511) 2025-05-03 00:02:24 +00:00
fake_class_registry.py [torchbind] fix error message when attr is a real tensor. (#151944) 2025-04-23 17:32:11 +00:00
fake_impl.py [custom ops] Override fake registration (#150806) 2025-04-12 02:43:47 +00:00
fake_profile.py Save/load op profiles (#151817) 2025-04-29 23:11:32 +00:00
infer_schema.py Custom Op handle 1-element tuples (#155447) 2025-06-11 03:43:40 +00:00
simple_registry.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
triton.py Typo Errors fixed in multiple files (#148262) 2025-03-09 12:21:40 +00:00
utils.py Add torch._C.Tag.needs_contiguous_strides (#152859) 2025-05-08 04:49:59 +00:00