fix numpy code op test

Author: cjld

python/jittor/__init__.py

@@ -7,7 +7,7 @@
 # This file is subject to the terms and conditions defined in
 # file 'LICENSE.txt', which is part of this source code package.
 # ***************************************************************
-__version__ = '1.1.7.4'
+__version__ = '1.1.7.5'
 from . import lock
 with lock.lock_scope():
     from . import compiler

python/jittor/test/test_numpy_code_op.py

@@ -58,9 +58,9 @@ def check():
             one=numpy.ones(a.shape)
             assert numpy.allclose(da.data,one*2.0)
 
-        jt.flags.use_cuda = 0
-        check()
-        jt.flags.use_cuda = 1
+        if jt.has_cuda:
+            with jt.flag_scope(use_cuda=1):
+                check()
         check()
 
     def test(self):
@@ -92,9 +92,9 @@ def check():
             one=numpy.ones(a.shape)
             assert numpy.allclose(da.data,one*2.0)
 
-        jt.flags.use_cuda = 0
-        check()
-        jt.flags.use_cuda = 1
+        if jt.has_cuda:
+            with jt.flag_scope(use_cuda=1):
+                check()
         check()
 
     def test_multi_input(self):
@@ -139,9 +139,9 @@ def check():
             assert numpy.allclose(dda.data,one)
             assert numpy.allclose(ddb.data,mone)
 
-        jt.flags.use_cuda = 0
-        check()
-        jt.flags.use_cuda = 1
+        if jt.has_cuda:
+            with jt.flag_scope(use_cuda=1):
+                check()
         check()
 
     @unittest.skipIf(True, "Memory leak testing is not in progress, Skip")

python/jittor/test/test_parallel_pass.py

@@ -36,7 +36,7 @@ def check(self, use_int32):
         b = jt.random((n, n))
         a.data, b.data
         with jt.profile_scope(compile_options = {
-            "compile_shapes":1, "parallel":1
+            "compile_shapes":1, "parallel":2, "try_use_32bit_index":use_int32
         }, try_use_32bit_index = use_int32) as rep:
             c = a + b
             nc = c.data