-`iJddlmZmZddlmZmZddlZGddeZdS))ABCabstractmethod)AnyNoReturnNceZdZdZedejjjddfdZ ede dzfdZ de dzfdZ de de fd ZdS) Torch25CustomGraphPassaI This class replaces CustomGraphPass from torch==2.6 when using torch<2.6. It conforms to the 2.6 interface but also supports pickling, as that's what the inductor code cache uses to determine the cache key before 2.6. (in 2.6 and above, uuid() is used.) Subclasses can just "pretend" that uuid is used. graphreturnNcdS)z4 Implementation of the custom pass. N)selfr s ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/compilation/torch25_custom_graph_pass.py__call__zTorch25CustomGraphPass.__call__cdS)z Return an ID to uniquely identify your custom pass implementation. Return None to skip inductor code caching entirely. Nr r s ruuidzTorch25CustomGraphPass.uuidrrc*|S)z Pickling is used instead of uuid() in torch<2.6. Just return uuid() to enable subclasses to only have to implement uuid. )rrs r __getstate__z#Torch25CustomGraphPass.__getstate__ s yy{{rstatec td)NzCannot unpickle CustomGraphPass because pickling is used for cache key uuid. Use torch>=2.6 with native uuid support for custom passes.) ValueError)r rs r __setstate__z#Torch25CustomGraphPass.__setstate__'s 6   r)__name__ __module__ __qualname____doc__rtorchfxr Graphrrrrrrr rrrr s ehn2 t   ^  cDj   ^ cDj # (      rr)abcrrtypingrrrrr rrr$st$####### # # # # # S# # # # # r