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

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

上级 668ffd59
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册