diff --git a/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py b/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py index c466cafe1ff3c393f4835988a39fff50044d0b9a..80d2ccb0d5ca6fcb3a802014a860bfb2ff9b3400 100644 --- a/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py +++ b/python/paddle/fluid/contrib/mixed_precision/fp16_lists.py @@ -18,7 +18,9 @@ from ... import core __all__ = ["CustomOpLists", "AutoMixedPrecisionLists"] # lookup_table fp16 is slower than fp32, though fp16 is supported. -_extra_unsupported_fp16_list = {'lookup_table', 'lookup_table_v2'} +_extra_unsupported_fp16_list = { + 'lookup_table', 'lookup_table_v2', 'scatter', 'scatter_grad' +} class AutoMixedPrecisionLists(object): diff --git a/python/paddle/fluid/dygraph/amp/auto_cast.py b/python/paddle/fluid/dygraph/amp/auto_cast.py index 14a2bce63f3b391d80cf218b9f84190d74b988b2..8f0e0dff2fcfb3e4431e8e1bd8216ebd6920635e 100644 --- a/python/paddle/fluid/dygraph/amp/auto_cast.py +++ b/python/paddle/fluid/dygraph/amp/auto_cast.py @@ -71,7 +71,9 @@ AMP_RELATED_FLAGS_SETTING = { } PURE_FP16_WHITE_LIST = {' '} -PURE_FP16_BLACK_LIST = {'lookup_table', 'lookup_table_v2'} +PURE_FP16_BLACK_LIST = { + 'lookup_table', 'lookup_table_v2', 'scatter', 'scatter_grad' +} #NOTE(zhiqiu): similar as paddle.fluid.contrib.mixed_precision.fp16_lists.AutoMixedPrecisionLists._update_list