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

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