Added nightly test for col2im

apache · Feb 20, 2020 · d24f284 · d24f284
1 parent 3f9fc15
commit d24f284
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -455,6 +455,19 @@ def npy_instance_norm(data, gamma, beta, axis, eps=1E-5):
         assert_almost_equal(out, out_nd.asnumpy(), forward_check_eps,
                             forward_check_eps)
 
+    def check_col2im():
+        data = nd.random_normal(shape=(1, 2**30, 4))
+        output_size = (2, 2, 1)
+        kernel = (1, 1, 1)
+
+        res = nd.col2im(data=data, output_size=output_size, kernel=kernel)
+
+        assert res.shape[0] == 1
+        assert res.shape[1] == 1073741824
+        assert res.shape[2] == 2
+        assert res.shape[3] == 2
+        assert res.shape[4] == 1
+
     check_gluon_embedding()
     check_fully_connected()
     check_dense()
@@ -474,6 +487,7 @@ def npy_instance_norm(data, gamma, beta, axis, eps=1E-5):
     check_linear_and_logistic_regression()
     check_l2_normalization()
     check_instance_norm()
+    check_col2im()
 
 
 def test_tensor():