diff --git a/test/test_symbolic_jit.py b/test/test_symbolic_jit.py index f983f027cd..fe819e6791 100644 --- a/test/test_symbolic_jit.py +++ b/test/test_symbolic_jit.py @@ -97,8 +97,8 @@ class TestSymbolicJit(unittest.TestCase): jf = TinyJit(f) a = Tensor.rand(10, 3) b = Tensor.rand(10, 3) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = jf(a[:vi], b[:vj]).reshape(i+j, 3).numpy() @@ -111,8 +111,8 @@ class TestSymbolicJit(unittest.TestCase): jf = TinyJit(f) a = Tensor.rand(3, 10) b = Tensor.rand(3, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = jf(a[:, :vi], b[:, :vj]).reshape(3, i+j).numpy() @@ -125,8 +125,8 @@ class TestSymbolicJit(unittest.TestCase): jf = TinyJit(f) a = Tensor.rand(10, 3) b = Tensor.rand(3, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = jf(a[:vi, :], b[:, :vj]).reshape(i, j).numpy() @@ -139,8 +139,8 @@ class TestSymbolicJit(unittest.TestCase): jf = TinyJit(f) a = Tensor.rand(10, 3) b = Tensor.rand(3, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = jf(a[:vj, :], b[:, :vi]).reshape(j, i).numpy() @@ -245,8 +245,8 @@ class TestSymbolicJit(unittest.TestCase): a = Tensor.rand(10, 10) b = Tensor.rand(10, 10) c = Tensor.rand(10, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) # axis = None @@ -297,8 +297,8 @@ class TestSymbolicJit(unittest.TestCase): a = Tensor.rand(10, 10) b = Tensor.rand(10, 10) c = Tensor.rand(10, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) # axis = None diff --git a/test/test_symbolic_ops.py b/test/test_symbolic_ops.py index 9560db974a..2c2cbebdd3 100644 --- a/test/test_symbolic_ops.py +++ b/test/test_symbolic_ops.py @@ -103,8 +103,8 @@ class TestSymbolicOps(unittest.TestCase): def f(a, b): return a.cat(b, dim=0).realize() a = Tensor.rand(10, 3) b = Tensor.rand(10, 3) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = f(a[:vi, :], b[:vj, :]).reshape(i+j, 3).numpy() @@ -115,8 +115,8 @@ class TestSymbolicOps(unittest.TestCase): def f(a, b): return a.cat(b, dim=1).realize() a = Tensor.rand(3, 10) b = Tensor.rand(3, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = f(a[:, :vi], b[:, :vj]).reshape(3, i+j).numpy() @@ -127,8 +127,8 @@ class TestSymbolicOps(unittest.TestCase): def f(a, b): return (a@b+1).realize() a = Tensor.rand(10, 3) b = Tensor.rand(3, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = f(a[:vi, :], b[:, :vj]).reshape(i, j).numpy() @@ -140,8 +140,8 @@ class TestSymbolicOps(unittest.TestCase): def f(a, b): return (a@b+1).realize() a = Tensor.rand(10, 3) b = Tensor.rand(3, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) symbolic = f(a[:vj, :], b[:, :vi]).reshape(j, i).numpy() @@ -225,8 +225,8 @@ class TestSymbolicOps(unittest.TestCase): def test_mean_2d(self): a = Tensor.rand(10, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) for axis in [None, 0, 1]: @@ -245,8 +245,8 @@ class TestSymbolicOps(unittest.TestCase): def test_var_2d(self): a = Tensor.rand(10, 10) - for i in range(1, 5): - for j in range(1, 5): + for i in range(2, 5): + for j in range(2, 5): vi = Variable("i", 1, 10).bind(i) vj = Variable("j", 1, 10).bind(j) for axis in [None, 0, 1]: