You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by ha...@apache.org on 2020/01/24 23:12:01 UTC
[incubator-mxnet] 02/02: Update symbol.py (#17408)
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 <li...@gmail.com>
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',