torch.compiler.cudagraph_mark_warmup_incomplete

torch.compiler.cudagraph_mark_warmup_incomplete() [исходный код]

Запросить дополнительный прогрев для активной функции CUDA Graph Trees.

Вызовите эту функцию синхронно из автотюнера или другого кода, выполняющегося во время прогрева CUDA Graph Trees, если текущей функции требуется дополнительная итерация прогрева. При следующем вызове функция снова выполнится в обычном режиме, а не будет записана. Эта функция ничего не делает вне прогрева CUDA Graph Trees, в том числе во время записи и воспроизведения графа или когда CUDA Graph Trees отключены.