4 Commits

Author	SHA1	Message	Date
Jane Xu	e8bfce9a43	Document how to use stack-based APIs with StableIValue (#155984) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/155984 Approved by: https://github.com/albanD, https://github.com/zou3519	2025-06-18 01:10:23 +00:00
Jane Xu	8a9c66bb70	Improve stable library apis per Scott's feedback (#152040) ... Following 3 suggestions: 1. inline at::Tensor arg 2. use uniq ptr of array vs std::vector 3. document the `std::optional<S>()` case Pull Request resolved: https://github.com/pytorch/pytorch/pull/152040 Approved by: https://github.com/swolchok, https://github.com/albanD	2025-04-24 20:51:03 +00:00
Scott Wolchok	2f74cffab2	Remove reinterpret_casts with undefined behavior from stable/library.h (#151595) ... There is a list of valid uses of `reinterpret_cast` (see https://en.cppreference.com/w/cpp/language/reinterpret_cast), and the use here was not on the list, hence undefined behavior. Implement what we meant using memcpy, which is well-defined. Differential Revision: [D73200791](https://our.internmc.facebook.com/intern/diff/D73200791/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/151595 Approved by: https://github.com/janeyx99	2025-04-22 20:24:47 +00:00
Jane Xu	988827cdfb	Use schema as source of truth + support ones_like/empty_like (#149052) ... This change does 2 important things: (a) Instead of relying on IValue type as source of truth, we use the schema as the source of truth, which is important as IValue types are overloaded and can ambiguously convert incorrectly. For example, a MemoryFormat will look like an int + get converted to an int64_t vs a MemoryFormat! (b) This PR expands support for many more types to encompass way more schemas, e.g., Optional, Device, dtype, etc. The main win from this PR is the ability for aoti_torch_call_dispatcher to call TensorFactory ops like ones_like/empty_like! Pull Request resolved: https://github.com/pytorch/pytorch/pull/149052 Approved by: https://github.com/albanD	2025-03-18 02:40:54 +00:00