tensorRT_Pro icon indicating copy to clipboard operation
tensorRT_Pro copied to clipboard

Is engine thread safe?

Open solaim opened this issue 3 years ago • 1 comments

Hi,

I want to use multi-thread to re-use engine, is this thread safe?

or any recommendations?

solaim avatar Mar 07 '22 08:03 solaim

the class infer under TRT is not thread-safe. But the class infer in yolo is thread-safe. Check the the design in yolo. TRT 下的infer 不是线程安全 但是yolo 下的infer 下是线程安全 所以如果想要考虑线程安全 可以参考yolo 的设计

Guanbin-Huang avatar Mar 09 '22 02:03 Guanbin-Huang