MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / loadEngineToBuildEnv

Function loadEngineToBuildEnv

samples/common/sampleEngines.cpp:1324–1344  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1322} // namespace
1323
1324bool 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
1346void dumpRefittable(nvinfer1::ICudaEngine& engine)
1347{

Callers 3

loadEngineFunction · 0.85
getEngineBuildEnvFunction · 0.85

Calls 5

checkSafeEngineFunction · 0.85
readMethod · 0.80
setBlobMethod · 0.80
dataMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected