torch.cuda.nvtx.range_push

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

Pushes a range onto a stack of nested range span. Returns zero-based depth of the range that is started.

Parameters: msg (str) – ASCII message to associate with range

© 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.range_push.html