Skip to content

Commit d24276f

Browse files
janeyx99pytorchmergebot
authored andcommitted
Add supported StableIValue types to docs (pytorch#168385)
Pull Request resolved: pytorch#168385 Approved by: https://github.com/albanD
1 parent 2254961 commit d24276f

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

docs/source/notes/libtorch_stable_abi.md

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -163,21 +163,20 @@ You can always work with StableIValue abstractions in your custom kernel for typ
163163
| -------- | ------- | ------- | ------- |
164164
| std::optional\<S> | if there is a value, raw bitwise copy into leading bytes of uint64_t of pointer to a new StableIValue representing S. if there is no value, nullptr. | std::optional\<T> | Type? |
165165
| torch::stable::Tensor | raw bitwise copy of underlying AtenTensorHandle into leading bytes of uint64_t | at::Tensor | Tensor |
166-
| RAIIATH (outdated) | raw bitwise copy of underlying AtenTensorHandle into leading bytes of uint64_t | at::Tensor | Tensor |
167166
| torch::headeronly::ScalarType | raw bitwise copy of the translated underlying enum into leading bytes of uint64_t | torch::headeronly::ScalarType | ScalarType |
168-
| int32_t | raw bitwise copy into leading bytes of uint64_t | at::Layout | Layout |
169-
| int32_t | raw bitwise copy into leading bytes of uint64_t | at::MemoryFormat | MemoryFormat |
167+
| torch::headeronly::Layout | raw bitwise copy of the translated underlying enum into leading bytes of uint64_t | at::Layout | Layout |
168+
| torch::headeronly::MemoryFormat | raw bitwise copy of the translated underlying enum into leading bytes of uint64_t | at::MemoryFormat | MemoryFormat |
170169
| bool | raw bitwise copy into leading bytes of uint64_t | bool | bool |
171170
| int64_t | raw bitwise copy into leading bytes of uint64_t | int64_t | int |
172171
| double | raw bitwise copy into leading bytes of uint64_t | double | float |
173-
| ? | ? | c10::Device | Device |
172+
| torch::stable::Device | raw bitwise copy of index and type into leading bytes of uint64_t | c10::Device | Device |
174173
| ? | ? | c10::Stream | Stream |
175174
| ? | ? | c10::complex<double> | complex |
176175
| ? | ? | at::Scalar | Scalar |
177-
| ? | ? | std::string/const char*/ivalue::ConstantString | str |
176+
| std::string/std::string_view | raw bitwise copy of underlying StringHandle into leading bytes of uint64_t | std::string/const char*/ivalue::ConstantString | str |
178177
| ? | ? | at::Storage | Storage |
179178
| ? | ? | at::Generator | Generator |
180-
| ? | ? | c10::List\<T> | Type[] |
179+
| std::vector<T>/torch::headeronly::HeaderOnlyArrayRef<T> | raw bitwise copy into leading bytes of uint64_t of pointer to a new StableIValue pointing to a list of StableIValues recursively representing the underlying elements. | c10::List\<T> | Type[] |
181180
| ? | ? | ivalue::Tuple\<T> | (Type, ...) |
182181
| ? | ? | c10::SymInt | SymInt |
183182
| ? | ? | c10::SymFloat | SymFloat |

0 commit comments

Comments
 (0)