提交 0c511700 编写于 作者: N nhzlx

fix the ut test error :)

上级 4801beb1
...@@ -40,7 +40,7 @@ TEST(TensorRTSubGraphPass, main) { ...@@ -40,7 +40,7 @@ TEST(TensorRTSubGraphPass, main) {
int* max_batch_size = new int(3); int* max_batch_size = new int(3);
int* workspace_size = new int(1 << 20); int* workspace_size = new int(1 << 20);
std::string* precision_mode = new std::string("FP32"); std::string* precision_mode = new std::string("FP32");
argument.Set<int>("minimun_subgraph_size", minimum_subgraph_size); argument.Set<int>("minimum_subgraph_size", minimum_subgraph_size);
argument.Set<int>("max_batch_size", max_batch_size); argument.Set<int>("max_batch_size", max_batch_size);
argument.Set<int>("workspace_size", workspace_size); argument.Set<int>("workspace_size", workspace_size);
argument.Set<std::string>("precision_mode", precision_mode); argument.Set<std::string>("precision_mode", precision_mode);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册