Add einsum to the default white_list. (#53586)

python/paddle/amp/amp_lists.py

@@ -16,6 +16,7 @@
 # safe and performance-critical. These ops are always converted to fp16.
 FP16_WHITE_LIST = {
     'conv2d',
+    'einsum',
     'matmul',
     'matmul_v2',
     'max_pool2d_with_index',
@@ -87,7 +88,7 @@ FP16_EXTRA_BLACK_LIST = {
     'depthwise_conv2d',
 }
 
-BF16_WHITE_LIST = {'conv2d', 'matmul_v2'}
+BF16_WHITE_LIST = {'conv2d', 'einsum', 'matmul_v2'}
 BF16_BLACK_LIST = set()
 
 

python/paddle/static/amp/fp16_lists.py

@@ -135,6 +135,7 @@ _only_supported_fp16_list = {'resnet_unit', 'fused_bn_add_activation'}
 
 white_list = {
     'conv2d',
+    'einsum',
     'matmul',
     'matmul_v2',
     'mul',