This is an automated email from the ASF dual-hosted git repository. haibin pushed a commit to branch v1.6.x in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 2c61787c78a51ea87d46c4820746a33b70fca64c Author: Haibin Lin <[email protected]> AuthorDate: Thu Jan 23 15:36:47 2020 -0800 Update symbol.py (#17408) --- python/mxnet/contrib/amp/lists/symbol.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py index 2146853..d501a7d 100644 --- a/python/mxnet/contrib/amp/lists/symbol.py +++ b/python/mxnet/contrib/amp/lists/symbol.py @@ -591,6 +591,10 @@ WIDEST_TYPE_CASTS = [ '_contrib_dgl_graph_compact', '_contrib_dgl_subgraph', '_contrib_edge_id', + '_contrib_interleaved_matmul_encdec_qk', + '_contrib_interleaved_matmul_encdec_valatt', + '_contrib_interleaved_matmul_selfatt_qk', + '_contrib_interleaved_matmul_selfatt_valatt', 'where', '_sparse_where', '_sparse_broadcast_add',
