From 715f87546e8defed7dd85ac116b7f542a537c818 Mon Sep 17 00:00:00 2001 From: kobayu858 Date: Wed, 10 Jul 2024 14:29:44 +0900 Subject: [PATCH] fix:shadowVariable Signed-off-by: kobayu858 --- common/tensorrt_common/src/tensorrt_common.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/common/tensorrt_common/src/tensorrt_common.cpp b/common/tensorrt_common/src/tensorrt_common.cpp index 48a80ba36727e..a22372f74a286 100644 --- a/common/tensorrt_common/src/tensorrt_common.cpp +++ b/common/tensorrt_common/src/tensorrt_common.cpp @@ -432,12 +432,12 @@ bool TrtCommon::buildEngineFromOnnx( auto inner_layer_type = inner_layer->getType(); std::string inner_name = inner_layer->getName(); if (inner_layer_type == nvinfer1::LayerType::kCONVOLUTION) { - layer->setPrecision(nvinfer1::DataType::kHALF); + inner_layer->setPrecision(nvinfer1::DataType::kHALF); std::cout << "Set kHALF in " << inner_name << std::endl; break; } if (inner_layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY) { - layer->setPrecision(nvinfer1::DataType::kHALF); + inner_layer->setPrecision(nvinfer1::DataType::kHALF); std::cout << "Set kHALF in " << inner_name << std::endl; break; }