pytorch / 2 / generated / torch.cuda.nvtx.mark.html

torch.cuda.nvtx.mark

torch.cuda.nvtx.mark(msg) [source]

Describe an instantaneous event that occurred at some point.

Parameters

msg (str) – ASCII message to associate with the event.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.cuda.nvtx.mark.html