optimum
Improve `use_io_binding` default value for different execution providers
#604
Merged

Improve `use_io_binding` default value for different execution providers #604

JingyaHuang merged 9 commits into main from fix-ort-pretrain-iobinding
JingyaHuang
JingyaHuang Fix iobinding default for TRT
a3b58863
HuggingFaceDocBuilderDev
fxmarty
fxmarty commented on 2022-12-19
fxmarty
JingyaHuang
fxmarty
JingyaHuang Merge branch 'main' into fix-ort-pretrain-iobinding
b2fe9c0f
JingyaHuang Add utils
c78d1d4d
JingyaHuang Improve utils
b41c8532
JingyaHuang Merge branch 'main' into fix-ort-pretrain-iobinding
4d907276
JingyaHuang Update to all ORTModel subclasses
0cb010ee
JingyaHuang JingyaHuang requested a review from fxmarty fxmarty 3 years ago
JingyaHuang JingyaHuang changed the title Fix `use_io_binding` default value for TensorRT EP Improve `use_io_binding` default value for different execution providers 3 years ago
fxmarty
fxmarty commented on 2022-12-21
fxmarty
fxmarty commented on 2022-12-21
fxmarty
fxmarty commented on 2022-12-21
JingyaHuang Update optimum/onnxruntime/modeling_decoder.py
39b0e141
fxmarty
fxmarty commented on 2022-12-21
fxmarty
fxmarty commented on 2022-12-21
JingyaHuang Update optimum/onnxruntime/modeling_seq2seq.py
bced0138
fxmarty
JingyaHuang
JingyaHuang Apply suggestion
9562c07f
fxmarty
fxmarty approved these changes on 2022-12-21
JingyaHuang JingyaHuang merged 34e782e6 into main 3 years ago
JingyaHuang JingyaHuang deleted the fix-ort-pretrain-iobinding branch 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone