pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

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 )