| 1322 | } // namespace |
| 1323 | |
| 1324 | bool loadEngineToBuildEnv(std::string const& engine, bool enableConsistency, BuildEnvironment& env, std::ostream& err) |
| 1325 | { |
| 1326 | std::ifstream engineFile(engine, std::ios::binary); |
| 1327 | SMP_RETVAL_IF_FALSE(engineFile.good(), "", false, err << "Error opening engine file: " << engine); |
| 1328 | engineFile.seekg(0, std::ifstream::end); |
| 1329 | int64_t fsize = engineFile.tellg(); |
| 1330 | engineFile.seekg(0, std::ifstream::beg); |
| 1331 | |
| 1332 | std::vector<uint8_t> engineBlob(fsize); |
| 1333 | engineFile.read(reinterpret_cast<char*>(engineBlob.data()), fsize); |
| 1334 | SMP_RETVAL_IF_FALSE(engineFile.good(), "", false, err << "Error loading engine file: " << engine); |
| 1335 | |
| 1336 | if (enableConsistency) |
| 1337 | { |
| 1338 | checkSafeEngine(engineBlob.data(), fsize); |
| 1339 | } |
| 1340 | |
| 1341 | env.engine.setBlob(engineBlob.data(), engineBlob.size()); |
| 1342 | |
| 1343 | return true; |
| 1344 | } |
| 1345 | |
| 1346 | void dumpRefittable(nvinfer1::ICudaEngine& engine) |
| 1347 | { |
no test coverage detected