From 1523ddc84d4ee0ce680967b8ace7fc86f03a790e Mon Sep 17 00:00:00 2001 From: Weilong Wu Date: Mon, 5 Sep 2022 15:50:00 +0800 Subject: [PATCH] [Eager, Performance optimization] use _C_ops.divide under eager mode (#45731) --- python/paddle/tensor/linalg.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/paddle/tensor/linalg.py b/python/paddle/tensor/linalg.py index 86ea2307ed4..7b7adc1eeca 100644 --- a/python/paddle/tensor/linalg.py +++ b/python/paddle/tensor/linalg.py @@ -906,9 +906,13 @@ def cond(x, p=None, name=None): keepdim, 'reduce_all', reduce_all) if porder == 2: + if in_dygraph_mode(): + return _C_ops.divide(max_out, min_out) return _legacy_C_ops.elementwise_div(max_out, min_out, 'aixs', axis, 'use_mkldnn', False) if porder == -2: + if in_dygraph_mode(): + return _C_ops.divide(min_out, max_out) return _legacy_C_ops.elementwise_div(min_out, max_out, 'aixs', axis, 'use_mkldnn', False) -- GitLab