-`iddlZddlmZddlZddlmZeeZdejj de e de fdZ ejde e dzdedfd ZdS) N) Generator) init_loggernode splitting_opsreturnc |jdkrdS|j}t|tjjr |j|vSt|tjjr(|}|d|j }||vp||vSdS)z Check if a node should be split for dynamo graph partition. It operates on dynamo graph, so the node.target can be anything. We need to check and split only on OpOverload and OpOverloadPacket. call_functionF.) optarget isinstancetorch_opsOpOverloadPacket_qualified_op_name OpOverloadname _overloadname)rrr packet_nameop_overload_names t/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/compilation/partition_rules.py should_splitrs w/!!u [F&%*566:(M99&%*/00Qkkmm *BBF,@BB=0PK=4PP 5)NNNc#K|s tddVdSttjjj}|tjj_tdt| dV|tjj_tddS#|tjj_tdwxYw)agContext manager to temporarily register Inductor partition rules. Registers custom partition rules for specified operators, forcing the Inductor scheduler to partition the graph at these operators. The rules are automatically restored to their previous state on exit. Args: splitting_ops: List of operator names to partition on. z6No partition ops provided; skipping rule registration.Nz4Registered inductor partition rules for %d operatorsz(Restored previous partition rules state.)loggerdebuglistr _inductorconfigcustom_should_partition_opslen)rsaved_splitting_opss rinductor_partition_rule_contextr#)s  MNNN &* :&&:GEO6 LL>M@R@RA >Q: ?@@@@@>Q: ?@@@@s B==2C/) contextlibcollections.abcrr vllm.loggerr__name__rfxNoderstrboolrcontextmanagerr#rrr.s%%%%%% ###### X  ux}T#Y46 !A9t#!A !A!A!A!A!A!Ar