diff --git a/python/jittor/test/test_matmul.py b/python/jittor/test/test_matmul.py index 34ba234c..8ce52b9b 100644 --- a/python/jittor/test/test_matmul.py +++ b/python/jittor/test/test_matmul.py @@ -341,6 +341,7 @@ class TestMatmul(unittest.TestCase): check([8,1,3,4], [10,4,5]) check([5,10,3,4], [5,10,4,5]) + @unittest.skipIf(not jt.compiler.has_cuda, "No CUDA found") @jt.flag_scope(use_cuda=1) def test_matmul_example2_cuda(self): self.test_matmul_example2() diff --git a/src/utils/log.cc b/src/utils/log.cc index 3844b598..36db65c0 100644 --- a/src/utils/log.cc +++ b/src/utils/log.cc @@ -335,7 +335,7 @@ void test_log_time(std::ostream* out) { auto finish = std::chrono::high_resolution_clock::now(); auto total_ns = std::chrono::duration_cast(finish-start).count(); LOGi << "total_ns" << total_ns << "each_ns" << total_ns/n; - CHECKop(total_ns/n,<=,6000); + CHECKop(total_ns/n,<=,6500); }; std::list ts; for (int i=0; i