B ;9\@sddlZddlZddlmZmZddlmZerDddlmZmZm Z ddZ ddZ d d Z d d Z d dZddZGdddeZdS)N)_set_num_threads MAX_THREADS)use_vml)_get_vml_version_set_vml_accuracy_mode_set_vml_num_threadscCstr tSdSdS)z Get the VML/MKL library version.N)rrrr,lib/python3.7/site-packages/numexpr/utils.pyget_vml_versionsr cCsXtrPddddd}dddd }|t|kr6td t||d}||Sd 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. r)Nlowhighfastrrr)r r r z9mode argument must be one of: None, 'high', 'low', 'fast'N)rlistkeys ValueErrorrget)modeZacc_dictZacc_reverse_dictZretvalrrr set_vml_accuracy_modes  rcCstr t|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)nthreadsrrr set_vml_num_threads=s rcCs t|}|S)aV Sets a number of threads to be used in operations. 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()`). If you are using Intel's VML, you may want to use `set_vml_num_threads(nthreads)` to perform the parallel job with VML instead. However, you should get very similar performance with VML-optimized functions, and VML's parallelizer cannot deal with common expresions like `(x+1)*(x-2)`, while Numexpr's one can. )r)rZ old_nthreadsrrr set_num_threadsMsrcCsttdrJdtjkr6td}t|trJ|dkrJ|SnttdddgSy"ttj dd}|dkrj|SWnt k rYnXd S) zC Detects the number of cores on a system. Cribbed from pp. sysconfSC_NPROCESSORS_ONLNrZsysctlz-nzhw.ncpuZNUMBER_OF_PROCESSORSr ) hasattros sysconf_namesr isinstanceint subprocessZ check_outputenvironrr)Zncpusrrr detect_number_of_coresas   r$c Csnyttjdd}WnFtk r\yttjdd}Wntk rVt}YnXYnX|tkrjt}|S)zt If this is modified, please update the note in: https://github.com/pydata/numexpr/wiki/Numexpr-Users-Guide ZNUMEXPR_NUM_THREADSrZOMP_NUM_THREADS)r!rr#rrr$r)rrrr detect_number_of_threadsxsr%cs,eZdZdZfddZfddZZS) CacheDictzB A dictionary that prevents itself from growing too much. cs||_tt||dS)N) maxentriessuperr&__init__)selfr') __class__rr r)szCacheDict.__init__cs\t||jkrF|jd}x,t|d|D]}tt||q.Wtt|||dS)N )lenr'rrr(r& __delitem__ __setitem__)r*keyvalueZentries_to_removek)r+rr r/s  zCacheDict.__setitem__)__name__ __module__ __qualname____doc__r)r/ __classcell__rr)r+r r&s r&)rr"Znumexpr.interpreterrrZnumexprrrrrr rrrr$r%dictr&rrrr  s