From f878c0c6ba3e91038e40ce6dc1a4eed3b16931b4 Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Fri, 1 Sep 2017 14:04:52 +0800 Subject: [PATCH] add use_mkldnn flag to v2 --- python/paddle/v2/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/paddle/v2/__init__.py b/python/paddle/v2/__init__.py index 5bea9806119..1c8d8f4b2f6 100644 --- a/python/paddle/v2/__init__.py +++ b/python/paddle/v2/__init__.py @@ -78,6 +78,8 @@ def init(**kwargs): if 'use_gpu' in kwargs: cp.g_command_config_args['use_gpu'] = kwargs['use_gpu'] + if 'use_mkldnn' in kwargs: + cp.g_command_config_args['use_mkldnn'] = kwargs['use_mkldnn'] assert 'parallel_nn' not in kwargs, ("currently 'parallel_nn' is not " "supported in v2 APIs.") -- GitLab