torch.cuda.graph_annotations.clear_kernel_annotations

torch.cuda.graph_annotations.clear_kernel_annotations() → None [исходный код]

Удаляет все зарегистрированные аннотации ядер.

Реестр аннотаций является глобальным для процесса и накапливает данные при захвате графов. Длительно работающим программам, захватывающим множество графов, следует очищать его после использования зарегистрированных аннотаций (например, после сохранения их вместе с трассировкой профилировщика).

Очистка удаляет все ранее зарегистрированные данные и отменяет регистрацию для всех областей, открытых до очистки: хуки обратного прохода, которые mark_kernels() добавлены к существующим узлам autograd, остаются в графе, но становятся неактивными. Области, открытые после очистки, регистрируют данные в обычном режиме.

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

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