[incubator-mxnet] branch amp-patch created (now 6052036)

[ha...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

haibin pushed a change to branch amp-patch
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.


      at 6052036  [AMP] Move topk from FP16_FP32_FUNCS to FP32_FUNCS

This branch includes the following new commits:

     new 6052036  [AMP] Move topk from FP16_FP32_FUNCS to FP32_FUNCS

The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails.  The revisions
listed as "add" were already present in the repository and have only
been added to this reference.

[incubator-mxnet] 01/01: [AMP] Move topk from FP16_FP32_FUNCS to FP32_FUNCS

[ha...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

haibin pushed a commit to branch amp-patch
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git

commit 6052036e8a4b37d91d8ce9b4a5ef059a021fab92
Author: Haibin Lin <li...@gmail.com>
AuthorDate: Mon Jun 24 14:51:01 2019 -0700

    [AMP] Move topk from FP16_FP32_FUNCS to FP32_FUNCS
---
 python/mxnet/contrib/amp/lists/symbol.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py
index 9c99340..066618b 100644
--- a/python/mxnet/contrib/amp/lists/symbol.py
+++ b/python/mxnet/contrib/amp/lists/symbol.py
@@ -340,7 +340,6 @@ FP16_FP32_FUNCS = [
     'take',
     'tanh',
     'tile',
-    'topk',
     'transpose',
     'trunc',
     'uniform',
@@ -463,6 +462,7 @@ FP32_FUNCS = [
     '_sparse_norm',
     '_sparse_rsqrt',
     'argsort',
+    'topk',
 
     # Neural network
     'SoftmaxOutput',