&`i NddlTddlmZmZmZddlZddlZddlZddlZddl Z ddl m Z m Z Gdde ZGdde ZGd d e ZGd d e ZGd de ZGdde ZGdde ZdZdZdZGddeZiejdejdejdejdejdejdej dej!dej"d ej#d!ej$d"ej%d#ej&d$ej'd%ej(d&ej)d'Z*Gd(d)eZ+Gd*d+eZ,gd,Z-Gd-d.e.Z/Gd/d0eZ0Gd1d2e.Z1Gd3d4e.Z2Gd5d6e.Z3Gd7d8eZ4Gd9d:eZ5da6e j7Z8da9e:Z;d;Zd>Z?d?Z@d@ZAdAZBdBZCdCZDdDZEdEZFdFZGdGZHdfdIZIdfdJZJdKZKdLZLdMZMdNZNdOZOdPZPdQZQdRZRdSZSdTZTdUZUdVZVdWZWdXZXdYZYdZZZd[Z[d\Z\d]Z]d^e^d_e_d`e_fdaZ`dbZadcZbddZcdeZddS)g)*)joinrealpathisfileN)IntEnumautoceZdZdS)ROCMLError_NotSupportedN__name__ __module__ __qualname__}/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/thirdparty/pyamdsmi/pyamdsmi.pyr r $Drr ceZdZdS)ROCMLError_FunctionNotFoundNr rrrrr(rrrceZdZdS)ROCMLError_LibraryNotFoundNr rrrrr,rrrceZdZdS)ROCMLError_DriverNotLoadedNr rrrrr0rrrceZdZdS)ROCMLError_UnknownNr rrrrr4rrrceZdZdS)ROCMLError_UninitializedNr rrrrr8rrrczeZdZeZ eZ eZ eZ eZdS) ROCMLStateN) r r rr UNINITIALIZED INITIALIZEDDISABLED_PYROCML_NOT_AVAILABLEDISABLED_CONFIGDISABLED_LIBRARY_NOT_FOUNDrrrrr<sUDFFM3$&&K.%)TVV"dffOZ!%44rrzlibrocm_smi64.so cReZdZdZdZdZdZdZdZdZ dZ d Z e Z d Z d Zd Zd ZdZdZdZdS) rsmi_status_tr N)r r rRSMI_STATUS_SUCCESSRSMI_STATUS_INVALID_ARGSRSMI_STATUS_NOT_SUPPORTEDRSMI_STATUS_FILE_ERRORRSMI_STATUS_PERMISSIONRSMI_STATUS_OUT_OF_RESOURCESRSMI_STATUS_INTERNAL_EXCEPTIONRSMI_STATUS_INPUT_OUT_OF_BOUNDSRSMI_STATUS_INIT_ERRORRSMI_INITIALIZATION_ERRORRSMI_STATUS_NOT_YET_IMPLEMENTEDRSMI_STATUS_NOT_FOUNDRSMI_STATUS_INSUFFICIENT_SIZERSMI_STATUS_INTERRUPTRSMI_STATUS_UNEXPECTED_SIZERSMI_STATUS_NO_DATARSMI_STATUS_UNKNOWN_ERRORrrrr'r'Psv" #  #& %("&)#  6&)#$'!"% *rr'zOperation was successfulzInvalid arguments providedz!Not supported on the given systemzProblem accessing a filezPermission deniedz*Unable to acquire memory or other resourcez An internal exception was caughtz0Provided input is out of allowable or safe rangez)Error occurred during rsmi initializationz3Requested function is not implemented on this setupzItem searched for but not foundz Insufficient resources availablez#Interrupt occurred during executionzUnexpected amount of data readz!No data found for the given inputzUnknown error occurredceZdZdZdS)rsmi_init_flags_tr(N)r r rRSMI_INIT_FLAG_ALL_GPUSrrrrIrIys!rrIc"eZdZdZeZdZdZeZdS)rsmi_memory_type_trr(r)N)r r rRSMI_MEM_TYPE_FIRSTRSMI_MEM_TYPE_VRAMRSMI_MEM_TYPE_VIS_VRAMRSMI_MEM_TYPE_GTTRSMI_MEM_TYPE_LASTrrrrLrL}s.,*rrL)VRAMVIS_VRAMGTTc$eZdZdefdefdefgZdS)rsmi_retired_page_record_t page_address page_sizestatusN)r r rc_uint64c_int_fields_rrrrVrVs-*h'5!#HHHrrVceZdZdZeZeZdS)rsmi_sw_component_trN)r r rRSMI_SW_COMP_FIRSTRSMI_SW_COMP_DRIVERRSMI_SW_COMP_LASTrrrr^r^s",+rr^c*eZdZdefdefdeezfgZdS)rsmi_frequencies_t num_supportedcurrent frequencyN)r r rc_int32c_uint32rZRSMI_MAX_NUM_FREQUENCIESr\rrrrcrcs4 '*H%h)AABDHHHrrcc$eZdZdefdeezfgZdS)rsmi_pcie_bandwidth_t transfer_ratelanesN)r r rrcrhrir\rrrrkrks- "45(%==>@HHHrrkc0eZdZdefdefdefdefdefgZdS)rsmi_process_info_t process_idpasid vram_usage sdma_usage cu_occupancyN)r r rrhrZr\rrrroros=x((#x(x(* ,HHHrroceZdZdZdZdZdS)rsmi_xgmi_status_trr(r)N)r r rRSMI_XGMI_STATUS_NO_ERRORSRSMI_XGMI_STATUS_ERROR RSMI_XGMI_STATUS_MULTIPLE_ERRORSrrrrvrvs"!"'($$$rrvcZeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdS)rsmi_io_link_typerr(r)r*r+r,r-r.r/r0r1r2r3r4r5r6N)r r rRSMI_IOLINK_TYPE_UNDEFINEDRSMI_IOLINK_TYPE_HYPERTRANSPORTRSMI_IOLINK_TYPE_PCIEXPRESSRSMI_IOLINK_TYPE_AMBARSMI_IOLINK_TYPE_MIPIRSMI_IOLINK_TYPE_QPI_1_1RSMI_IOLINK_TYPE_RESERVED1RSMI_IOLINK_TYPE_RESERVED2RSMI_IOLINK_TYPE_RAPID_IORSMI_IOLINK_TYPE_INFINIBANDRSMI_IOLINK_TYPE_RESERVED3RSMI_IOLINK_TYPE_XGMIRSMI_IOLINK_TYPE_XGOPRSMI_IOLINK_TYPE_GZRSMI_IOLINK_TYPE_ETHERNET_RDMARSMI_IOLINK_TYPE_RDMA_OTHERRSMI_IOLINK_TYPE_OTHERRSMI_IOLINK_TYPE_NUMIOLINKTYPESRSMI_IOLINK_TYPE_SIZErrrr{r{s&'&'#&'&'&'&'&'&'&'&'&(&(&(&(&("&(&(&(#&0rr{ct|tvr t|St tdkrt t t|t|<t|tS#t$rtwxYw#twxYw)N) _rocml_get_function_ptr_cache lib_load_lockacquirerocm_librgetattrreleaseAttributeErrorr)names r_rocml_get_function_ptrrs ,,,,T22  t  * * .29(D2I2I )$ /06   . . .- - . sB)BBBB7ctdkrt tdkr tjdddkrt dt }t|t|an#t$rtdwxYwtdkrtdt dS#t wxYwdS)z'Load ROCm library if not already loadedNr*winz%Windows platform is not supported yetzROCm library not found) rrrsysplatformr _find_lib_rocmcdll LoadLibraryCDLLOSErrorrr) path_librocms r_load_rocm_libraryrs 4 $4 O|BQB'50056]^^^(6'7'7 ((666#' #5#5OOO45MNNNOt##45MNNN  ! ! # # # # #M ! ! # # # #'s# C#ABC#B++C##C>ctjdd}t|dt}t |r|ndS)zRsearch for librocm and returns path if search fails, returns empty string ROCM_PATHz /opt/rocmzlib/)osenvirongetr LIBROCM_NAMEr) rocm_path rocm_lib_paths rrrsH {K88I$9<$9$9::M"=11 9==r9rcd} ttjddtj}n#tj$rYnwxYwt |dkS)zH Returns true if amdgpu is found in the list of initialized modules rz+cat /sys/module/amdgpu/initstate |grep liveT)shellstderrr)str subprocess check_outputDEVNULLCalledProcessErrorlen) initializeds r_driver_initializedrs~K *12_gkt~uGHHHII  (     {  a s.3AAc^ttrEtd}|dkr$t jd|t dnt dttdz a t dS)zInitialize ROCm binding of SMIrz ROCm SMI init returned value: %szROCm SMI initialization failedz ROCm driver initilization failedr(N) rrr rsmi_initloggingdebug RuntimeErrorrr_rocm_lib_refcountr)ret_inits rsmi_initializers?%%a(( q== M>>!rc|tjkrdt}t|t |t jd|j dSdS)z Returns true if RSMI call status is 0 (success) @param device: DRM device identifier @param my_ret: Return of RSMI call (rocm_smi_lib API) @param metric: Parameter of GPU currently being analyzed zROCm RSMI error: %sFT) r'r7c_char_prrsmi_status_stringbyrefrrvaluedecode)my_reterr_strs r rsmi_ret_okr%sb222**##FE'NN;;; +W]-A-A-C-CDDDu 4rctttt dzatdS)z4leave the library loaded, but shutdown the interfacer(N)rrrsmi_shut_downrrrrrrr smi_shutdownr4sS''))***!rctd}ttj|d}t |r|jndS)z#returns ROCm kernerl driver versionr$r)create_string_bufferrrsmi_version_str_getr^r`rrr)ver_strrets rsmi_get_kernel_versionr?sR"3''G  ' '(;(OQXZ] ^ ^C%0%5%5 =7=   ! ! !2=rct}t|t|}t |r|jndS)z0returns device id of the device as 64bit integer)rZrrsmi_dev_id_getrrrdevuidrs rsmi_get_device_idrEs? **C  " "3c 3 3C#C(( 0399b0rctd}tt|}t |r|jndS)zreturns a list of GPU devices rr)rhrrsmi_num_monitor_devicesrrr) num_devicers rsmi_get_device_countrKsC!J  + +E*,=,= > >C*3// 7:  R7rctt}t||t}t |r|jndS)z returns the name of a GPU devicer)rRSMI_MAX_BUFFER_LENGTHrrsmi_dev_name_getrrr)rseriesrs rsmi_get_device_namerRsN !"8 9 9F  $ $S&2H I IC$/$4$4 <6<   "z2smi_get_device_compute_process..s HHHA ! 'HHHr)rhrrsmi_compute_process_info_getrrrrorange) num_procsrbuff_szret2rKs @rsmi_get_device_compute_processrRs I  0 0uY7G7G H HC3/B&(7255 55eI6F6FiHXHXYYLWX\L]L]eHHHHy1G1GHHHHcee r device_idproc_idsreturncg}|D]]}t}t||t|}t |r||^|S)a&Returns list of process info running compute on the specified device by process IDs. Args: device_id: The device index to query proc_ids: List of process IDs to get info for Returns: List of process info structures for the specified device and process IDs )ror'rsmi_compute_process_info_by_device_getrrappend)rSrT proc_infosproc_idrKrs r&smi_get_compute_process_info_by_devicer[ soJ))')) >>w SXYbScScdd s   )   i ( ( ( rct}t|dt|}t |r |jdzndS)z&returns average power of device_id devrgư>r)rhrrsmi_dev_power_ave_getrrr)rpowerrs rsmi_get_device_average_powerr_sG JJE  ) )#q%,, ? ?C!,S!1!1 95;  r9rct}t|t|}t |r|jndS)z&returns XGMI error status for a devicer)rvrrsmi_dev_xgmi_error_statusrrr)rrYrs r smi_get_device_xgmi_error_statusrb'sA  ! !F  - -c5== A AC&s++ 36<<3rcTt|}t|S)z%resets XGMI error status for a device)rrsmi_dev_xgmi_error_resetrr$s rsmi_reset_device_xgmi_errorre.s#  , ,S 1 1C s  rct}t|t|}t |r|jndS)z!returns XGMI hive ID for a devicer)rZrrsmi_dev_xgmi_hive_id_getrrr)rhive_idrs rsmi_get_device_xgmi_hive_idri4s?jjG  , ,S%.. A AC',, 47=="4r)rR)ectypesos.pathrrrrrrr threadingenumrr Exceptionr rrrrrrrrrir[r'r7r8r9r:r;r<r=r>r@rArBrCrDrErFrGrsmi_status_verbose_err_outrIrLr StructurerVr^rcrkrorvr{rLockrrdictrrrrrrrrrrrrrrrrrrrrrrrrr!r%r(r+r.r1r7r=r9rFrRintlistr[r_rbrerirrrrus/0**********        i        )                                y    5 5 5 5 5 5 5 5" +++++E+++*%'A*,H+-P(*D  (*=  .0\ 02T13e+-X13h')J/1S')N-/O%'J +-E!(""""""""++++++++,++ ######## ,,,,,%,,, DDDDDDDD @@@@@I@@@ ,,,,,),,,)))))))) 11111111.    !%   *$$$4:::   &   >>> 111 888===111 :::22223333:::BBB111333777TTT555@@@ @@@ 777444RRREEE888   cTd$:::444 55555r