torch.cuda.nvtx.range_end

torch.cuda.nvtx.range_end(range_id) [source]

Отметить окончание диапазона с заданным range_id.

Параметры:

range_id (int) – уникальный идентификатор начального диапазона.