Rate this Page
★ ★ ★ ★ ★

torch.cuda.graphs.register_graph_instantiate_hook#

torch.cuda.graphs.register_graph_instantiate_hook(fn)[source]#

Register a hook run with each CUDA graph right after it is instantiated. Returns a RemovableHandle; call .remove() to unregister.

Return type:

RemovableHandle

On this page
Show Source
PyTorch Libraries