o vZhm@s\ddlmZddlmZeddGdddZdd d Zdd dZdddZdddZdS)) annotations) dataclassT)frozenc@s\eZdZUded<ded<ded<ded<ded<edd dZedddZdddZdS)SelectiveBuildOperatorstrnameboolis_root_operatoris_used_for_traininginclude_all_overloadstuple[str, ...] | None _debug_infoop_nameop_infodict[str, object]returncCshd}tt||dkr tddt||d|vr,||dks,J|dd}t|ts9J|dd}t|tsFJ|d d}t|tsSJd}d |vrm|d }t|t sdJt d d |D}t |||||d S)N>r r debug_infor r rz!Got unexpected top level keys: {},rr Tr r rcss|]}t|VqdSN)r).0xrP/var/www/auris/lib/python3.10/site-packages/torchgen/selective_build/operator.py Wsz8SelectiveBuildOperator.from_yaml_dict..rr r r r ) lensetkeys Exceptionformatjoinget isinstancerlisttupler)rrZ allowed_keysr r r rZdi_listrrrfrom_yaml_dict4s6   z%SelectiveBuildOperator.from_yaml_dictcCst|dddddS)NTr)r)rrrr*from_legacy_operator_name_without_overloadaszASelectiveBuildOperator.from_legacy_operator_name_without_overloadcCs*|j|j|jd}|jdur|j|d<|S)N)r r r r)r r r r )selfretrrrto_dictms  zSelectiveBuildOperator.to_dictN)rrrrrr)rrrr)rr)__name__ __module__ __qualname____annotations__ staticmethodr%r&r)rrrrr s   ,  rlhsr rhsrcCs,|dur |dur dStt|pd|pdS)Nr)r$rr/r0rrrmerge_debug_infoysr2cCslt|jt|jkrtdt|jdt|jdt|j|jp#|j|jp(|j|jp-|jt|j|jdS)Nz8Expected both arguments to have the same name, but got 'z' and 'z ' insteadr) rrrrr r r r2r r1rrrcombine_operatorss    r3!dict[str, SelectiveBuildOperator]cCsLi}t|t|D]\}}|}||vrt|||}|||<q|Sr)r#itemsr3)r/r0 operatorsropZnew_oprrrmerge_operator_dictss  r8rrcCs|ddS)N.r)split)rrrrstrip_operator_overload_namesr;N)r/r r0r rr )r/rr0rrr)r/r4r0r4rr4)rrrr) __future__r dataclassesrrr2r3r8r;rrrrs   l