Skip to content

Commit

Permalink
Add scatter into black list for dygraph.
Browse files Browse the repository at this point in the history
  • Loading branch information
@limin2021
limin2021 committed Dec 27, 2021
1 parent 0311ebb commit ec09228
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion python/paddle/fluid/dygraph/amp/auto_cast.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

1 comment on commit ec09228

@paddle-bot-old
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

✅Congratulation! Your pull request passed all required CI. You could ask reviewer(s) to approve and merge. 🎉

Please sign in to comment.