B d @s&dZddlZddlZddlZddlmZdZdZej ddsLej ddrRdZn ej dkrbdZnddl Z dZdZej ddre ejdZ ed e ndererd d Znd d Zed ZedZedZedZ edZedZedZedZedZddZddZGdddeZdS)z This module defines optimization flags and determines hardware features that some other modules and packages may use in addition to some optimized utilities. N) OrderedDictFLEVEL2_CACHE_SIZEZ NO_GETCONFdarwinTz0opt: using LEVEL2_CACHE_SIZE %d from environmentcCsttd|gS)Ngetconf)int subprocess check_output)confvarr V/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/pycbc/opt.pyr,srcCstd|}t|dS)Nzgetconf )commandsgetstatusoutputr)r Zretlistr r r r/sLEVEL1_DCACHE_SIZELEVEL1_DCACHE_ASSOCLEVEL1_DCACHE_LINESIZELEVEL2_CACHE_ASSOCLEVEL2_CACHE_LINESIZELEVEL3_CACHE_SIZELEVEL3_CACHE_ASSOCLEVEL3_CACHE_LINESIZEcCs*|d}|jddd|jddddS)z Adds the options used to specify optimization-specific options. Parameters ---------- parser : object OptionParser instance z4Options for selecting optimization-specific settingsz--cpu-affinityzz A set of CPUs on which to run, specified in a format suitable to pass to taskset.)helpz--cpu-affinity-from-envz The name of an enivornment variable containing a set of CPUs on which to run, specified in a format suitable to pass to taskset.N)add_argument_group add_argument)parserZoptimization_groupr r r insert_optimization_option_group>s rcCsd}|jdk r~|jdk r,tdtdtj|j}|dkr\td|jtd|dkr~td|jtd|dkr|j}|dk rd|t f}t |}|dkrtd ||ftdt d |dS) a:Parses the CLI options, verifies that they are consistent and reasonable, and acts on them if they are Parameters ---------- opt : object Result of parsing the CLI with OptionParser, or any object with the required attributes parser : object OptionParser instance. Nz9Both --cpu_affinity_from_env and --cpu_affinity specifiedr zTCPU affinity requested from environment variable %s but this variable is not definedzNCPU affinity requested from environment variable %s but this variable is emptyztaskset -pc %s %drz/taskset command <%s> failed with return code %dzPinned to CPUs %s ) Zcpu_affinity_from_envZ cpu_affinityloggingerrorsysexitosenvirongetgetpidsysteminfo)optrZrequested_cpuscommandretcoder r r verify_optimization_optionsSs0           r*c@s(eZdZdZddZddZddZdS) LimitedSizeDictz" Fixed sized dict for FIFO cachingcOs,|dd|_tj|f|||dS)N size_limit)popr,r__init___check_size_limit)selfargskwdsr r r r.szLimitedSizeDict.__init__cCst||||dS)N)r __setitem__r/)r0keyvaluer r r r3szLimitedSizeDict.__setitem__cCs.|jdk r*xt||jkr(|jddq WdS)NF)last)r,lenpopitem)r0r r r r/s z!LimitedSizeDict._check_size_limitN)__name__ __module__ __qualname____doc__r.r3r/r r r r r+sr+)r<r!rr collectionsrZ_USE_SUBPROCESSZ HAVE_GETCONFr"r#platformrrrr&rrrrrrrrrrr*r+r r r r s>   /