torch.compiler.keep_portable_guards_unsafe

torch.compiler.keep_portable_guards_unsafe(guard_entries) [source]

Общая функция, которая оставляет только проверки, применимые как в средах Python, так и в средах без Python. К ним относятся: - метаданные тензоров и сведения о динамических формах; - состояние глобального контекста (например, autocast, no_grad и т. д.).

По умолчанию использовать эту функцию небезопасно. Чтобы воспользоваться этим API, укажите аргумент guard_filter_fn при вызове torch.compile

>> opt_mod = torch.compile( >> mod, >> options={“guard_filter_fn”: torch.compiler.keep_global_context_and_tensor_guards_unsafe}, >> )