diff --git a/python/mxnet/contrib/amp/lists/symbol.py b/python/mxnet/contrib/amp/lists/symbol.py
index <HASH>..<HASH> 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',
[AMP] Move topk from FP<I>_FP<I>_FUNCS to FP<I>_FUNCS (#<I>)