o i~@sXdZddlZddlmZddlmZddlmZddlm Z m Z ej Gdd d Z dS) aHook system for Dynamo's guard functionality. This module provides a way to register callback functions that are triggered during guard-related operations. The Hooks class manages two types of hook functions: - guard_export_fn: Called when guards need to be exported, taking a GuardsSet as input - guard_fail_fn: Called when a guard check fails, taking a GuardFail object as input These hooks enable customization of guard export and failure handling behaviors. N)Callable)Optional) GuardsSet) GuardFailGuardFilterEntryc@sdeZdZUdZeeegdfed<dZeee gdfed<dZ eee e ge e fed<dS)HooksNguard_export_fn guard_fail_fnguard_filter_fn)__name__ __module__ __qualname__r rrr__annotations__r rr listrboolrrU/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/_dynamo/hooks.pyrs &r) __doc__ dataclassescollections.abcrtypingr torch._guardsrtypesrr dataclassrrrrrs