Hi,
I am trying to register an external function which pack data. I want to be able to return a handle to created data structure ( a templatized class) using ret.
Gist:
TVM_REGISTER_GLOBAL(“tvm.contrib.fbgemm.pack_matB”)
.set_body([](TVMArgs args, TVMRetValue *ret) {
auto packB = new PackBMatrix<std::int8_t, std::int32_t> ( blah, blah);
*ret = *packB;
})
This doesn’t currently work. Any pointers on how to achieve this will be helpful