o ki)@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]returncCs8hd}tt||dkr tddt||d|vr6||dkr6td|d|d|dd }t|t sJtd t ||d d }t|t s^td t ||d d }t|t srtdt |d}d|vr|d}t|t stdt |t dd|D}t |||||dS)N>r debug_infor r r rz!Got unexpected top level keys: {},rzop_name mismatch: z != r Tz,Expected 'is_root_operator' to be bool, got r z0Expected 'is_used_for_training' to be bool, got r z1Expected 'include_all_overloads' to be bool, got rz&Expected 'debug_info' to be list, got css|]}t|VqdSN)r).0xrk/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torchgen/selective_build/operator.py fsz8SelectiveBuildOperator.from_yaml_dict..rr r r r )lensetkeys ExceptionformatjoinAssertionErrorget isinstancertypelisttupler)rr allowed_keysr r r rdi_listrrrfrom_yaml_dict4sT            z%SelectiveBuildOperator.from_yaml_dictcCst|dddddS)NTr)r)rrrr*from_legacy_operator_name_without_overloadpszASelectiveBuildOperator.from_legacy_operator_name_without_overloadcCs*|j|j|jd}|jdur|j|d<|S)N)r r r r)r r r r )selfretrrrto_dict|s  zSelectiveBuildOperator.to_dictN)rrrrrr)rrrr)rr)__name__ __module__ __qualname____annotations__ staticmethodr)r*r-rrrrr s   ;  rlhsr rhsrcCs,|dur |dur dStt|pd|pdS)Nr)r&rr3r4rrrmerge_debug_infosr6cCslt|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 r6r r5rrrcombine_operatorss    r7!dict[str, SelectiveBuildOperator]cCsLi}t|t|D]\}}|}||vrt|||}|||<q|Sr)r%itemsr7)r3r4 operatorsropnew_oprrrmerge_operator_dictss  r=rrcCs|jddddS)N.)maxsplitr)split)rrrrstrip_operator_overload_namesrBN)r3r r4r rr )r3rr4rrr)r3r8r4r8rr8)rrrr) __future__r dataclassesrrr6r7r=rBrrrrs   {