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

Merge branch 'develop' into dw_conv

This commit is contained in:
cxjyxx_me 2020-12-07 22:43:29 +08:00
parent 895f6ecee2 6e761f89da
commit 6f5a4cb336
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.src.md
View File

@ -263,16 +263,16 @@ import jittor as jt
jt.flags.use_cuda = 1
```
### Optional Step 5: Run full tests
### Optional Step 5: Test Resnet18 training
### 可选步骤五：进行完整测试
### 可选步骤五：测试训练Resnet18
To check the integrity of Jittor, you can run full tests.
To check the integrity of Jittor, you can run Resnet18 training test.
要检查Jittor的完整性，您可以运行完整的测试。
要检查Jittor的完整性，您可以运行Resnet18训练测试。
```bash
python3.7 -m jittor.test -v
python3.7 -m jittor.test.test_resnet
```
if those tests are failed, please report bugs for us, and feel free to contribute ^_^

8
python/jittor/compile_extern.py
View File

@ -101,8 +101,12 @@ def setup_cub():
global cub_home
from pathlib import Path
cub_path = os.path.join(str(Path.home()), ".cache", "jittor", "cub")
cub_home = install_cub(cub_path)
setup_cuda_lib("cub", link=False, extra_flags=f"-I{cub_home}")
cuda_version = int(get_version(nvcc_path)[1:-1].split('.')[0])
extra_flags = ""
if cuda_version < 11:
cub_home = install_cub(cub_path)
extra_flags = f"-I{cub_home}"
setup_cuda_lib("cub", link=False, extra_flags=extra_flags)
def setup_cuda_extern():
if not has_cuda: return