Add scatter into black list for dygraph.

limin2021 · limin2021 · commit ec0922829c25 · 2021-12-27T14:31:25.000Z
diff --git a/python/paddle/fluid/dygraph/amp/auto_cast.py b/python/paddle/fluid/dygraph/amp/auto_cast.py
@@ -71,7 +71,9 @@
 }
 
 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

Original file line number	Diff line number	Diff line change
`@@ -71,7 +71,9 @@`
`71`	`71`	`}`
`72`	`72`
`73`	`73`	`PURE_FP16_WHITE_LIST = {' '}`
`74`		`-PURE_FP16_BLACK_LIST = {'lookup_table', 'lookup_table_v2'}`
	`74`	`+PURE_FP16_BLACK_LIST = {`
	`75`	`+ 'lookup_table', 'lookup_table_v2', 'scatter', 'scatter_grad'`
	`76`	`+}`
`75`	`77`
`76`	`78`
`77`	`79`	`#NOTE(zhiqiu): similar as paddle.fluid.contrib.mixed_precision.fp16_lists.AutoMixedPrecisionLists._update_list`