diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 1eb57d34490edd3557dbe09a541b0a173ac7bbdf..1bb34990f2000e7294f8ecbda7da2fb6f19b3336 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -84,6 +84,9 @@ def parse_args(): parser.add_argument("--enable_mkldnn", type=str2bool, default=False) parser.add_argument("--use_zero_copy_run", type=str2bool, default=False) + + parser.add_argument("--use_pdserving", type=str2bool, default=False) + return parser.parse_args()