onnxruntime
[TensorRT EP] Call cudaSetDevice at compute function for handling multithreading scenario
#24010
Merged
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
5
Changes
View On
GitHub
Commits
update
chilo-ms
committed
1 year ago
update
chilo-ms
committed
1 year ago
update
chilo-ms
committed
1 year ago
move cudaSetDevice to OnRunStart()
chilo-ms
committed
1 year ago
move cudaSetDevice to compute_func
chilo-ms
committed
1 year ago
Loading