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

fix ci

This commit is contained in:
Dun Liang 2020-07-18 10:38:46 +08:00
parent c28d606f5a
commit 0e5c19eb72
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/ops/tape_op.h
View File

@ -17,6 +17,8 @@ struct Tapes;
struct GradCallback {
typedef jittor::VarHolder VarHolder;
typedef VarHolder* VarHolderPtr;
typedef jittor::Var Var;
typedef jittor::VarPtr VarPtr;
std::function<void(int,Var**,int,VarPtr*)> func;
std::function<void()> deleter;
inline ~GradCallback() { if (deleter) deleter(); }

2
src/pyjt/py_converter.h
View File

@ -602,7 +602,7 @@ DEF_IS(GradCallback, T) from_py_object(PyObject* obj) {
Py_INCREF(obj);
T func(
// callback
[obj](int n_o, Var** douts, int n_i, VarPtr* dins) {
[obj](int n_o, typename T::Var** douts, int n_i, typename T::VarPtr* dins) {
PyObjHolder list(PyTuple_New(n_o));
for (int i=0; i<n_o; i++) {
if (douts[i]) {