From 0e5c19eb72df0b4bcdcdbe3f2366fd3826360bf3 Mon Sep 17 00:00:00 2001
From: Dun Liang <randonlang@gmail.com>
Date: Sat, 18 Jul 2020 10:38:46 +0800
Subject: [PATCH] fix ci

---
 src/ops/tape_op.h       | 2 ++
 src/pyjt/py_converter.h | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/ops/tape_op.h b/src/ops/tape_op.h
index 14cf0134..d6354c91 100644
--- a/src/ops/tape_op.h
+++ b/src/ops/tape_op.h
@@ -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(); }
diff --git a/src/pyjt/py_converter.h b/src/pyjt/py_converter.h
index 65ec6320..50c9d304 100644
--- a/src/pyjt/py_converter.h
+++ b/src/pyjt/py_converter.h
@@ -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]) {