fix ci bug

python/jittor/init_cupy.py

@@ -9,7 +9,7 @@
 
 has_cupy = 0
 try:
-    import cupy
+    import cupy as cp
     has_cupy = 1
 except:
     pass

python/jittor/test/__main__.py

@@ -34,4 +34,6 @@ if __name__ == "__main__":
         suite.addTest(unittest.defaultTestLoader.loadTestsFromName(
             "jittor.test."+test_name))
 
-    unittest.TextTestRunner(verbosity=3).run(suite)
+    result = unittest.TextTestRunner(verbosity=3).run(suite)
+    if len(result.errors) or len(result.failures):
+        exit(1)

src/opt/pass/atomic_tuner_pass.cc

@@ -223,7 +223,7 @@ void AtomicTunerPass::run() {
     for (uint i=0; i<ir->before.size(); i++) {
         auto& func_call = ir->before[i];
         // TODO: remove this if
-        if (func_call->get_attr("dtype") != "__global__ void") continue;
+        if (func_call->get_attr("dtype").find("__global__ void") == string::npos) continue;
         tune_atomic(this, func_call.get(), is_cuda, 4, sorder, sfunc);
     }