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

fix array args bug

This commit is contained in:
Dun Liang 2020-07-08 13:07:38 +08:00
parent 0f0b08e30d
commit f250fdd53b
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

@ -21,7 +21,7 @@ with open(os.path.join(path, "README.md"), "r", encoding='utf8') as fh:
setuptools.setup(
name='jittor',
version='1.1.5.2',
version='1.1.5.3',
# scripts=[],
author="Jittor Group",
author_email="ran.donglang@gmail.com",

2
src/pyjt/py_converter.h
View File

@ -303,10 +303,10 @@ DEF_IS(ArrayArgs, T) from_py_object(PyObject* obj) {
auto arr = (PyArray_Proxy*)holder.obj;
int64 size = PyArray_Size(arr);
T args;
args.ptr = arr->data;
args.shape = vector<int64>(arr->dimensions, arr->dimensions+arr->nd);
args.dtype = get_type_str(arr);
args.buffer.reset(new char[size]);
args.ptr = (void*)args.buffer.get();
memcpy((void*)args.buffer.get(), (void*)arr->data, size);
return args;
}