未验证 提交 6ecfe806 编写于 作者: J JingZhuangzhuang 提交者: GitHub

CUDA sparsity support (#36413) (#36659)

上级 8ebee864
...@@ -238,6 +238,7 @@ void TensorRTEngine::FreezeNetwork() { ...@@ -238,6 +238,7 @@ void TensorRTEngine::FreezeNetwork() {
infer_engine_.reset(infer_builder_->buildEngineWithConfig( infer_engine_.reset(infer_builder_->buildEngineWithConfig(
*network(), *infer_builder_config_)); *network(), *infer_builder_config_));
#else #else
infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kSPARSE_WEIGHTS);
infer_ptr<nvinfer1::IHostMemory> plan(infer_builder_->buildSerializedNetwork( infer_ptr<nvinfer1::IHostMemory> plan(infer_builder_->buildSerializedNetwork(
*network(), *infer_builder_config_)); *network(), *infer_builder_config_));
infer_ptr<nvinfer1::IRuntime> runtime(createInferRuntime(&logger_)); infer_ptr<nvinfer1::IRuntime> runtime(createInferRuntime(&logger_));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册