1j)&@ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZdd lmZmZd Zd efd Zd edeeeefd efdZd ed dfdZ d"dddddddeedeedeed eedeeeefdeeee fd eefdZd"deed dfdZd#deeded dfdZd$deedeed dfdZ d%deedeedeed eedeedeeee ff dZ d"ddddddd deedeedeed eedeedeeee ff d!Zy)&N)proton)getenv)flags) HookManager LaunchHookInstrumentationHook)Hook)BaseMode)OptionalUnionrreturnctjjjj j }|dk(ry|dk(ryt d)Ncudacuptihip roctracerz/No backend is available for the current target.)tritonruntimedriveractiveget_current_targetbackend ValueError)rs V/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/triton/profiler/profile.py_select_backendrsHnn##**==?GGG& E JKKrmodec|dk(rGtjjjj j }|r|d|S|S|r t |SdS)Ninstrumentation:)rrrrrrstr)rrprefixs r _get_mode_strr%sX##&&--@@BJJ%)&4&!5v53t9$"$rc|dk(r&ddg}|D]}t|dtd|dtjjj j D]\}}|j}t|dttjj|}|Ftjj|x}shtjj||dy)NrHIP_VISIBLE_DEVICESCUDA_VISIBLE_DEVICESz3Proton does not work when the environment variable zK is set on AMD GPUs. Please unset it and use `ROCR_VISIBLE_DEVICES` insteadr) rrrknobsrknob_descriptorsitemskeygetattrtoenvsetenv)rhip_device_envsenvattrdescr,valenv_vals r _check_envr6 s+02HI" Cc4 , I#OZ[ ll))::@@B9 dhh #t  $&,,--t4C$ll005575LL''WQZ8 9rshadowtreecontextdatarrhooknamer:r;r<ctjs$tjjj rydt_|tn|}| tn|}t||}t|tj|||||}t|trtj ||n4|dk(rtj t#|n|t%d||dk(rtj t'|||S)a- Start profiling with the given name and backend. Usage: ```python proton.start("my_profile") # do something proton.finalize() ``` Args: name (str, optional): The name (with path) of the profiling session. If not provided, the default name is "~/proton.", where suffix is the default format according to the data type. For example, if data is "tree", the default name is "~/proton.hatchet". context (str, optional): The context to use for profiling. Available options are ["shadow", "python"]. Defaults to "shadow". data (str, optional): The data structure to use for profiling. Available options are ["tree", "trace"]. Defaults to "tree". backend (str, optional): The backend to use for profiling. Available options are [None, "cupti", "roctracer", "instrumentation"]. Defaults to None, which automatically selects the backend matching the current active runtime. mode (Union[str, BaseMode], optional): The "mode" to use for profiling, which is specific to the backend. Can be a string or an instance of BaseMode (or any subclass thereof). Defaults to None. For "cupti", available options are [None, "pcsampling", "periodic_flushing"]. For "roctracer", available options are ["periodic_flushing"]. For "instrumentation", available options are [None]. Each mode has a set of control knobs following with the mode name. For example, "periodic_flushing" mode has a knob: - format: The output format of the profiling results. Available options are ["hatchet", "hatchet_msgpack", "chrome_trace"]. Default is "hatchet". The can be set via `mode="periodic_flushing:format=chrome_trace"`. hook (Union[str, Hook], optional): The hook to use for profiling. You may pass either: - a string hook name, e.g. "triton" (kernel launch metadata), or - a custom Hook instance. Defaults to None. Returns: session (Optional[int]): The session ID of the profiling session, or None if profiling is disabled. NTrzUnsupported hook: r )r command_linerr)rdisable profiling_onDEFAULT_PROFILE_NAMErr%r6 libprotonstart isinstancer rregisterrrr )r=r:r;rrr<mode_strsessions rrDrD3sf V\\0088E#'< TD#*?oGWd+HwoodGT7HEG$T7+  Z\73 -dX677##06@ NrrHctjr|dk7r tdtj||t j yt j|y)z Activate the specified session. The profiling session will be active and data will be recorded. Args: session (int): The session ID of the profiling session. Defaults to None (all sessions) Returns: None rzEOnly one session can be activated when running from the command line.N)rr?rractivaterC activate_all)rHs rrJrJsL gl`aa! 7#rflushingctjr|dk7r tdtj||t j |yt j||y)ap Stop the specified session. The profiling session's data will still be in the memory, but no more data will be recorded. Args: session (int): The session ID of the profiling session. Defaults to None (all sessions) flushing (bool): Whether to flush the profiling data before deactivating. Defaults to True. Returns: None rzGOnly one session can be deactivated when running from the command line.N)rr?rr deactivaterCdeactivate_all)rHrLs rrNrNsP glbcc7#  *Wh/r output_formatctj||!dt_t j |ytj r|dk7r tdt j||y)a Finalizes a profiling session. Flush and write the profiling data to the file specified by the session name. Args: session (int, optional): The session ID to finalize. If None, all sessions are finalized. Defaults to None. output_format (str, optional): The output format for the profiling results. Available options are ["hatchet", "hatchet_msgpack", "chrome_trace"]. Returns: None NFrzEOnly one session can be finalized when running from the command line.) r unregisterrrArC finalize_allr?rfinalize)rHrPs rrTrTsX7#"}-   'Q,de e7M2rcZtjfd}|S)z Context manager for profiling. Internally use only. Args: See start() for the arguments. Returns: wrapper (function): The wrapped function. cPt  }|i|}t||S)Nr9)rDrN) argskwargsrHretrr:r;funcr<rr=s rwrapperz_profiling..wrappers5gD'PT[_`D#F#7 r) functoolswraps)rZr=r:r;rrr<r[s``````` r _profilingr^s,&__T Nrr=r:r;rrr<c N| fd}|St|S)z Decorator for profiling. Usage: ```python @proton.profile def foo(): pass ``` Args: See start() for the arguments. Returns: decorator (function): The decorator function. c (t|S)Nr_r^)frr:r;r<rr=s r decoratorzprofile..decoratorsadG$PW^bimn nrr_rb)rZr=r:r;rrr<rds `````` rprofileres96 | o o$T7w]ahlmmr)N)NF)Nr")Nr7r8NNN) r\rtriton._C.libprotonrrCtriton._C.libtritonrrhooksrrr hooks.hookr rr typingr r rBr#rr%r6intrDrJboolrNrTr^rerrrnsf 3&??"LL%3%huS(]/C&D%%999(K& !+/'+K 3-Kc]K 3- K c] K 5h' ( K 5d# $Kc]K\$hsm$t$,0 000.3hsm38C=3RV34% !'+ 3-c] 3-  c]  3-  5d# $< #n% !'+#n 3-#nc] #n 3- #n c] #n 3-#n 5d# $#nr