frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace edge_type with Edge and create Variable::gradient_edge() (#5030)

Browse Source
This commit is contained in:
Peter Goldsborough
2018-02-07 10:50:42 -08:00
committed by Zachary DeVito
parent 0390587d12
commit 25e946bf78
20 changed files with 191 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/csrc/autograd/python_cpp_function.cpp
View File

@ -105,10 +105,10 @@ PyObject* THPCppFunction_next_functions(THPCppFunction* self, PyObject* hook)
auto& c_tuple = next_functions[i];
THPObjectPtr tuple(PyTuple_New(2));
if (!tuple) return NULL;
PyObject *py_fn = functionToPyObject(c_tuple.first);
PyObject *py_fn = functionToPyObject(c_tuple.function);
if (!py_fn) return NULL;
PyTuple_SET_ITEM(tuple.get(), 0, py_fn);
PyObject *py_idx = PyLong_FromLong(c_tuple.second);
PyObject *py_idx = PyLong_FromLong(c_tuple.input_nr);
if (!py_idx) return NULL;
PyTuple_SET_ITEM(tuple.get(), 1, py_idx);
PyTuple_SET_ITEM(py_functions.get(), i, tuple.release());