CUDAGraph

class torch.cuda.CUDAGraph [source]

Обёртка вокруг графика CUDA.

Предупреждение

Этот API находится в бета-версии и может быть изменён в будущих выпусках.

capture_begin(pool=None) [source]

Начинает захват работы CUDA в текущем потоке.

Обычно вы не должны вызывать capture_begin самостоятельно. Используйте graph или make_graphed_callables(), которые вызывают capture_begin внутри.

Параметры:

pool (необязательно) – Токен (возвращённый graph_pool_handle() или other_Graph_instance.pool()), который указывает, что этот график может использовать память указанного пула. См. Управление памятью графика.

capture_end() [source]

Завершает захват графика CUDA в текущем потоке. После capture_end, replay может быть вызван для этого экземпляра.

Обычно вы не должны вызывать capture_end самостоятельно. Используйте graph или make_graphed_callables(), которые вызывают capture_end внутри.

pool() [source]

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

replay() [source]

Переигрывает работу CUDA, захваченную этим графиком.

reset() [source]

Удаляет график, хранящийся в данном экземпляре.