mirror of
https://github.com/pytorch/pytorch.git
synced 2026-10-28 18:56:51 +08:00
## Summary
- Add `torch::stable::{index_select,floor_divide,is_pinned}` Tensor–Tensor wrappers (gated at 2.14) via `torch_call_dispatcher`
- Document the new APIs in the stable operators docs
- Cover them in the 2.14 libtorch-agnostic extension tests
'stacked' on top of:
https://github.com/pytorch/pytorch/pull/192083
https://github.com/pytorch/pytorch/pull/191973
Will need to rebase once those are merged.
see commits https://github.com/pytorch/pytorch/pull/192097/changes/c4e0c5c1ef85bb93cd0d6e3cc46cff6446222db7..cb760bb831420502c80f49fa7881c6050a029807 for new edits in this PR
## Test plan
- [x] `python -m pytest test/cpp_extensions/test_libtorch_agnostic.py -k 'test_my_index_select or test_my_floor_divide or test_my_is_pinned' -v`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/192097
Approved by: https://github.com/janeyx99