onnxruntime
Support onnx data types (bfloat16, float8) in python I/O binding APIs
#22306

Merged

Support onnx data types (bfloat16, float8) in python I/O binding APIs #22306

tianleiwu merged 6 commits into main from tlwu/io_binding_onnx_type

support onnx type in python api

289553fb

tianleiwu marked this pull request as draft 1 year ago

refactoring

d549f75f

tianleiwu changed the title ~~Support onnx data types in python API~~ Support onnx data types in python I/O binding APIs 1 year ago

tianleiwu marked this pull request as ready for review 1 year ago

tianleiwu requested a review from

yuslepukhin 1 year ago

tianleiwu requested a review from

kunal-vaishnavi 1 year ago

tianleiwu requested a review from

yufenglee 1 year ago

clean up

4a9cfe75

tianleiwu changed the title ~~Support onnx data types in python I/O binding APIs~~ Support onnx data types (bfloat16, float8) in python I/O binding APIs 1 year ago

refactoring

0d3bfc16

update comment

4b859c2f

yuslepukhin dismissed these changes on 2024-10-04

kunal-vaishnavi dismissed these changes on 2024-10-04

fix test (float8 is not supported by torch 2.0 used in training pipel…

482e8be7

tianleiwu dismissed their stale review via 482e8be7 1 year ago

tianleiwu requested a review from

yuslepukhin 1 year ago

tianleiwu requested a review from

kunal-vaishnavi 1 year ago

kunal-vaishnavi approved these changes on 2024-10-04

yuslepukhin requested changes on 2024-10-04

yuslepukhin approved these changes on 2024-10-04

tianleiwu merged b5ef8555 into main 1 year ago

tianleiwu deleted the tlwu/io_binding_onnx_type branch 1 year ago

Reviewers

yuslepukhin

kunal-vaishnavi

yufenglee

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

onnxruntime Support onnx data types (bfloat16, float8) in python I/O binding APIs #22306 Merged

Support onnx data types (bfloat16, float8) in python I/O binding APIs #22306

onnxruntime
Support onnx data types (bfloat16, float8) in python I/O binding APIs
#22306

Merged