From e00e38ba694a4d06605f825cf5728bb563b044b0 Mon Sep 17 00:00:00 2001 From: Artur Lesniak Date: Wed, 2 Feb 2022 12:13:08 +0100 Subject: [PATCH] amp list updated --- .../paddle/fluid/contrib/mixed_precision/bf16/amp_lists.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/contrib/mixed_precision/bf16/amp_lists.py b/python/paddle/fluid/contrib/mixed_precision/bf16/amp_lists.py index f39be90101959..3f799809af977 100644 --- a/python/paddle/fluid/contrib/mixed_precision/bf16/amp_lists.py +++ b/python/paddle/fluid/contrib/mixed_precision/bf16/amp_lists.py @@ -91,7 +91,11 @@ def _update_list(self): } # depends on the prev_op type -gray_list = gray_list_fp16 +gray_list = { + 'elementwise_add', 'elementwise_sub', 'elementwise_mul', 'elementwise_div', + 'relu', 'layer_norm', 'slice', 'concat', 'uniform_random', 'reshape2', + 'transpose2', 'pool2d', 'sigmoid', 'cast', 'scale', 'fill_constant', 'split' +} _, _, _sys_unsupported_bf16_list = core.op_supported_infos( 'CPU', core.VarDesc.VarType.BF16)