fix the ut test error :)

revert-13637-optimize-opyreader
nhzlx 6 years ago
parent 4801beb101
commit 0c51170052

@ -40,7 +40,7 @@ TEST(TensorRTSubGraphPass, main) {
int* max_batch_size = new int(3);
int* workspace_size = new int(1 << 20);
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>("workspace_size", workspace_size);
argument.Set<std::string>("precision_mode", precision_mode);

Loading…
Cancel
Save