k=1e3dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z dZ GddZ d Zd Zd Ze eZGd d eZdS)zETypes and functions to manage information on CPU microarchitectures. N)FEATURE_ALIASES)LazyDictionarycFtjfd}|S)zhDecorator that automatically converts a known target name to a proper Microarchitecture object. ct|tr:|tvr$d}t||t|}||S)Nz "{0}" is not a valid target name) isinstancestrTARGETS ValueErrorformat)selfothermsgfuncs >lib/python3.11/site-packages/archspec/cpu/microarchitecture.py_implz"coerce_target_names.._implsZ eS ! ! #G##8 E!2!2333ENEtD%   ) functoolswraps)rrs` rcoerce_target_namesrs: _T!!!!! Lrc eZdZdZeZddZedZdZ e dZ e dZ e dZ e d Ze d Ze d Zd Zd ZdZedZedZddZdZdS)MicroarchitectureanRepresents a specific CPU micro-architecture. Args: name (str): name of the micro-architecture (e.g. skylake). parents (list): list of parents micro-architectures, if any. Parenthood is considered by cpu features and not chronologically. As such each micro-architecture is compatible with its ancestors. For example "skylake", which has "broadwell" as a parent, supports running binaries optimized for "broadwell". vendor (str): vendor of the micro-architecture features (list of str): supported CPU flags. Note that the semantic of the flags in this field might vary among architectures, if at all present. For instance x86_64 processors will list all the flags supported by a given CPU while Arm processors will list instead only the flags that have been added on top of the base model for the current micro-architecture. compilers (dict): compiler support to generate tuned code for this micro-architecture. This dictionary has as keys names of supported compilers, while values are list of dictionaries with fields: * name: name of the micro-architecture according to the compiler. This is the name passed to the ``-march`` option or similar. Not needed if the name is the same as that passed in as argument above. * versions: versions that support this micro-architecture. generation (int): generation of the micro-architecture, if relevant. rcZ||_||_||_||_||_||_dSN)nameparentsvendorfeatures compilers generation)r rrrrrr s r__init__zMicroarchitecture.__init__Ks0     "$rc|jdd|jD](}fd|jD)S)z,All the ancestors of this microarchitecture.Nc3$K|] }|v|V dSr).0avalues r z.Microarchitecture.ancestors..Xs'GGqGGr)rextend ancestors)r parentr's @rr*zMicroarchitecture.ancestorsSsY QQQl H HF LLGGGGF$4GGG G G G G rc`tt|gd|jDzS)z9Returns a set of the nodes in this microarchitecture DAG.c,g|]}t|Sr$r r%xs r z-Microarchitecture._to_set.._s!A!A!AQ#a&&!A!A!Ar)setr r*r s r_to_setzMicroarchitecture._to_set[s0CII;!A!A$.!A!A!AABBBrct|tstS|j|jkoO|j|jko?|j|jko/|j|jko|j|jko|j|jkSr) rrNotImplementedrrrrrr r rs r__eq__zMicroarchitecture.__eq__as%!233 "! ! I # 4 u|+ 4 / 4  - 4%/1  4 5#33  rc||k Srr$r7s r__ne__zMicroarchitecture.__ne__os5=  rct|tstS||kSrrrr6r4r7s r__lt__zMicroarchitecture.__lt__s6%!233 "! !||~~ //rc||kp||kSrr$r7s r__le__zMicroarchitecture.__le__z 04%<0rct|tstS||kSrr<r7s r__gt__zMicroarchitecture.__gt__~r>rc||kp||kSrr$r7s r__ge__zMicroarchitecture.__ge__rArcN|jj}|dz}||S)Nz\({0.name!r}, {0.parents!r}, {0.vendor!r}, {0.features!r}, {0.compilers!r}, {0.generation!r})) __class____name__r )r cls_namefmts r__repr__zMicroarchitecture.__repr__s4>* A A zz$rc|jSrrr3s r__str__zMicroarchitecture.__str__s yrc t|ts>d}t|tt |||jvrdSt j|d}||S)Nz3only objects of string types are accepted [got {0}]TcdS)NFr$r0s rz0Microarchitecture.__contains__..sur) rr TypeErrorr typerrfeature_aliasesget)r featurer match_aliass r __contains__zMicroarchitecture.__contains__s'3'' <GCCJJs4=='9'9::;; ; dm # #4(7;;G__UU {4   rcd|g|jzD}d}|ddd|Ddz }t|dks J||S)z9Returns the architecture family a given target belongs toc g|] }|j | Sr$)r*r/s rr1z,Microarchitecture.family..sGGGq1;GGGGrz>a target is expected to belong to just one architecture familyz[found , c34K|]}t|VdSrr.r/s rr(z+Microarchitecture.family..s("9"9a3q66"9"9"9"9"9"9r]r)r*joinlenpop)r rootsrs rfamilyzMicroarchitecture.familys~HGTFT^3GGGN <"9"95"9"9"999<<<<5zzQyy{{rcPd|g|jzD}t|dS)zBReturns the best generic architecture that is compatible with selfc(g|]}|jdk |S)generic)rr/s rr1z-Microarchitecture.generic..s$PPP!!(i:O:OA:O:O:Orc*t|jSr)r`r*rQs rrRz+Microarchitecture.generic..s3q{+;+;r)key)r*max)r genericss rrfzMicroarchitecture.generics8QPv6PPP8!;!;<<<.s(>>1A>>>>>>rr rc,g|]}t|Sr$r.r/s rr1z-Microarchitecture.to_dict..s666AQ666r)r rrsortedrr rdict)r return_list_of_items list_of_itemss rto_dictzMicroarchitecture.to_dictsS^^ $ s4;'' ( >> >>>>> ? 4? + 66666 7    ! M"""rc~|jjvrdS|jvrFfd|jDd}d}|||j}t ||j}d}|D]n}|||r`|d}|d|j|dd } | rtj | |jdi|} | cSod }|r)d |D} |d d | dz }n|dz }||j|}t |)aReturns a string containing the optimization flags that needs to be used to produce code optimized for this micro-architecture. If there is no information on the compiler passed as argument the function returns an empty string. If it is known that the compiler version we want to use does not support this architecture the function raises an exception. Args: compiler (str): name of the compiler to be used version (str): version of the compiler to be used c&g|] }|jv |Sr$)r)r%r0compilers rr1z8Microarchitecture.optimization_flags..s%PPPAK8O8O18O8O8Orrzf'{0}' compiler is known to optimize up to the '{1}' microarchitecture in the '{2}' architecture familyc|dd\}}t|\}}t|\}}t|\}}d}||}|r||}||krdS|r||}||krdSdS)Nversions:cZtd|dDS)Nc34K|]}t|VdSr)int)r%ys rr(zfMicroarchitecture.optimization_flags..satisfies_constraint..tuplify..s(<<SVV<<<<<.satisfies_constraint..tuplifys)<.satisfies_constraints',Z'8'>'>s'C'C $K0 <.s===!* ===rz" [supported compiler versions are r\r^z! [no supported compiler versions]r$) rcrr*r UnsupportedMicroarchitecture setdefaultrrVrwarnr_) r rvr best_targetr compiler_inforcompiler_entry flags_fmtwarning_messagerrxs ` roptimization_flagsz$Microarchitecture.optimization_flagss 4;0 0 02 4> ) )PPPPdnPPPQRSKF **X{K4FGGC.s33 3x0    2,  N##NG<< *73 ))&$)<<<#1"4"4Z"F"F"3M/222( (::>::   X   7==}===H N (8K8KNNN NCC 6 6CjjHg66*3///rN)r)F)rH __module__ __qualname____doc__rrUr!propertyr*r4rr8r:r=r@rCrErKrNrYrcrfrrrr$rrrr&sD&O%%%%XCCC      !!!000 111000 111    ! ! !X==X= ####&Q0Q0Q0Q0Q0rrc*t|gdgiS)zReturns a generic micro-architecture with no vendor and no features. Args: name (str): name of the micro-architecture rf)rrrr)rrMs rgeneric_microarchitecturers&  bR2   rctjdt|}|sdS|d}|d}||fS)aDecomposes the version passed as input in version number and suffix and returns them. If the version number or the suffix are not present, an empty string is returned. Args: version (str): version to be decomposed into its components z([\d.]*)(-?)(.*))rtrtr)rematchr group)rrversion_numbersuffixs rrr%sR H(#g,, 7 7E v[[^^N [[^^F 6 !!rcfdi}tjjjd}|D]}||vr|||t j}||t||S)zReturns a dictionary of the known micro-architectures. If the current host platform is unknown adds it too as a generic target. c4||}|d}|D]}|vr ||fd|D}|d}t|d}|di} |dd} t||||| | |<dS) avRecursively fills targets by adding the micro-architecture passed as argument and all its ancestors. Args: name (str): micro-architecture to be added to targets. data (dict): raw data loaded from JSON. targets (dict): dictionary that maps micro-architecture names to ``Microarchitecture`` objects fromc:g|]}|Sr$)rV)r%r+targetss rr1zL_known_microarchitectures..fill_target_from_dict..Qs%BBB67;;v&&BBBrrrrr rN)r2rVr) rdatarvalues parent_namesr+rrrrr fill_target_from_dicts ` rrz8_known_microarchitectures..fill_target_from_dict>sdf~ " 9 9F   ! !&$ 8 8 8 8BBBB\BBB!vj)**JJ{B// ZZ a00 ) '68Y    rmicroarchitectures)archspeccpuschema TARGETS_JSONplatformmachinerr) known_targetsrr host_platformrs @r_known_microarchitecturesr9s      <M <  +,@ AD99 = dD-8888$&&M],Em,T,TUUU rceZdZdZdS)rzcRaised if a compiler version does not support optimization for a given micro-architecture. N)rHrrrr$rrrrosrr)rrrrrrarchspec.cpu.aliasarchspec.cpu.schemaaliasrrrrrrrrr r rr$rrrs1  """"""""""""$q0q0q0q0q0q0q0q0h"""(///f .2 3 3:r