3j*ddlmZgdZddlZddlZddlZddddddZddddddZd Zdd Z d Z d Z Gd dejZ edk(rddlZej e yy)) annotations)cpus runParallelrunNonParallelsafeToParallizeNFupdateFunctionupdateMultiplyunpackIterableupdateSendsIterablectst||Sttd}gfd}|dddlm}m} |5} | ||krft|zz} t|| } |r| fd| D} n| fd| D} | }j| |||krfdddS#1swYSxYw) a3 runs parallelFunction over iterable in parallel, optionally calling updateFunction after each common.cpus * updateMultiply calls. Setting updateMultiply too small can make it so that cores wait around when they could be working if one CPU has a particularly hard task. Setting it too high can make it seem like the job has hung. updateFunction should take three arguments: the current position, the total to run, and the most recent results. It does not need to be pickleable, and in fact, a bound method might be very useful here. Or updateFunction can be "True" which just prints a generic message. If unpackIterable is True then each element in iterable is considered a list or tuple of different arguments to parallelFunction. If updateSendsIterable is True then the update function will get the iterable content, after the output. As of Python 3, partial functions are pickleable, so if you need to pass the same arguments to parallelFunction each time, make it a partial function before passing it to runParallel. Note that parallelFunction, iterable's contents, and the results of calling parallelFunction must all be pickleable, and that if pickling the contents or unpickling the results takes a lot of time, you won't get nearly the speedup from this function as you might expect. The big culprit here is definitely music21 streams. >>> files = ['bach/bwv66.6', 'schoenberg/opus19', 'AcaciaReel'] >>> def countNotes(fn): ... c = corpus.parse(fn) # this is the slow call that is good to parallelize ... return len(c.recurse().notes) >>> #_DOCS_SHOW outputs = common.runParallel(files, countNotes) >>> outputs = common.runNonParallel(files, countNotes) #_DOCS_HIDE cannot pickle doctest funcs. >>> outputs [165, 50, 131] Set updateFunction=True to get an update every 3 * numCpus (-1 if > 2) >>> #_DOCS_SHOW outputs = common.runParallel(files, countNotes, updateFunction=True) >>> outputs = common.runNonParallel(files, countNotes, updateFunction=True) #_DOCS_HIDE Done 0 tasks of 3 Done 3 tasks of 3 With a custom updateFunction that gets each output: >>> def yak(position, length, output): ... print(f'{position}:{length} {output} is a lot of notes!') >>> #_DOCS_SHOW outputs = common.runParallel(files, countNotes, updateFunction=yak) >>> outputs = common.runNonParallel(files, countNotes, updateFunction=yak) #_DOCS_HIDE 0:3 165 is a lot of notes! 1:3 50 is a lot of notes! 2:3 131 is a lot of notes! Or with updateSendsIterable, we can get the original files data as well: >>> def yik(position, length, output, fn): ... print(f'{position}:{length} ({fn}) {output} is a lot of notes!') >>> #_DOCS_SHOW outputs = common.runParallel(files, countNotes, updateFunction=yik, >>> outputs = common.runNonParallel(files, countNotes, updateFunction=yik, #_DOCS_HIDE ... updateSendsIterable=True) 0:3 (bach/bwv66.6) 165 is a lot of notes! 1:3 (schoenberg/opus19) 50 is a lot of notes! 2:3 (AcaciaReel) 131 is a lot of notes! unpackIterable is useful for when you need to send multiple values to your function call as separate arguments. For instance, something like: >>> def pitchesAbove(fn, minPitch): # a two-argument function ... c = corpus.parse(fn) # again, the slow call goes in the function ... return len([p for p in c.pitches if p.ps > minPitch]) >>> inputs = [('bach/bwv66.6', 60), ... ('schoenberg/opus19', 72), ... ('AcaciaReel', 66)] >>> #_DOCS_SHOW outputs = common.runParallel(inputs, pitchesAbove, unpackIterable=True) >>> outputs = common.runNonParallel(inputs, pitchesAbove, unpackIterable=True) #_DOCS_HIDE >>> outputs [99, 11, 123] r rNcdurt|g}td|dydvrOt| zz |D]9}|dkr |tk\rd}n|} s ||,|||;yy)NTDone tasks of FNrminprintrangelen) ii tasksDone thisPosition thisResult iterLengthiterablenumCpus resultsListr r r s D/DATA/.local/lib/python3.12/site-packages/music21/common/parallel.py callUpdatezrunParallel..callUpdates T !R,-I E)Jzl; < = 0 %bNW,D&Er J !#3{#33!%J!,\!:J*"<ZH"<ZR^I_`!K1)Paralleldelayed)n_jobsc3.K|] }|ywN.0i delayFunctionrs r zrunParallel..sHx!-!5xsc34K|]}|ywr'r(r)s r r-zrunParallel..sGh- 4hs) rrrrjoblibr#r$rrextend)rparallelFunctionr r r r totalRunr!r#r$para endPositionrangeGen_rr,rrrs` `` ` @@@@r rrsp  h(8-;-;-;2E G G fGXJH#Kaa&qM(  !T 01 #h>)AA:NKX{3HHxHHGhGG"H   r " x # "  " s %A3C##C-c t g fd}|dt D]4}|r ||}n ||} j|||dz6 S)z This is intended to be a perfect drop in replacement for runParallel, except that it runs on one core only, and not in parallel. Used automatically if we're already in a parallelized function. c|zdk7rydurt|g}td|dydvrLt|z |D]9}|dkr |tk\rd}n|} s ||,|||;yy)NrTrrrr) rrrrrrrr r r s r r!z"runNonParallel..callUpdates  ! #  T !R,-I E)Jzl; < = 0 %b>&92 > !#3{#33!%J!,\!:J*"<ZH"<ZR^I_`!?1r"r)rrappend) rr1r r r r r!r+r6rrs ` `` ` @@r rrsxXJKaa,qM :  !8A;/B!(1+.B21q5 r"cBtj}|dk\r|dz S|S)zf Returns the number of CPUs or if >= 3, one less (to leave something out for multiprocessing) rr9)multiprocessing cpu_count)cpuCounts r rrs(((*H1}!|r"crtdkDxr)tj xrdtjvS)a Check to see if it is safe or even useful to start a parallel process. Will return False if we are in a multiprocessing child process or if there is only one CPU or if pytest's x-dist worker flag is in the environment. * New in v10. r9PYTEST_XDIST_WORKER)rr<parent_processosenvironr(r"r rrs7   4..0 0 4 ! 3r"cvddlm}|j|}t|j j S)Nr)corpus)music21rEparserrecursenotes)fnrEcs r _countNrLs* RA qyy{  !!r"c|dk\ry|dvryy)NrFz bach/bwv66.6zschoenberg/opus19 AcaciaReelTr()r+rJs r _countUnpackedrPsAv DD r"ceZdZdZdZdZy)TestcXgd}ddlm}m}t||}|j |gdt|||j t|||j dttt||d}|j t|d |jd |y) NrNr)rLrP2)r T)r r )r rF) music21.common.parallelrLrPr assertEqual_customUpdate1_customUpdate2list enumerater assertNotIn)selffilesrLrPoutputpasseds r &x_figure_out_segfault_testMultiprocessz+Test.x_figure_out_segfault_testMultiprocesssCCUG, 0E#'#6#6 8 E#'#6#6(, .T)E"23+,02 Va( 'r"ct|j|d|j|d|j|gdy)NrrT)rY assertLessassertIn)r_r+totalras r rZzTest._customUpdate1&s. " 1 fn-r"c,|j|gdy)NrN)rf)r_r+ unused_total unused_outputrJs r r[zTest._customUpdate2+s bMNr"N)__name__ __module__ __qualname__rcrZr[r(r"r rRrRs((. Or"rR__main__)returnbool) __future__r__all__r<rBunittestrrrrrLrPTestCaserRrkrFmainTestr(r"r rvs#   $ $$) Nh#'"#"'', 1h 6" O8  O> zGTr"