TiSSKJr SSKrSSKrS Sjr"SS5rSr"SS5rS S S jjrSSS jjr g)) annotationsNcdSSKnURU5nUR5RnU$)z Returns the physical memory used by a process. Args: pid: Process id, current one is `os.getpid()`. Returns: Physical memory. It relies on the module *psutil*. rN)psutilProcess memory_inforss)pidrprocessmems [/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/tools/memory_peak.pyget_memory_rssr s.nnS!G     # #C JcV\rSrSrSrS S SjjrS SjrS SjrSr\ SSj5r Sr g )MonitorcJSUlSUlSUlSUlSUlg)Nr)max_peakaverage n_measuresbeginendselfs r __init__Monitor.__init__s%     rc[U5n[URU- URS-UR- U- URU- UR U- UR U- S9$)Ng?)peakmeannrr)floatdictrrrrr)runitfunits r to_dictMonitor.to_dict%sbd &#doo5=oo%**u$5   rcURRSURSURSURS3$)Nz(peak=z , average=z, n=)) __class____name__rrrrs r __repr__Monitor.__repr__/sA~~&&'vdmm_=||nD(9 < rcURS:XaXl[XR5UlU=RU- slXlU=RS- slg)Nr)rrmaxrrr)rr s r updateMonitor.update5sD ??a JC/    1rcURUR5 URUR5 URUR5 URUR5 URUR 5 gN)sendrrrrr)rconns r r3 Monitor.send=sR $--  $,, $//" $** $((rcU"5nUR5UlUR5UlUR5UlUR5UlUR5UlU$r2)recvrrrrr)clsr4ms r r7 Monitor.recvDsO EYY[ IIK yy{ ))+ r)rrrrrN)r-)r"intreturndict[str, float])r<str)r r )r<r) r) __module__ __qualname____firstlineno__rr$r*r/r3 classmethodr7__static_attributes__rr rrs/  rrcz^^URS5 UR5nUR5nUR5nSSKnURU5nU(ahSSKJnJnJmJnJ n U"5 U"5n [U 5V s/sH o"U 5PM sn mUU4Sjn [U 5V s/sH n [5PM n n O/n [5nURS5 UR5RnURU5 U(a,[W "5U 5HunnURU5 M UR!US9(aUR5nUS:XaOMUR5RnURU5 U(a,[W "5U 5HunnURU5 M URU5 UR[#U 55 U HnURU5 M U(aW "5 UR%5 gs sn fs sn f)Nr)nvmlDeviceGetCountnvmlDeviceGetHandleByIndexnvmlDeviceGetMemoryInfonvmlInit nvmlShutdowncT>TVs/sHnT"U5RPM sn$s snfr2)used)hhandlesrIs r gpu_used%_process_memory_spy..gpu_usedms'=DEW+A.33WE EEs%)timeout)r3r7rrpynvmlrGrHrIrJrKrangerrrr/zippolllenclose)r4r rRcudarr rGrHrJrKn_gpusirPgpuscpur rgcoderrOrIs @@r _process_memory_spyrbOsIIbM ))+CiikG 99;DnnS!G     #%:?-H-Q-a0-H F$)=1=a =1 )CIIbM !!#'' 3 HJ-1 . 99W9 %99;Drz      # #CJJsO  D)DAq HHQK*HHTNIIc$i  t  JJLOI 2s H31H8c:\rSrSrSrSS SjjrS SjrS SjrSrg) MemorySpyz Information about the spy. It class method `start`. Method `stop` can be called to end the measure. Args: pid: process id of the process to spy on delay: spy on every delay seconds cuda: enable cuda monitoring cHXlX lX0lUR5 gr2)r delayrZstart)rr rgrZs r rMemorySpy.__init__s   rc|[R"5uUlUl[R"[ UR4S9UlUR R5 URR5nUS:wa[SUS35eURRUR5 URRUR5 URRUR(aSOS5 URR5nUS:wa[SUS35eU$)z+Starts another process and tells it to spy.)targetargsrFz0The child processing is supposed to send -2 not .r-rz6The child processing is supposed to send -2 again not )multiprocessingPipe parent_conn child_connrrb child_processrhr7 RuntimeErrorr3r rgrZ)rdatas r rhMemorySpy.starts,;,@,@,B)$/,44&doo-?    "$$& 2:!QRVQWWXYZ Z dhh' djj) 499a!4$$& 2:HaP  rcURRS5 [RUR5/nURR5n/n[ U5H1nUR [RUR55 M3 URR 5 URR5 [US9nUR(aX5S'U$)zStops spying on.rS)r^r]) rpr3rr7rUappendrYrrjoinr!rZ)rr^r[r]_ress r stopMemorySpy.stops b!||D,,-.!!&&(vA KK T%5%56 7   !sm 99K r)rqrrrZrgrpr N){Gz?F)r r;rgr rZbool)r<rd)r<dict[str, list[Monitor]]) r)r?r@rA__doc__rrhr{rCrDrr rdrds (rrdcJUc[R"5n[XU5$)aStarts the memory spy. The function starts another process spying on the one sent as an argument. Example:: .. code-block:: python from onnxscript.tools.memory_peak import start_spying_on, flatten p = start_spying_on() # ... # code to measure # ... stat = p.stop() print(stat) print(flatten(stat)) Args: pid: process id to spy or the the current one. delay: delay between two measures. cuda: True or False to get memory for cuda devices )osgetpidrd)r rgrZs r start_spying_onrs!2 {iik S &&rc6USSRSS9nSU;aG[US5H5up4URSS9R5HupVXbSUSU3'M M7 U(a)UR5VVs0sH upVUU3U_M nnnU$s snnf)z9Flattens a dictionary produced by :meth:`MemorySpy.stop`.r^ri)r"r]gpury)r$ enumerateitems)psprefixobsr\r`kvs r flattenrs U)A,  E  *C |bj)DA u -335%&c!AaSM"6*-0YY[9[TQ&!~q [9 J:s=B)r r;r<r;)Nr}F)r z int | Nonergr rZr~r<rd))rrrr>r<r=) __future__rrnrr rrbrdrrrDrr rsb# &//dCL55r?D' '#('7;''< r