Functions 4
clear_kernel_annotations
functionFull reference ↗- tensorplay.cuda.graph_annotations.clear_kernel_annotations()[source]
Clear all recorded kernel annotations.
get_kernel_annotations
functionFull reference ↗- tensorplay.cuda.graph_annotations.get_kernel_annotations()[source]
Return the recorded kernel annotations mapping.
is_available
functionFull reference ↗mark_kernels
functionFull reference ↗- tensorplay.cuda.graph_annotations.mark_kernels(name, metadata=None)[source]
Tag the GPU work captured within its scope with user metadata.
