onnxruntime
[TensorRT EP] Call cudaSetDevice at compute function for handling multithreading scenario
#24010

Merged

[TensorRT EP] Call cudaSetDevice at compute function for handling multithreading scenario #24010

chilo-ms merged 5 commits into main from chi/set_device_fix

update

61dc6a12

update

4e4f26ae

update

1654c17c

chilo-ms requested a review from

jywu-msft 1 year ago

move cudaSetDevice to OnRunStart()

c5c527e3

tianleiwu commented on 2025-03-12

tianleiwu dismissed these changes on 2025-03-12

move cudaSetDevice to compute_func

1c2ff53a

chilo-ms dismissed their stale review via 1c2ff53a 1 year ago

tianleiwu approved these changes on 2025-03-13

chilo-ms merged 6bb6d791 into main 1 year ago

chilo-ms deleted the chi/set_device_fix branch 1 year ago

yf711 added release:1.21.1

snnn removed release:1.21.1

Reviewers

tianleiwu

jywu-msft

Assignees

No one assigned

Labels

None yet

Milestone

No milestone