reduce input size to fix oom (apache#4653)

alexwong · Feb 28, 2020 · ec061d4 · ec061d4
1 parent eaa9a23
commit ec061d4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/topi/tests/python/test_topi_conv2d_NCHWc.py b/topi/tests/python/test_topi_conv2d_NCHWc.py
@@ -229,7 +229,7 @@ def test_conv2d_NCHWc():
     verify_conv2d_NCHWc(1,  64,   56,  64,  1, 1, "VALID")
     verify_conv2d_NCHWc(1, 388,   56,  64,  3, 1, "VALID")
     verify_conv2d_NCHWc(1, 512,   19,  64,  1, 1, "SAME")
-    verify_conv2d_NCHWc(1,  64, 2048,  32,  2, 1, "SAME")
+    verify_conv2d_NCHWc(1,  64,   16,  32,  2, 1, "SAME")
     verify_conv2d_NCHWc(1,  64,   56,  64,  3, 1, (1, 2, 2, 1), add_relu=True)
     verify_conv2d_NCHWc(1,  64,   56,  64,  5, 2, (1, 3), add_bias=True)
     verify_conv2d_NCHWc(1,  64,   56,  64,  3, 1, "VALID", add_bias=True, add_relu=True)

diff --git a/topi/tests/python/test_topi_conv2d_int8.py b/topi/tests/python/test_topi_conv2d_int8.py
@@ -197,7 +197,7 @@ def test_conv2d_nchw():
         verify_conv2d_NCHWc_int8(1,  64,   56,  64,  1, 1, "VALID")
         verify_conv2d_NCHWc_int8(1, 388,   56,  64,  3, 1, "VALID")
         verify_conv2d_NCHWc_int8(1, 512,   19,  64,  1, 1, "SAME")
-        verify_conv2d_NCHWc_int8(1,  64, 2048,  32,  2, 1, "SAME")
+        verify_conv2d_NCHWc_int8(1,  64,   16,  32,  2, 1, "SAME")
         verify_conv2d_NCHWc_int8(1,  64,   56,  64,  3, 1, (1, 2, 2, 1), add_relu=True)
         verify_conv2d_NCHWc_int8(1,  64,   56,  64,  5, 2, (1, 3), add_bias=True)
         verify_conv2d_NCHWc_int8(1,  64,   56,  64,  3, 1, "VALID", add_bias=True, add_relu=True)

diff --git a/topi/tests/python/test_topi_conv2d_nchw.py b/topi/tests/python/test_topi_conv2d_nchw.py
@@ -194,9 +194,9 @@ def test_conv2d_nchw():
     verify_conv2d_nchw(1,  64,  384,  64,  3, 1, (1, 2), use_cudnn=True)
     verify_conv2d_nchw(1,  64,   56,  64,  1, 1, "VALID")
     verify_conv2d_nchw(1, 388,   56,  64,  3, 1, "VALID")
-    verify_conv2d_nchw(1,  64, 1280,  48,  3, 1, "VALID", use_cudnn=True)
+    verify_conv2d_nchw(1,  64,   20,  48,  3, 1, "VALID", use_cudnn=True)
     verify_conv2d_nchw(1, 512,   19,  64,  1, 1, "SAME")
-    verify_conv2d_nchw(1,  64, 2048,  32,  2, 1, "SAME")
+    verify_conv2d_nchw(1,  64,   16,  32,  2, 1, "SAME")
     verify_conv2d_nchw(1,  64,    8,  64,  3, 1, "SAME", use_cudnn=True)
     verify_conv2d_nchw(1,  64,   56,  64,  3, 1, (1, 2, 2, 1), add_relu=True)
     verify_conv2d_nchw(1,  64,   56,  64,  5, 2, (1, 3), add_bias=True)