[MXNET-1413] Adding Large Tensor support for sort operators

python/mxnet/test_utils.py

@@ -425,6 +425,12 @@ def rand_coord_2d(x_low, x_high, y_low, y_high):
     return x, y
 
 
+def create_2d_tensor(rows, columns):
+    a = np.arange(0, rows).reshape(rows, 1)
+    b = np.broadcast_to(a, shape=(a.shape[0], columns))
+    return mx.nd.array(b, dtype=np.int64)
+
+
 def np_reduce(dat, axis, keepdims, numpy_reduce_func):
     """Compatible reduce for old version of NumPy.
 

src/operator/mxnet_op.h

@@ -630,7 +630,7 @@ struct Kernel<OP, cpu> {
       }
     }
 #else
-    for (size_t i = 0; i < N; ++i) {
+    for (index_t i = 0; i < static_cast<index_t>(N); ++i) {
       OP::Map(i, args...);
     }
 #endif