未验证 提交 e821688b 编写于 作者: Z zhangbo9674 提交者: GitHub

use empty set for amp black/white list (#46266)

上级 668ffd59
...@@ -76,7 +76,7 @@ AMP_RELATED_FLAGS_SETTING = { ...@@ -76,7 +76,7 @@ AMP_RELATED_FLAGS_SETTING = {
'FLAGS_cudnn_batchnorm_spatial_persistent': 1, 'FLAGS_cudnn_batchnorm_spatial_persistent': 1,
} }
PURE_FP16_WHITE_LIST = {''} PURE_FP16_WHITE_LIST = set()
PURE_FP16_BLACK_LIST = { PURE_FP16_BLACK_LIST = {
'lookup_table', 'lookup_table',
'lookup_table_v2', 'lookup_table_v2',
...@@ -91,10 +91,10 @@ PURE_FP16_BLACK_LIST = { ...@@ -91,10 +91,10 @@ PURE_FP16_BLACK_LIST = {
} }
BF16_WHITE_LIST = {'conv2d', 'matmul_v2'} BF16_WHITE_LIST = {'conv2d', 'matmul_v2'}
BF16_BLACK_LIST = {''} BF16_BLACK_LIST = set()
PURE_BF16_WHITE_LIST = {''} PURE_BF16_WHITE_LIST = set()
PURE_BF16_BLACK_LIST = {''} PURE_BF16_BLACK_LIST = set()
_g_amp_state_ = None _g_amp_state_ = None
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册