Thanks! This is really cool.
I wonder why we need to call torch::CapturedTraceback::gather twice with the same arguments.
torch::CapturedTraceback::gather