未验证 提交 80a4a2e5 编写于 作者: W wanghuancoder 提交者: GitHub

speedup eager_legacy_trace_op (#52467)

上级 5b7c8f9e
...@@ -90,6 +90,10 @@ name_mapping = { ...@@ -90,6 +90,10 @@ name_mapping = {
# } # }
} }
core_ops_args_info = _legacy_C_ops.get_core_ops_args_info()
core_ops_args_type_info = _legacy_C_ops.get_core_ops_args_type_info()
core_ops_returns_info = _legacy_C_ops.get_core_ops_returns_info()
class Tracer(core.Tracer): class Tracer(core.Tracer):
""" """
...@@ -120,10 +124,6 @@ class Tracer(core.Tracer): ...@@ -120,10 +124,6 @@ class Tracer(core.Tracer):
): ):
function_ptr = _legacy_C_ops.__dict__[op_type] function_ptr = _legacy_C_ops.__dict__[op_type]
core_ops_args_info = _legacy_C_ops.get_core_ops_args_info()
core_ops_args_type_info = _legacy_C_ops.get_core_ops_args_type_info()
core_ops_returns_info = _legacy_C_ops.get_core_ops_returns_info()
op_args = core_ops_args_info[op_type] op_args = core_ops_args_info[op_type]
op_args_type = core_ops_args_type_info[op_type] op_args_type = core_ops_args_type_info[op_type]
op_returns = core_ops_returns_info[op_type] op_returns = core_ops_returns_info[op_type]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册