Nigel
/
jittor
forked from jittor/jittor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

bug fix

This commit is contained in:
Dun Liang 2020-05-17 23:12:30 +08:00
parent 725c9d3544
commit f5a12baf1e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -16,7 +16,7 @@ from setuptools import setup, find_packages
import os
path = os.path.dirname(__file__)
with open(os.path.join(path, "README.src.md"), "r", encoding='utf8') as fh:
with open(os.path.join(path, "README.md"), "r", encoding='utf8') as fh:
long_description = fh.read()
setuptools.setup(

2
src/mem/mem_info.cc
View File

@ -99,7 +99,7 @@ MemInfo::MemInfo() {
total_cpu_ram = info.totalram;
total_cuda_ram = 0;
#ifdef HAS_CUDA
cudaDeviceProp prop = {0};
cudaDeviceProp prop;
cudaGetDeviceProperties(&prop, 0);
total_cuda_ram = prop.totalGlobalMem;
#endif