frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-04 08:00:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix previous_functions when it contains Variables

Browse Source
This commit is contained in:
Sam Gross
2017-02-16 12:22:05 -08:00
committed by Soumith Chintala
parent 7117a9012e
commit dd844f741b
3 changed files with 27 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
torch/csrc/autograd/python_function.cpp
View File

@ -932,9 +932,10 @@ PyObject *THPFunction_previous_functions(THPFunction *self, void *_unused)
return NULL;
for (int i = 0; i < size; i++) {
THPObjectPtr fn_tuple = PyTuple_New(2);
if (!fn_tuple)
return NULL;
PyTuple_SET_ITEM(fn_tuple.get(), 0, functionToPyObject(prev_fns[i].first));
if (!fn_tuple) return NULL;
PyObject* fn = functionToPyObject(prev_fns[i].first);
if (!fn) return NULL;
PyTuple_SET_ITEM(fn_tuple.get(), 0, fn);
PyTuple_SET_ITEM(fn_tuple.get(), 1, PyInt_FromLong(prev_fns[i].second));
PyTuple_SET_ITEM(result.get(), i, fn_tuple.release());
}