add fp16 to mkldnn
This commit is contained in:
parent
1c25ac896b
commit
24040eb4ac
|
@ -268,10 +268,11 @@ def create_predictor(args, mode, logger):
|
|||
# cache 10 different shapes for mkldnn to avoid memory leak
|
||||
config.set_mkldnn_cache_capacity(10)
|
||||
config.enable_mkldnn()
|
||||
|
||||
if args.precision == "fp16":
|
||||
config.enable_mkldnn_bfloat16()
|
||||
# enable memory optim
|
||||
config.enable_memory_optim()
|
||||
#config.disable_glog_info()
|
||||
config.disable_glog_info()
|
||||
|
||||
config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass")
|
||||
if mode == 'table':
|
||||
|
|
Loading…
Reference in New Issue