onnxruntime
07aabcc3 - Set cuda device before create cuda stream for IOBinding case (#18583)

Commit
2 years ago
Set cuda device before create cuda stream for IOBinding case (#18583) ### Description <!-- Describe your changes. --> Set cuda device before create cuda stream for IOBinding case ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> This is to fix the issue #18432 , which the inference will fail for IOBinding case when there are multiple cuda devices. The reason is that the cuda device is not set properly before the cuda stream is created
Author
Parents
Loading