Iriw"ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZmZd Zd efd Zd ed eeee fd efdZd ed dfdZ d!dddddddeedeedeed eed eeee fdeef dZd!deed dfdZd!deed dfdZd"deedeed dfdZ d#deedeedeed eed eedeef dZ d!ddddddddeedeedeed eed eedeef d ZdS)$N)proton)getenv)flags) HookManager LaunchHookInstrumentationHook)BaseMode)OptionalUnionrreturnctjjjj}|dkrdS|dkrdSt d)Ncudacuptihip roctracerz/No backend is available for the current target.)tritonruntimedriveractiveget_current_targetbackend ValueError)rs o/var/www/html/bestrading.cuttalo.com/models/btc_v9/venv/lib/python3.11/site-packages/triton/profiler/profile.py_select_backendrsMn#*==??GG&w E  {JKKKrmodec|dkr8tjjjj}|r|d|n|S|rt |ndS)Ninstrumentation:)rrrrrrstr)rrprefixs r _get_mode_strr$s]###&-@@BBJ%)5&!!4!!!v5 $3t999"$rc|dkr,ddg}|D]%}t|dtd|d&tjjjD]\}}|j}t|dhttjj|}|Gtj |x}r&tj ||ddS)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_envr5s+02HI"  Cc4  , [#[[[- l):@@BB99 dh #t   $&,-t44C$l0055579L''WQZ888 99rshadowtreecontextdatarrhooknamer9r:r;ctjstjjjrdSdt_|tn|}|tn|}t||}t|tj |||||}|dkr!tjt||dkr"tjt!|||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"]. For "roctracer", available options are [None]. For "instrumentation", available options are [None]. Each mode has a set of control knobs following with the mode name. For example, "pcsampling" has an "interval" control knob, expressed as "pcsampling:interval=1000". hook (str, optional): The hook to use for profiling. Available options are [None, "launch"]. Defaults to None. Returns: session (int): The session ID of the profiling session. NTrr)r command_linerr(rdisable profiling_onDEFAULT_PROFILE_NAMErr$r5 libprotonstartrregisterrr )r<r9r:rrr;mode_strsessions rrCrC2s^ V\08E#'<  TD#*?oGWd++HwodGT7HEEG xZ\\7333###066@@@ NrrFctjr|dkrtdtj||t jdSt j|dS)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>rractivaterB activate_allrFs rrHrHxsi bgll`aaa!!!     7#####rctjr|dkrtdtj||t jdSt j|dS)a  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) Returns: None rzGOnly one session can be deactivated when running from the command line.N)rr>rr deactivaterBdeactivate_allrJs rrLrLsi dgllbccc7### """""W%%%%%rr! output_formatctj||"dt_t j|dStjr|dkrtdt j||dS)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", "chrome_trace"]. Returns: None NFrzEOnly one session can be finalized when running from the command line.) r unregisterrr@rB finalize_allr>rfinalize)rFrNs rrRrRsu7###"}-----   f'Q,,dee e7M22222rc^tjfd}|S)z Context manager for profiling. Internally use only. Args: See start() for the arguments. Returns: wrapper (function): The wrapped function. c`t  }|i|}t||S)Nr8)rCrL) argskwargsrFretrr9r:funcr;rr<s rwrapperz_profiling..wrappersFgD'PT[_```dD#F##7 r) functoolswraps)rXr<r9r:rrr;rYs``````` r _profilingr\s^&_T Nrr<r9r:rrr;c V| 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. Nc 0t|S)Nr]r\)frr9r:r;rr<s r decoratorzprofile..decorators#adG$PW^bimnnn nrr]r`)rXr<r9r:rrr;rbs `````` rprofilercst6 | o o o o o o o o o o$T7w]ahlmmmmr)N)Nr!)Nr6r7NNN)rZrtriton._C.libprotonrrBtriton._C.libtritonrrhooksrrr rr typingr r rAr"rr$r5rCintrHrLrRr\rcrrrjso 333333&&&&&&??????????""""""""LLLLL%3%huS(]/C&D%%%%%999999(C& !+/CCC 3-Cc]C 3- C c] C 5h' ( C 3-CCCCL$$hsm$t$$$$,&& &&&&&,33hsm38C=3RV33334% ! 3-c] 3-  c]  3-  3-< #n% !#n#n#n 3-#nc] #n 3- #n c] #n 3-#n 3-#n#n#n#n#n#nr