| 171 | } |
| 172 | |
| 173 | bool initNvparsers() |
| 174 | { |
| 175 | #if !TRT_STATIC |
| 176 | static LibraryPtr libnvparsersPtr{}; |
| 177 | auto fetchPtrs = [](DynamicLibrary* l) { |
| 178 | // TODO: get equivalent Windows symbol names |
| 179 | pCreateUffParser = l->symbolAddress<nvuffparser::IUffParser*()>(kMANGLED_UFF_PARSER_CREATE_NAME.c_str()); |
| 180 | pCreateCaffeParser |
| 181 | = l->symbolAddress<nvcaffeparser1::ICaffeParser*()>(kMANGLED_CAFFE_PARSER_CREATE_NAME.c_str()); |
| 182 | pShutdownUffLibrary = l->symbolAddress<void()>(kMANGLED_UFF_PARSER_SHUTDOWN_NAME.c_str()); |
| 183 | pShutdownCaffeLibrary = l->symbolAddress<void(void)>(kMANGLED_CAFFE_PARSER_SHUTDOWN_NAME.c_str()); |
| 184 | }; |
| 185 | return initLibrary(libnvparsersPtr, kNVPARSERS_LIBNAME, fetchPtrs); |
| 186 | #else |
| 187 | pCreateUffParser = nvuffparser::createUffParser; |
| 188 | pCreateCaffeParser = nvcaffeparser1::createCaffeParser; |
| 189 | pShutdownUffLibrary = nvuffparser::shutdownProtobufLibrary; |
| 190 | pShutdownCaffeLibrary = nvcaffeparser1::shutdownProtobufLibrary; |
| 191 | return true; |
| 192 | #endif // !TRT_STATIC |
| 193 | } |
| 194 | |
| 195 | } // namespace |
| 196 |
no test coverage detected