JTh SSKrSSKr/SQrS S\S\SS4SjjrSr\RS S\S\4Sjj5rS\4S jr S\4S jr \RS \4S j5r g)N)startstopprofile metal_captureis_metal_capture_enabledis_capturing_metalmodewait_until_completedreturncUR5RSS5n[RR X!5 g)aEStart OS Signpost tracing from MPS backend. The generated OS Signposts could be recorded and viewed in XCode Instruments Logging tool. Args: mode(str): OS Signpost tracing mode could be "interval", "event", or both "interval,event". The interval mode traces the duration of execution of the operations, whereas event mode marks the completion of executions. See document `Recording Performance Data`_ for more info. wait_until_completed(bool): Waits until the MPS Stream complete executing each encoded GPU operation. This helps generating single dispatches on the trace's timeline. Note that enabling this option would affect the performance negatively. .. _Recording Performance Data: https://developer.apple.com/documentation/os/logging/recording_performance_data  N)lowerreplacetorch_C_mps_profilerStartTrace)r r mode_normalizeds J/var/www/auris/envauris/lib/python3.13/site-packages/torch/mps/profiler.pyrrs.(jjl**33O HH$$_Kc@[RR5 g)z6Stops generating OS Signpost tracing from MPS backend.N)rr_mps_profilerStopTracerrrr)s HH##%rc#^# [X5 Sv [5 g![5 f=f7f)a Context Manager to enabling generating OS Signpost tracing from MPS backend. Args: mode(str): OS Signpost tracing mode could be "interval", "event", or both "interval,event". The interval mode traces the duration of execution of the operations, whereas event mode marks the completion of executions. See document `Recording Performance Data`_ for more info. wait_until_completed(bool): Waits until the MPS Stream complete executing each encoded GPU operation. This helps generating single dispatches on the trace's timeline. Note that enabling this option would affect the performance negatively. .. _Recording Performance Data: https://developer.apple.com/documentation/os/logging/recording_performance_data N)rr)r r s rrr.s$ d)  s- - *-c>[RR5$)zlChecks if `metal_capture` context manager is usable To enable metal capture, set MTL_CAPTURE_ENABLED envvar )rr_mps_isCaptureEnabledrrrrrGs 88 ) ) ++rc>[RR5$)z%Cheks if metal capture is in progress)rr_mps_isCapturingrrrrrNs 88 $ $ &&rfnamec## [RRU5 Sv [RR 5 [RR 5 g![RR 5 f=f7f)zBConext manager that enables capturing of Metal calls into gputraceN)rr_mps_startCapturemps synchronize_mps_stopCapture)rs rrrSsO$ ""5)   !!#!!#sBAA$B$ BB)intervalF) contextlibr__all__strboolrrcontextmanagerrrrrrrrr+s  LLLL0&  #$0,$,'D'  $$$r