How to debug packedfunc

xqdan · October 10, 2018, 12:27pm

Hi,

see code below, ret seems to be a invalid value, how to step into packedfunc f, if f is a python function, how can know which function f is.

github.com

dmlc/tvm/blob/master/nnvm/src/compiler/packed_func_ext.cc#L98


// custom version of TVM compute
TVM_REGISTER_GLOBAL("nnvm._register_compute")
.set_body([](TVMArgs args, TVMRetValue *rv) {
  // Intentionally copy and not de-allocate it, to avoid free pyobject during shutdown
  PackedFunc* f = new PackedFunc(args[1].operator PackedFunc());
  Op& op = ::dmlc::Registry<nnvm::Op>::Get()->__REGISTER_OR_GET__(args[0]);
  auto fcompute = [f](const NodeAttrs& attrs,
                      const Array<Tensor>& inputs,
                      const Array<Tensor>& out_info)
      -> Array<Tensor> {
    TVMRetValue ret = (*f)(GetAttrDict(attrs), inputs, out_info);
    if ((*ret.ptr<::tvm::NodePtr<tvm::Node> >())->derived_from<tvm::TensorNode>()) {
      return {ret.operator Tensor()};
    } else {
      return ret;
    }
  };
  op.set_attr<FTVMCompute>("FTVMCompute", fcompute, args[2]);
});


TVM_REGISTER_GLOBAL("nnvm._register_schedule")

inputs=…, out_info=…) at nnvm/src/compiler/packed_func_ext.cc:98
98 TVMRetValue ret = (*f)(GetAttrDict(attrs), inputs, out_info);
(gdb) n
97 -> Array {
(gdb)
98 TVMRetValue ret = (*f)(GetAttrDict(attrs), inputs, out_info);