From ec7164e533b9feed0dad9139bc3cc391e30e6847 Mon Sep 17 00:00:00 2001 From: Dun Liang Date: Thu, 13 Aug 2020 14:04:35 +0800 Subject: [PATCH] tiny fix cu --- python/jittor/test/test_matmul.py | 1 + src/utils/log.cc | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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