Update trt_executor.cc

Add cast data type casting to fix the bug `cannot convert from const char [] to char*`
OAID · Jan 11, 2022 · ac9fc21 · ac9fc21
1 parent 2cac48a
commit ac9fc21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/device/tensorrt/trt_executor.cc b/source/device/tensorrt/trt_executor.cc
@@ -84,7 +84,7 @@ TensorRTEngine::TensorRTEngine()
 
     this->precision = nvinfer1::DataType::kFLOAT;
 
-    this->option.dev_name = TRT_DEVICE_NAME;
+    this->option.dev_name = (char *)TRT_DEVICE_NAME;
     this->option.precision = TENGINE_DT_FP32;
     this->option.gpu_index = 0;
     this->option.dla_index = -1;