viable/strict/1780597927: Add torch.autograd.graph.region_activation_memory_budget (#185979)
🔒
https://github.com
«Summary
Adds torch.autograd.graph.region_activation_memory_budget(budget), a context manager that overrides torch._functorch.config.activation_memory_budget for the region of a compiled forward traced under it. This lets...»
Automatische Weiterleitung...
1.5s