Commit adaabc11 authored by Jie's avatar Jie
Browse files

[DEBUG] 

  multiple GPU crash with [cuda_illigal_memory_address]
    added cudaSetDevice before ICudaEngine::createExecutionContext()

  To make sure TRT engine gets allocated on the same GPU (to access IO memory)
parent b52390f0
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment