torch.cuda.graph_annotations.dump_kernel_py_stacks#
- torch.cuda.graph_annotations.dump_kernel_py_stacks(path)[source]#
Save recorded CUDA graph launch stacks as gzip-compressed JSON.
The file maps decimal node-id strings to the stack strings described by
get_kernel_py_stacks(). Call after instantiation and before resetting or destroying the graphs.- Parameters:
path (str) – Output file path.
Warning
This API is in prototype and may change in future releases.