torch.cuda.nvtx.range_start

torch.cuda.nvtx.range_start(msg) [исходный код]

Отмечает начало диапазона строковым сообщением. Возвращает уникальный дескриптор этого диапазона, который передаётся соответствующему вызову rangeEnd().

Ключевое отличие от range_push/range_pop заключается в том, что версия range_start/range_end поддерживает диапазоны, начинающиеся и заканчивающиеся в разных потоках (начало в одном потоке, конец — в другом).

Возвращает: дескриптор диапазона (uint64_t), который можно передать в range_end().

Параметры:

msg (str) – сообщение в формате ASCII, связанное с диапазоном.

Тип возвращаемого значения:

int