diff --git a/convert_tf2npu/ast_impl.py b/convert_tf2npu/ast_impl.py index c586c4585ae8dce4934f2e76a9cbe1ac26b2a4f2..80701afbf894740c14eae0b70bff0f20f7c87040 100644 --- a/convert_tf2npu/ast_impl.py +++ b/convert_tf2npu/ast_impl.py @@ -97,6 +97,11 @@ def ast_if(node): return node def ast_call(node): + if _call_name_match(node.func, "set_experimental_options"): + log_msg(getattr(node, 'lineno', 'None'), 'change set_experimental_options(*) to set_experimental_options(experimental_options)') + node.args = [ast.Name(id='experimental_options', ctx=ast.Load())] + node.keywords = [] + util_global.set_value('need_conver', True) if isinstance(node.func, ast.Name) and node.func.id == 'check_available_gpus': log_msg(getattr(node, 'lineno', 'None'), "change check_available_gpus() to ['/device:CPU:0']") util_global.set_value('need_conver', True) diff --git a/tf_adapter/python/npu_bridge/npu_init.py b/tf_adapter/python/npu_bridge/npu_init.py index cd1d648cfc869451258dc7d1b4f5aedccfc32a13..dbba95df77434bdcbaab524b3db0295d2e842edb 100644 --- a/tf_adapter/python/npu_bridge/npu_init.py +++ b/tf_adapter/python/npu_bridge/npu_init.py @@ -47,6 +47,18 @@ from npu_bridge.estimator.npu.npu_plugin import npu_close import atexit atexit.register(npu_close) +experimental_options = { + "disable_model_pruning": True, + "function_optimization": RewriterConfig.OFF, + "constant_folding": RewriterConfig.OFF, + "shape_optimization": RewriterConfig.OFF, + "arithmetic_optimization": RewriterConfig.OFF, + "loop_optimization": RewriterConfig.OFF, + "dependency_optimization": RewriterConfig.OFF, + "layout_optimizer": RewriterConfig.OFF, + "memory_optimization": RewriterConfig.OFF +} + def npu_hooks_append(hooks_list=[]): if (not isinstance(hooks_list, list)): hooks_list = []