onnxruntime
c46515cd
- [TensorRT EP] Remove cudaDeviceSynchronize and use cudaAllocator for scratch buffers (#5714)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
[TensorRT EP] Remove cudaDeviceSynchronize and use cudaAllocator for scratch buffers (#5714) * use cuda allocator, remove cudaDeviceSync call * use unique_ptr for scratch buffers
References
#5714 - [TensorRT EP] Remove cudaDeviceSynchronize and use cudaAllocator for scratch buffers
Author
jywu-msft
Parents
fd9d0c4e
Loading