lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

tiny fix cu

This commit is contained in:
Dun Liang 2020-08-13 14:04:35 +08:00
parent 535523063f
commit ec7164e533
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
python/jittor/test/test_matmul.py
View File

@ -341,6 +341,7 @@ class TestMatmul(unittest.TestCase):
check([8,1,3,4], [10,4,5]) check([8,1,3,4], [10,4,5])
check([5,10,3,4], [5,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) @jt.flag_scope(use_cuda=1)
def test_matmul_example2_cuda(self): def test_matmul_example2_cuda(self):
self.test_matmul_example2() self.test_matmul_example2()

2
src/utils/log.cc
View File

@ -335,7 +335,7 @@ void test_log_time(std::ostream* out) {
auto finish = std::chrono::high_resolution_clock::now(); auto finish = std::chrono::high_resolution_clock::now();
auto total_ns = std::chrono::duration_cast<std::chrono::nanoseconds>(finish-start).count(); auto total_ns = std::chrono::duration_cast<std::chrono::nanoseconds>(finish-start).count();
LOGi << "total_ns" << total_ns << "each_ns" << total_ns/n; LOGi << "total_ns" << total_ns << "each_ns" << total_ns/n;
CHECKop(total_ns/n,<=,6000); CHECKop(total_ns/n,<=,6500);
}; };
std::list<std::thread> ts; std::list<std::thread> ts;
for (int i=0; i<nthread; i++) ts.emplace_back(log_lot); for (int i=0; i<nthread; i++) ts.emplace_back(log_lot);