diff --git a/python/paddle/fluid/dygraph/amp/auto_cast.py b/python/paddle/fluid/dygraph/amp/auto_cast.py index d70c976858b96d49e88f00915f3d51491ddbe1a4..6d9be5de058c60fd4eff087d20846c7c9162d8e8 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