attempt to solve memory issue

apache · Aug 8, 2019 · a151181 · a151181
1 parent 7f1c2b5
commit a151181
Showing 1 changed file with 71 additions and 59 deletions.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -353,97 +353,109 @@ def test_topk():
 
 
 def test_add():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__add__(b)
-    np_res = a.asnumpy().__add__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__add__(a)    
+    assert c[0][-1] == 2
+    assert c.shape == a.shape
 
 
 def test_sub():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__sub__(b)
-    np_res = a.asnumpy().__sub__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__sub__(a)    
+    assert c[0][-1] == -2
+    assert c.shape == a.shape
 
 
 def test_rsub():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__rsub__(b)
-    np_res = a.asnumpy().__rsub__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__rsub__(a)
+    assert c[0][-1] == 2
+    assert c.shape == a.shape
 
 
 def test_neg():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__neg__()
-    np_res = a.asnumpy().__neg__()
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = nd.ones(shape=(LARGE_X, SMALL_Y))    
+    c = a
+    c = c.__neg__()
+    assert c[0][-1] == -1
+    assert c.shape == a.shape
 
 
 def test_mul():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__mul__(b)
-    np_res = a.asnumpy().__mul__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__mul__(a)    
+    assert c[0][-1] == 6
+    assert c.shape == a.shape
 
 
 def test_div():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__div__(b)
-    np_res = a.asnumpy().__truediv__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__div__(a)    
+    assert c[0][-1] == 3/2
+    assert c.shape == a.shape
 
 
 def test_rdiv():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__rdiv__(b)
-    np_res = a.asnumpy().__rtruediv__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__rdiv__(a)    
+    assert c[0][-1] == 2/3
+    assert c.shape == a.shape
 
 
 def test_mod():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__mod__(b)
-    np_res = a.asnumpy().__mod__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__mod__(a)    
+    assert c[0][-1] == 1
+    assert c.shape == a.shape
 
 
 def test_rmod():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__rmod__(b)
-    np_res = a.asnumpy().__rmod__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__rmod__(a)    
+    assert c[0][-1] == 2
+    assert c.shape == a.shape
 
 
 def test_imod():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__imod__(b)
-    np_res = a.asnumpy().__imod__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__imod__(a)    
+    assert c[0][-1] == 1
+    assert c.shape == a.shape
 
 
 def test_pow():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__pow__(b)
-    np_res = a.asnumpy().__pow__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__pow__(a)    
+    assert c[0][-1] == 9
+    assert c.shape == a.shape
 
 def test_rpow():
-    a = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y, dtype=np.float64)
-    mx_res = a.__rpow__(b)
-    np_res = a.asnumpy().__rpow__(b.asnumpy())
-    assert mx_res.asnumpy()[-1][0]==np_res[-1][0]
+    a = 2*nd.ones(shape=(LARGE_X, SMALL_Y))
+    b = 3*nd.ones(shape=(LARGE_X, SMALL_Y))
+    c = b
+    c = c.__rpow__(a)    
+    assert c[0][-1] == 8
+    assert c.shape == a.shape
 
 if __name__ == '__main__':
     import nose