From e821688b7cc310544b00398f2e3ba7d9531eeedc Mon Sep 17 00:00:00 2001
From: zhangbo9674 <82555433+zhangbo9674@users.noreply.github.com>
Date: Tue, 20 Sep 2022 15:10:24 +0800
Subject: [PATCH] use empty set for amp black/white list (#46266)

---
 python/paddle/fluid/dygraph/amp/auto_cast.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/paddle/fluid/dygraph/amp/auto_cast.py b/python/paddle/fluid/dygraph/amp/auto_cast.py
index d70c976858b..6d9be5de058 100644
--- a/python/paddle/fluid/dygraph/amp/auto_cast.py
+++ b/python/paddle/fluid/dygraph/amp/auto_cast.py
@@ -76,7 +76,7 @@ AMP_RELATED_FLAGS_SETTING = {
     'FLAGS_cudnn_batchnorm_spatial_persistent': 1,
 }
 
-PURE_FP16_WHITE_LIST = {''}
+PURE_FP16_WHITE_LIST = set()
 PURE_FP16_BLACK_LIST = {
     'lookup_table',
     'lookup_table_v2',
@@ -91,10 +91,10 @@ PURE_FP16_BLACK_LIST = {
 }
 
 BF16_WHITE_LIST = {'conv2d', 'matmul_v2'}
-BF16_BLACK_LIST = {''}
+BF16_BLACK_LIST = set()
 
-PURE_BF16_WHITE_LIST = {''}
-PURE_BF16_BLACK_LIST = {''}
+PURE_BF16_WHITE_LIST = set()
+PURE_BF16_BLACK_LIST = set()
 
 _g_amp_state_ = None
 
-- 
GitLab