[runtime] How can I access defined private variables in CUDAModuleNode from module.cc or graph_runtime.cc?？

Questions
#1

Hi~

I try to access the defined private variable in cuda_module class after op execution in graph_runtime. How can I do it? Do I need to write packedfunc in cuda_module.cc?

#2

@junrushao @vinx13 @masahi @Huyuwei

#3

What variable do you want to extract from cuda_module.cc? For cuda kernel source, you can use runtime.imported_modules[0].get_source()", where runtime is tvm.runtime.Module.

#4

Hi, thanks for your response. Actually I would like to extract(return) grid/block info as well as void**args to graph_runtime.cc, any ideas? I am trying to write packedfunc in cuda_module.cc

#5

Hi, thanks for your response. Actually I would like to extract(return) grid/block info as well as void**args to graph_runtime.cc, any ideas? I am trying to write packedfunc in cuda_module.cc

#6

That would be difficult, since graph runtime doesn’t know anything about CUDAWrappedFunc or CUDAModuleNode.

#7

Is it possible to set some global variables to be accessed in graphruntime