ó YO\c@sãddlZejeƒZddlZddlZddlZddlmZm Z ddl m Z e rŠddlm Z m Z mZnd„Zd„Zd„Zd„Zd „Zd „Zd „Zd efd „ƒYZdS(iÿÿÿÿN(t_set_num_threadst MAX_THREADS(tuse_vml(t_get_vml_versiont_set_vml_accuracy_modet_set_vml_num_threadscCstr tƒSdSdS(s Get the VML/MKL library version.N(RRtNone(((s,lib/python2.7/site-packages/numexpr/utils.pytget_vml_versionscCs‘tr‰idd 6dd6dd6dd6}idd6dd6dd6}||jƒkrdtdƒ‚nt|j|dƒƒ}|j|ƒSd Sd S( s> 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. iitlowithighitfasts9mode argument must be one of: None, 'high', 'low', 'fast'N(RRtkeyst ValueErrorRtget(tmodetacc_dicttacc_reverse_dicttretval((s,lib/python2.7/site-packages/numexpr/utils.pytset_vml_accuracy_mode"s"  cCstrt|ƒndS(s[ 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(tnthreads((s,lib/python2.7/site-packages/numexpr/utils.pytset_vml_num_threadsAs cCst|ƒ}|S(sV 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(Rt old_nthreads((s,lib/python2.7/site-packages/numexpr/utils.pytset_num_threadsQs cCsdtjƒkr-tjdƒtdƒdSt}tƒ}dtjkrZt }t }nL|t kr€tj d|t fƒn|dkr¦tj d|ƒd}ndtjkrËt tjdƒ}nEd tjkrðt tjd ƒ}n |}|stj d |ƒnt|ƒ|S( s  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. tsparcsÃThe 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.itNUMEXPR_MAX_THREADSssNote: detected %d virtual cores but NumExpr set to maximum of %d, check "NUMEXPR_MAX_THREADS" environment variable.is`Note: NumExpr detected %d cores but "NUMEXPR_MAX_THREADS" not set, so enforcing safe limit of 8.tNUMEXPR_NUM_THREADStOMP_NUM_THREADSs!NumExpr defaulting to %d threads.( tplatformtmachinetlogtwarningRtFalsetdetect_number_of_corestostenvirontTrueRtinfotint(tenv_configuredtn_corestrequested_threads((s,lib/python2.7/site-packages/numexpr/utils.pyt_init_num_threadsds.        cCsµttdƒrndtjkrOtjdƒ}t|tƒrk|dkrk|SqnttjdddgƒƒSny/ttjj ddƒƒ}|dkrœ|SWnt k r°nXd S( sC Detects the number of cores on a system. Cribbed from pp. tsysconftSC_NPROCESSORS_ONLNitsysctls-nshw.ncputNUMBER_OF_PROCESSORSti( thasattrR!t sysconf_namesR*t isinstanceR%t subprocesst check_outputR"R R (tncpus((s,lib/python2.7/site-packages/numexpr/utils.pyR ”s  cCs•tjdƒyttjjddƒƒ}WnMtk r{yttjjddƒƒ}Wq|tk rwtƒ}q|XnX|tkr‘t}n|S(s¥ DEPRECATED: use `_init_num_threads` instead. If this is modified, please update the note in: https://github.com/pydata/numexpr/wiki/Numexpr-Users-Guide s+Deprecated, use `init_num_threads` instead.RR.R( RRR%R!R"R R R R(R((s,lib/python2.7/site-packages/numexpr/utils.pytdetect_number_of_threads«s     t CacheDictcBs eZdZd„Zd„ZRS(sB A dictionary that prevents itself from growing too much. cCs#||_tt|ƒj|ƒdS(N(t maxentriestsuperR6t__init__(tselfR7((s,lib/python2.7/site-packages/numexpr/utils.pyR9Äs cCsst|ƒ|jkrV|jd}x1|jƒ| D]}tt|ƒj|ƒq3Wntt|ƒj||ƒdS(Ni (tlenR7R R8R6t __delitem__t __setitem__(R:tkeytvaluetentries_to_removetk((s,lib/python2.7/site-packages/numexpr/utils.pyR=Ès  (t__name__t __module__t__doc__R9R=(((s,lib/python2.7/site-packages/numexpr/utils.pyR6¿s (tloggingt getLoggerRBRR!R2Rtnumexpr.interpreterRRtnumexprRRRRRRRRR)R R5tdictR6(((s,lib/python2.7/site-packages/numexpr/utils.pyt s         0