Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

Home Page:https://cg.cs.tsinghua.edu.cn/jittor/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

jittor.test.test_cudnn_op有时ok有时fail

LXY666def opened this issue · comments

Describe the bug

jittor.test.test_cudnn_op有时ok有时fail,不清楚原因,终端命令可以正常查到安装的cudnn,jittor/pytorch都可以正常训练

Full Log

======================================================================
FAIL: test (main.TestCudnnConvOp)

Traceback (most recent call last):
File "/home/cityofstars/anaconda3/envs/jittor/lib/python3.8/site-packages/jittor/test/test_cudnn_op.py", line 72, in test
check([10,40,50,4], [5,4,5,5], stride=1, padding=1, dilation=1)
File "/home/cityofstars/anaconda3/envs/jittor/lib/python3.8/site-packages/jittor/test/test_cudnn_op.py", line 70, in check
assert np.allclose(y.data, cy.data), np.abs(y.data-cy.data).max()
AssertionError: 0.0039253235

======================================================================
FAIL: test_backward (main.TestCudnnConvOp)

Traceback (most recent call last):
File "/home/cityofstars/anaconda3/envs/jittor/lib/python3.8/site-packages/jittor/test/test_cudnn_op.py", line 130, in test_backward
check([10,4,40,50], [5,4,5,5], stride=1, padding=1, dilation=1)
File "/home/cityofstars/anaconda3/envs/jittor/lib/python3.8/site-packages/jittor/test/test_cudnn_op.py", line 125, in check
assert np.allclose(y.data, cy.data)
AssertionError


Ran 5 tests in 1.407s

FAILED (failures=2)

Minimal Reproduce

Reproduce this error with a file or several lines of code.
If it is not possible, leave it blank.

Expected behavior

test应该一直是ok才对