Hi, I’m wondering if there is any easy way to generate a high-level trace/compute graph for DGL scripts. I have tried using the torchviz/hiddenlayer libraries, but they are unable to include information from the graph update_all function (both the edge functions like copy, mul, etc and the aggregation function). Any ideas on how to get a good trace from DGL?
Thanks