amp list updated

PaddlePaddle · Feb 2, 2022 · e00e38b · e00e38b
1 parent 0e0c5f4
commit e00e38b
Showing 1 changed file with 5 additions and 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
@@ -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)