Q%dY38ddlZddlZddlZddlZddlZddlZdejvrrddlZej dZ ej e ddZ ejjdkreje ejdxxeje zz cc<ddlTdd lmZddlTdd lmZd d gezezZd Zd a d a dZddZ ddZGddejjZ ddZ GddejjZ!Gdd Z"dZ#dZ$dZ%dS)NWindowsLibrarybinPATH)*)__all__Monkey is_activez Python API for Intel(R) oneAPI Threading Building Blocks (oneTBB) extended with standard Python's pools implementation and monkey-patching. Command-line interface example: $ python3 -m tbb $your_script.py Runs your_script.py in context of tbb.Monkey Fz libirml.so.1cddl}|jdkrCtjtdt jdt jzdzksJddlm}||td dS) z Some testsrNLinuxzldd z#| grep -E 'libimf|libsvml|libintlc'r)testdone) platformsystemctypesCDLLlibirmlos_api__file__rprint)argrrs ,lib/python3.11/site-packages/tbb/__init__.py_testr;sOOOxG## Gbit} 45Z Z[[[[[[DIII &MMMMMcddlm}||||||trY tjt }|dS#tdt tj YdSxYwdSNr)workerzWarning: Can not load file multiprocessing.poolr" ipc_enabledrrrrelease_resourcesrsysstderr)inqueueoutqueue initializerinitargsmaxtasksr"librmls rtbb_process_pool_worker27r1Fs++++++ F7Hk8X>>>F F[))F  $ $ & & & & & F *G#* E E E E E E E FFs -A #A3c eZdZdZdZdZdS)TBBProcessPool27c ddlm}t|jt |jz D]}|t|j|j |j |j |j f}|j ||jdd|_d|_||ddS zBring the number of pool processes up to the specified number, for use after reaping workers which have exited. r)debug)targetargsProcess PoolWorkerTz added workerN)multiprocessing.utilr6range _processeslen_poolr9r1_inqueue _outqueue _initializer _initargs_maxtasksperchildappendnamereplacedaemonstartselfr6iws r_repopulate_poolz!TBBProcessPool27._repopulate_poolSs /.....tTZ899 " "A $=#'=$.#'#4#'>43I#KA J  a V^^I|<r?r9rar@rArBrCrD_wrap_exceptionrErFrGrHrIrJs rrNz TBBProcessPool3._repopulate_pool}s /.....tTZ899 " "A $<#'=$.#'#4#'>43I#'#7#9A J  a V^^I|< ! 23326.BJ. /K$$&& B BD DM$'t{4/@ A A A A B Br)NF) r\r]r^__doc__rzrxrsrrrZrrrr r sp  FH 7 7 7 7 $ $ $" B B B B Brc tjt}||dS#t $r'}t d|tjYd}~dSd}~wwxYw)Nz6Warning: Can not initialize name of shared semaphores:r#) rrrset_active_sem_nameset_stop_sem_name Exceptionrr)r*)r0es r init_sem_namersW%%""$$$  """""  F:          sAA A6A11A6ctrS tjt}|dS#t dt jYdSxYwdS)Nz*Warning: Can not release shared semaphoresr#)r'rrrrelease_semaphoresrr)r*)r0s r tbb_atexitrsw# #[))F  % % ' ' ' ' ' # >z # # # # # # # ##s -8Acddl}ddl}|dd|j}|jdkrI|ddd |d d dd |d dd |ddt tdd|dddd |dddd |dddd|dd |d!|jd"#| }|j rd$tj d%<|jdkr|j rd&|_|jrtj d'sd(}d)}d$tj d'<t!dtj |d*d+}||vrt%d,|d-|d.n3d+|gt)|ztj |<|j r|jdksJ t+d/d05}t|}dddn #1swxYwY|dkrt%d1d$tj d2<n*#t%d3t/jd4YnxYwtjt.jt.jdd5gt.jd4dRJd7|jg|jzt._|jdko|jat>rd$nd8tj d9<t>r'tAj!tDtGtj d:sd8tj d:<d;|jztIvr$tId;|jz|jSddl%} |j&r| j'n| j(} tS|j*|j+<5| |jd=>ddddS#1swxYwYdS)?Nrzpython3 -m tbba Run your Python script in context of tbb.Monkey, which replaces standard Python pools and threading layer of Intel(R) oneAPI Math Kernel Library (oneMKL) by implementation based on Intel(R) oneAPI Threading Building Blocks (oneTBB). It enables multiple parallel tasks to be executed on the same thread pool and coordinate number of threads across multiple processes thus avoiding overheads from oversubscription. )prog descriptionformatter_classrz--ipc store_truezAEnable inter-process (IPC) coordination between oneTBB schedulers)actionhelpz-az --allocatorzOEnable oneTBB scalable allocator as a replacement for standard memory allocatorz--allocator-huge-pagesz4Enable huge pages for oneTBB allocator (implies: -a)z-pz--max-num-threadsz:Initialize oneTBB with P max number of threads per processP)defaulttypermetavarz-bz --benchmarkzBlock oneTBB initialization until all the threads are created before continue the script. This is necessary for performance benchmarks that want to exclude lazy scheduler initialization effects from the measurementsz-vz --verbosez'Request verbose and version informationz-mmodulezExecutes following as a module)rdestrrFzScript or module name)rr8zCommand line arguments)nargsr1 TBB_VERSIONT_TBB_MALLOC_PRELOADzlibtbbmalloc_proxy.so.2 LD_PRELOAD:zInfo:containszalready z/proc/sys/vm/nr_hugepagesrzoneTBB: Pre-allocated huge pages are not currently reserved in the system. To reserve, run e.g.: sudo sh -c 'echo 2000 > /proc/sys/vm/nr_hugepages'TBB_MALLOC_USE_HUGE_PAGESzoneTBB: Failed to read number of pages from /proc/sys/vm/nr_hugepages Is the Linux kernel configured with the huge pages feature?rtbbFzRe-execution failed0 IPC_ENABLE KMP_BLOCKTIME_)rqrr__main__)run_name),rargparseArgumentParserArgumentDefaultsHelpFormatterr add_argumentrorm REMAINDER parse_argsverboserrallocator_huge_pages allocatorgetfiltersplitrrSlistopenreadr)exitexecl executableargvrFr8ipcr'atexitregisterrrrvrunpyr run_modulerun_pathr rqrr) rrparserr8libtbbmalloc_lib ld_preload preload_listfpagesrrunfs r_mainrsw OOOOOO  $ $*:I#+"H%JJFxG##GL`  b b bD- n  p p p4\S  U U U 1;N;P;PWZYcfhhh mLXYYY k,FHHH \=??? %<=== h&85777     D |($' =!xG##  $ "!DN > 0"*..1F"G"G 08 %J03BJ, -!$ z2(F(F(L(LS(Q(QRRL<//gz:7GUUUU),3C2DtLGYGY2Y)Z)Z :&( &x((G3333  93??.1 #AFFHH ...............zzUVVV>ABJ:;; Z[[[HQKKKKK HS^S^T5 P38ABB< P P P P // / / / {TY&CH!(/##w.;48K&1:sssBJ| ### :>>/ * **&) ?# TY'))##)wyyty)4955 #';BuEN D$8DN S S S 1 1 DZ 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1sBK -"J K JK "J#'K %K2Q22Q69Q6rP)NrN)NrNF)&r&multiprocessingrrr)rrrsitegetsitepackages path_to_envpathrS path_to_libsrradd_dll_directoryrpathseprpr api__allpool pool__allrr r'rrr1rr3rarcr rrrrrrrsp"  !!!!KKK&$&((+K7<< Y>>L ""\***Jv"*|33$$$$$$&&&&&& [ !H ,y 8   5 - MO%) F F F F+0<LN:? F F F Fo*/>HBHBHBHBHBHBHBHBV###S1S1S1S1S1r