mri~vdZddlZddlmZddlmZddlmZddlm Z m Z ejGdd Z y) 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) GuardFailGuardFilterEntryczeZdZUdZeeegdfed<dZeee gdfed<dZ eee e ge e fed<y)HooksNguard_export_fn guard_fail_fnguard_filter_fn)__name__ __module__ __qualname__r rrr__annotations__r rr listrbool>/usr/local/lib/python3.12/dist-packages/torch/_dynamo/hooks.pyr r s^=AOXh {D'89:A;?M8Hi[$%678?PTOXh-=(>'?d'KLMTrr ) __doc__ dataclassescollections.abcrtypingr torch._guardsrtypesrr dataclassr rrrrs= $#. UUUr