e ddlZejeZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZe r ddlmZmZmZmZdZdZd Zd Zd Zd Zd ZdZdZGddeZdS)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_threadsc0trtSdS)z* Get the VML/MKL library version. N)rr -lib/python3.11/site-packages/numexpr/utils.pyget_vml_versionrs!!!trctrwddddd}dddd }|t|vrtd t ||d}||Sd S) 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 &s& Aq!<<$F;; tHMMOO,, , ,KMM M' T1(=(=>>##F+++trc2trt|dS)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$Es'X&&&Drc0trtSdS)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&Us&#%%% 4rc$t|}|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)es$H--L rctS)zE Gets the number of threads currently in use for operations. )rrrrget_num_threadsr+qs   rcdtjvr+tdt ddSd}t }dt jvr d}t}nT|tkr$t d|tfz|dkrt d |zd}d t jvr tt jd }nOd t jvr tt jd }n!|}|st d |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_THREADSTzsNote: detected %d virtual cores but NumExpr set to maximum of %d, check "NUMEXPR_MAX_THREADS" environment variable.z`Note: NumExpr detected %d cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 8.NUMEXPR_NUM_THREADSOMP_NUM_THREADSz!NumExpr defaulting to %d threads.) rplatform_machinelogwarningr)detect_number_of_coresosenvironrinfoint)env_configuredn_coresrequested_threadss r_init_num_threadsr=wsn'***     qN$&&G ** [ HHKMTVaLbb c c c Q;; HHwx A A AG ** +@ ABB bj ( ( +< =>># B HH8@ A A A%&&& rcttdrcdtjvr2tjd}t |t r|dkr|Sn#t t jgdS t tj dd}|dkr|Sn#t$rYnwxYwdS)zC Detects the number of cores on a system. Cribbed from pp. sysconfSC_NPROCESSORS_ONLNr)sysctlz-nzhw.ncpuNUMBER_OF_PROCESSORSr) hasattrr6 sysconf_namesr? isinstancer9 subprocess check_outputr7rr)ncpuss rr5r5s r9M B$4 4 4J455E%%% %!)) z./J/J/JKKLL L BJNN#92>>?? 199L       1s:4B00 B=<B=cvtd ttjdd}n]#t $rP ttjdd}n#t $rt}YnwxYwYnwxYw|tkrt}|S)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.r0rCr1) r3r4r9r6r7rrr5rr"s rdetect_number_of_threadsrKs KK=>>>0rz~~&;R@@AA 000 02:>>*;R@@AAHH 0 0 0-//HHH 00+ Os5-A B$-BB$BB$BB$#B$c,eZdZdZfdZfdZxZS) CacheDictzB A dictionary that prevents itself from growing too much. cf||_tt||dS)N) maxentriessuperrM__init__)selfrO __class__s rrQzCacheDict.__init__s.$ i''-----rcFt||jkr^|jdz}t|d|D]*}t t ||+t t |||dS)N )lenrOrrrPrM __delitem__ __setitem__)rRkeyvalueentries_to_removekrSs rrXzCacheDict.__setitem__s t99t & & $2 5 $))++&&'9(9'9: 6 6i&&2215555 i**366666r)__name__ __module__ __qualname____doc__rQrX __classcell__)rSs@rrMrMs[.....777777777rrM)logging getLoggerr]r3r6rGplatformnumexpr.interpreterrrrnumexprrrCrr r r r rr r$r&r)r+r=r5rKdictrMrrrrhsg!! OOOOOOOOOO  >       ---`   .(7777777777r