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

Function setPreviewFeatures

samples/common/sampleEngines.cpp:685–697  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

683}
684
685void setPreviewFeatures(IBuilderConfig& config, BuildOptions const& build)
686{
687 auto const setFlag = [&](PreviewFeature feat) {
688 int32_t featVal = static_cast<int32_t>(feat);
689 if (build.previewFeatures.find(featVal) != build.previewFeatures.end())
690 {
691 config.setPreviewFeature(feat, build.previewFeatures.at(featVal));
692 }
693 };
694 setFlag(PreviewFeature::kFASTER_DYNAMIC_SHAPES_0805);
695 setFlag(PreviewFeature::kDISABLE_EXTERNAL_TACTIC_SOURCES_FOR_CORE_0805);
696 setFlag(PreviewFeature::kPROFILE_SHARING_0806);
697}
698
699} // namespace
700

Callers 1

setupNetworkAndConfigFunction · 0.85

Calls 2

setPreviewFeatureMethod · 0.80
findMethod · 0.45

Tested by

no test coverage detected