diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py index 2146853a6866..d501a7d6c5b5 100644 --- a/python/mxnet/contrib/amp/lists/symbol.py +++ b/python/mxnet/contrib/amp/lists/symbol.py @@ -591,6 +591,10 @@ '_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',