Skip to content

GitLab

"example/01_gemm/gemm_xdl_fp16.cpp" did not exist on "41cdd3801a873def1c1220da9860f5202f36edbd"
Commit d7cbe526 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

adjust the min_subgraph_size to 15

parent 71ed68c9
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/infer/utility.py
View file @ d7cbe526
...@@ -35,7 +35,7 @@ def init_args(): ...@@ -35,7 +35,7 @@ def init_args():
parser.add_argument("--use_gpu", type=str2bool, default=True) parser.add_argument("--use_gpu", type=str2bool, default=True)
parser.add_argument("--ir_optim", type=str2bool, default=True) parser.add_argument("--ir_optim", type=str2bool, default=True)
parser.add_argument("--use_tensorrt", type=str2bool, default=False) parser.add_argument("--use_tensorrt", type=str2bool, default=False)
parser.add_argument("--min_subgraph_size", type=int, default=10) parser.add_argument("--min_subgraph_size", type=int, default=15)
parser.add_argument("--precision", type=str, default="fp32") parser.add_argument("--precision", type=str, default="fp32")
parser.add_argument("--gpu_mem", type=int, default=500) parser.add_argument("--gpu_mem", type=int, default=500)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment