fix: remove set_mkldnn_cache_capacity (#1695)

use set_mkldnn_cache_capacity() to avoid memory leak and the bug is fixed in Paddle so that it is not needed in cv now.
pull/1683/head
Tingquan Gao 2022-02-28 09:53:49 +08:00 committed by GitHub
parent 7abf42b7a1
commit b27acf6ace
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions

View File

@ -50,8 +50,6 @@ class Predictor(object):
else:
config.disable_gpu()
if args.enable_mkldnn:
# cache 10 different shapes for mkldnn to avoid memory leak
config.set_mkldnn_cache_capacity(10)
config.enable_mkldnn()
config.set_cpu_math_library_num_threads(args.cpu_num_threads)