!gYddlZejeZddlZddlZddlmZmZm Z ddl m Z ddl m Z e r ddlmZmZmZmZdZdZd Zd Zd Zd Zd ZdZdZGddeZy)N)_set_num_threads_get_num_threads MAX_THREADS)use_vml)version)_get_vml_version_set_vml_accuracy_mode_set_vml_num_threads_get_vml_num_threadsc$tr tSy)z* Get the VML/MKL library version. N)rr -lib/python3.12/site-packages/numexpr/utils.pyget_vml_versionrs!!rctr_ddddd}dddd }|t|jvr td t |j |d}|j |Sy ) a> Set the accuracy mode for VML operations. The `mode` parameter can take the values: - 'high': high accuracy mode (HA), <1 least significant bit - 'low': low accuracy mode (LA), typically 1-2 least significant bits - 'fast': enhanced performance mode (EP) - None: mode settings are ignored This call is equivalent to the `vmlSetMode()` in the VML library. See: http://www.intel.com/software/products/mkl/docs/webhelp/vml/vml_DataTypesAccuracyModes.html for more info on the accuracy modes. Returns old accuracy settings. rr)Nlowhighfastrrr)rrrz9mode argument must be one of: None, 'high', 'low', 'fast'N)rlistkeys ValueErrorr get)modeacc_dictacc_reverse_dictretvals rset_vml_accuracy_moder %sp&Aq!<$F; tHMMO, ,KM M' T1(=>##F++rc&tr t|y)a[ Suggests a maximum number of threads to be used in VML operations. This function is equivalent to the call `mkl_domain_set_num_threads(nthreads, MKL_DOMAIN_VML)` in the MKL library. See: http://www.intel.com/software/products/mkl/docs/webhelp/support/functn_mkl_domain_set_num_threads.html for more info about it. N)rr nthreadss rset_vml_num_threadsr$DsX&rc$tr tSy)a Gets the maximum number of threads to be used in VML operations. This function is equivalent to the call `mkl_domain_get_max_threads (MKL_DOMAIN_VML)` in the MKL library. See: http://software.intel.com/en-us/node/522118 for more info about it. N)rr rrrget_vml_num_threadsr&Ts#%% rct|}|S)a Sets a number of threads to be used in operations. DEPRECATED: returns the previous setting for the number of threads. During initialization time NumExpr sets this number to the number of detected cores in the system (see `detect_number_of_cores()`). )r)r# old_nthreadss rset_num_threadsr)ds$H-L rctS)zE Gets the number of threads currently in use for operations. )rrrrget_num_threadsr+ps   rc$dtjvr!tjdt dyd}t }dt jvrt jddk7s(dt jvrt jddk7r d}t}nF|tkDrtjd |tfz|d kDrtjd |zd }d t jvr3t jd dk7rtt jd }nadt jvr3t jddk7rtt jd}n|}|stjd |zt ||S)a  Detects the environment variable 'NUMEXPR_MAX_THREADS' to set the threadpool size, and if necessary the slightly redundant 'NUMEXPR_NUM_THREADS' or 'OMP_NUM_THREADS' env vars to set the initial number of threads used by the virtual machine. sparczThe number of threads have been set to 1 because problems related to threading have been reported on some sparc machine. The number of threads can be changed using the "set_num_threads" function.rFNUMEXPR_MAX_THREADSOMP_NUM_THREADSTzsNote: detected %d virtual cores but NumExpr set to maximum of %d, check "NUMEXPR_MAX_THREADS" environment variable.zaNote: NumExpr detected %d cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 16.NUMEXPR_NUM_THREADSz!NumExpr defaulting to %d threads.) rplatform_machinelogwarningr)detect_number_of_coresosenvironrinfoint)env_configuredn_coresrequested_threadss r_init_num_threadsr>vsv'***   N$&G+ ;P0QUW0WRZZ'BJJ7H,IR,O [ HHKMTVaLbb c R< HHxzAA BG *rzz:O/PTV/V +@ AB bjj (RZZ8I-Jb-P +< =># HH8@ A%& rcbttdr^dtjvr,tjd}t |t r'|dkDr"|St t jgdS t tjjdd}|dkDr|S y#t$rYywxYw)zC Detects the number of cores on a system. Cribbed from pp. sysconfSC_NPROCESSORS_ONLNr)sysctlz-nzhw.ncpuNUMBER_OF_PROCESSORSr/r) hasattrr7 sysconf_namesr@ isinstancer: subprocess check_outputr8rr)ncpuss rr6r6s r9 B$4$4 4JJ45E%%%!) z../JKL L BJJNN#92>? 19L      s0/B"" B.-B.cHtjd ttjj dd}|tkDrt}|S#t $rG ttjj dd}n#t $r t}YnwxYwY`wxYw)z DEPRECATED: use `_init_num_threads` instead. If this is modified, please update the note in: https://github.com/pydata/numexpr/wiki/Numexpr-Users-Guide z+Deprecated, use `init_num_threads` instead.r2r/r0) r4r5r:r7r8rrr6rr"s rdetect_number_of_threadsrKs KK=>0rzz~~&;R@A+ O 0 02::>>*;R@AH 0-/H 00s5)A B!)BB!BB!BB! B!c,eZdZdZfdZfdZxZS) CacheDictzB A dictionary that prevents itself from growing too much. c:||_tt||y)N) maxentriessuperrM__init__)selfrO __class__s rrQzCacheDict.__init__s$ i'-rct||jkDrC|jdz}t|jd|D]}tt ||tt |||y)N )lenrOrrrPrM __delitem__ __setitem__)rRkeyvalueentries_to_removekrSs rrXzCacheDict.__setitem__s` t9t & $2 5 $))+&'9(9:i215; i*36r)__name__ __module__ __qualname____doc__rQrX __classcell__)rSs@rrMrMs.77rrM)logging getLoggerr]r4r7rGnumexpr.interpreterrrrnumexprrr/rr r r r rr r$r&r)r+r>r6rKdictrMrrrrgsug! OO  >   /d .(77r