habana-pyhlml 1.17.1.40__py3-none-any.whl → 1.19.0.561__py3-none-any.whl

{habana_pyhlml-1.17.1.40.dist-info → habana_pyhlml-1.19.0.561.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: habana-pyhlml
-Version: 1.17.1.40
+Version: 1.19.0.561
 Summary: Python3 wrapper for the HLML library.
 Home-page: UNKNOWN
 Author: HabanaAI

habana_pyhlml-1.19.0.561.dist-info/RECORD

@@ -0,0 +1,10 @@
+pyhlml/__init__.py,sha256=VjbHqSSypSMmVK7UTxuCyOs_AE8HhTXX9EGr6NbaQNk,25
+pyhlml/hlml_error.py,sha256=GQulGXf-jX38jEASXF22qJ0yL8eC57vUo8s9lVI7q_E,2113
+pyhlml/hlml_lib.py,sha256=JiHRLl3vpAozhCSs9eJC8xlhIMxR2wNVZdmahjlgDfc,1430
+pyhlml/hlml_types.py,sha256=AwehWbTpq3D3fAi3U1_hZg88sljr7c4pldoXOo8Yv5Q,11677
+pyhlml/main.py,sha256=U-4PQKZ0RWFVnDWoJrDJ_TBOOhMqtTfshXi7CbR72M8,39118
+habana_pyhlml-1.19.0.561.dist-info/LICENSE,sha256=_J8fTEtZF8y69onNU5_EAdOn0AC9RM2uHPh_q36-5bc,1065
+habana_pyhlml-1.19.0.561.dist-info/METADATA,sha256=eMNrpQ6Do4ZmTt7PBjDX6Guvd1aOtLEHivRJJb1ifDI,1457
+habana_pyhlml-1.19.0.561.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+habana_pyhlml-1.19.0.561.dist-info/top_level.txt,sha256=0jXJHTTotoeuyzpI1dcVr0mzCVB0NKY7LyCF4OwoMZU,7
+habana_pyhlml-1.19.0.561.dist-info/RECORD,,

pyhlml/hlml_types.py

@@ -8,7 +8,7 @@ class HLML_DEVICE:
     TYPE                                    = ctypes.c_void_p
 
 class HLML_DEFINE:
-    PCI_DOMAIN_LEN                          = 5
+    PCI_DOMAIN_LEN                          = 9
     PCI_ADDR_LEN                            = ( PCI_DOMAIN_LEN + 10 )
     PCI_LINK_INFO_LEN                       = 10
     HL_FIELD_MAX_SIZE                       = 32
@@ -88,6 +88,22 @@ class HLML_ENABLE_STATE:
 class HLML_P_STATES:
     TYPE                                    = ctypes.c_uint
     HLML_PSTATE_0                           = 0
+    HLML_PSTATE_1                           = 1
+    HLML_PSTATE_2                           = 2
+    HLML_PSTATE_3                           = 3
+    HLML_PSTATE_4                           = 4
+    HLML_PSTATE_5                           = 5
+    HLML_PSTATE_6                           = 6
+    HLML_PSTATE_7                           = 7
+    HLML_PSTATE_8                           = 8
+    HLML_PSTATE_9                           = 9
+    HLML_PSTATE_10                          = 10
+    HLML_PSTATE_11                          = 11
+    HLML_PSTATE_12                          = 12
+    HLML_PSTATE_13                          = 13
+    HLML_PSTATE_14                          = 14
+    HLML_PSTATE_15                          = 15
+    HLML_PSTATE_SENTINEL                    = 16
     HLML_PSTATE_UNKNOWN                     = 32
 
 class HLML_MEMORY_ERROR:
@@ -188,6 +204,7 @@ class c_hlml_pci_info(_PrintS):
         * device - The device's id on the bus, 0 to 31
         * domain - The PCI domain on which the device's bus resides
         * pci_device_id - The combined 16b deviceId and 16b vendor id
+        * pci_subsys_device_id - The combined 16b subsys_id and 16b subsys_vendor_id
     */
     """
     _fields_ = [("bus", ctypes.c_uint),
@@ -196,11 +213,18 @@ class c_hlml_pci_info(_PrintS):
                 ("domain", ctypes.c_uint),
                 ("pci_device_id", ctypes.c_uint),
                 ("caps", c_hlml_pci_cap),
-                ("pci_rev", ctypes.c_uint)
+                ("pci_rev", ctypes.c_uint),
+                ("pci_subsys_id", ctypes.c_uint)
                ]
 
 class c_hlml_utilization(_PrintS):
-    _fields_ = [("aip", ctypes.c_uint)]
+    _fields_ = [("aip", ctypes.c_uint),
+                ("memory", ctypes.c_uint)
+               ]
+
+class c_hlml_process_utilization(_PrintS):
+    _fields_ = [("aip_util", ctypes.c_uint)
+               ]
 
 class c_hlml_memory(_PrintS):
     _fields_ = [("free", ctypes.c_ulonglong),

pyhlml/main.py

@@ -211,7 +211,24 @@ def hlmlDeviceGetMaxClockInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 ) ->
     check_return(ret)
     return speed.value
 
-def hlmlDeviceGetUtilizationRates(device: hlml_t.HLML_DEVICE.TYPE) -> int:
+def hlmlDeviceGetClockLimitInfo(device: hlml_t.HLML_DEVICE.TYPE, clock_type=0 ) -> int:
+    """ Retrives the frequency limit of the selected clock (MHz)
+        Parameters:
+            device (HLML_DEVICE.TYPE) - The handle for a habana device.
+            clock_type ( 0-SOC ( GAUDI ) / 1-IC ( GOYA ) / 2-MME ( GOYA ) / 3-TPC ( GOYA ) )
+        Returns:
+            speed (int) - The frequency LIMIT of the selected clock in MHz.
+    """
+    global _hlmlOBJ
+    speed = ctypes.c_uint()
+
+    fn = _hlmlOBJ.get_func_ptr("hlml_device_get_clock_limit_info")
+    ret = fn(device, clock_type, ctypes.byref(speed))
+
+    check_return(ret)
+    return speed.value
+
+def hlmlDeviceGetUtilizationRates(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_utilization:
     """ Returns utilization over the past second as a percentage
         Parameters:
             device (HLML_DEVICE.TYPE) - The handle for a habana device.
@@ -225,7 +242,23 @@ def hlmlDeviceGetUtilizationRates(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     ret = fn(device, ctypes.byref(hlml_util))
 
     check_return(ret)
-    return hlml_util.aip
+    return hlml_util
+
+def hlmlDeviceGetProcessUtilization(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_process_utilization:
+    """ Returns process utilization over the past second as a percentage
+        Parameters:
+            device (HLML_DEVICE.TYPE) - The handle for a habana device.
+        Returns:
+            hlml_util (int) - The utilization rate over the last second as a percentage.
+    """
+    global _hlmlOBJ
+    hlml_util = hlml_t.c_hlml_process_utilization()
+
+    fn = _hlmlOBJ.get_func_ptr("hlml_device_get_process_utilization")
+    ret = fn(device, ctypes.byref(hlml_util))
+
+    check_return(ret)
+    return hlml_util
 
 def hlmlDeviceGetMemoryInfo(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.c_hlml_memory:
     """ Returns the total, used, and free memory in bytes
@@ -326,6 +359,22 @@ def hlmlDeviceGetPowerUsage(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     check_return(ret)
     return power.value
 
+def hlmlDeviceSetPowerManagementLimit(device: hlml_t.HLML_DEVICE.TYPE, limit : ctypes.c_uint) -> None:
+    """ Sets power management limit on this device in mW
+        Parameters:
+            device (HLML_DEVICE.TYPE) - The handle for a habana device.
+            limit (ctypes.c_uint) - The power limit on the device in mW.
+        Returns:
+            ret (HLML_RETURN) - The status of the operation.
+    """
+    global _hlmlOBJ
+
+    fn = _hlmlOBJ.get_func_ptr("hlml_device_set_power_management_limit")
+    ret = fn(device, limit)
+
+    check_return(ret)
+    return None
+
 def hlmlDeviceGetPowerManagementDefaultLimit(device: hlml_t.HLML_DEVICE.TYPE) -> int:
     """ Retrieves default power management limit on this device in mW
         Parameters:
@@ -1063,3 +1112,34 @@ def hlmlDeviceGetOperationStatus(device: hlml_t.HLML_DEVICE.TYPE) -> str:
 
     check_return(ret)
     return sts_str.value
+
+def hlmlDeviceGetPowerManagementMode(device: hlml_t.HLML_DEVICE.TYPE) -> hlml_t.HLML_ENABLE_STATE:
+    """ Acquire name of device from handle
+        Parameters:
+            device (HLML_DEVICE.TYPE) - The handle for a habana device.
+        Returns:
+            mode (int) - The power management mode of the device.
+    """
+    global _hlmlOBJ
+    mode = ctypes.c_uint()
+    fn = _hlmlOBJ.get_func_ptr("hlml_device_get_power_management_mode")
+    ret = fn(device, ctypes.byref(mode))
+
+    check_return(ret)
+    return mode.value
+
+def hlmlDeviceGetPowerManagementLimit(device: hlml_t.HLML_DEVICE.TYPE) -> int:
+    """ Returns the pending status of replaced rows.
+        Parameters:
+            device (HLML_DEVICE.TYPE) - The handle for a habana device.
+        Returns:
+            limit (int) - Reference in which to return the power management
+                          limit in mW
+    """
+    global _hlmlOBJ
+    power_max = ctypes.c_uint()
+    fn = _hlmlOBJ.get_func_ptr("hlml_device_get_power_management_limit")
+    ret = fn(device, ctypes.byref(power_max))
+
+    check_return(ret)
+    return power_max.value

habana_pyhlml-1.17.1.40.dist-info/RECORD

@@ -1,10 +0,0 @@
-pyhlml/__init__.py,sha256=VjbHqSSypSMmVK7UTxuCyOs_AE8HhTXX9EGr6NbaQNk,25
-pyhlml/hlml_error.py,sha256=GQulGXf-jX38jEASXF22qJ0yL8eC57vUo8s9lVI7q_E,2113
-pyhlml/hlml_lib.py,sha256=JiHRLl3vpAozhCSs9eJC8xlhIMxR2wNVZdmahjlgDfc,1430
-pyhlml/hlml_types.py,sha256=znPABbbwxHfSRiUO4lepgnJumVweRx3AWoU9R84aTss,10603
-pyhlml/main.py,sha256=r9wnlzK9qVY2RdxoSpGxbnYyQbS1HKijPkHjogeEEwY,36135
-habana_pyhlml-1.17.1.40.dist-info/LICENSE,sha256=_J8fTEtZF8y69onNU5_EAdOn0AC9RM2uHPh_q36-5bc,1065
-habana_pyhlml-1.17.1.40.dist-info/METADATA,sha256=vXftSslewDYDHkse6mu7xD3G4aiWD8N5TDqnUI6bkOw,1456
-habana_pyhlml-1.17.1.40.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-habana_pyhlml-1.17.1.40.dist-info/top_level.txt,sha256=0jXJHTTotoeuyzpI1dcVr0mzCVB0NKY7LyCF4OwoMZU,7
-habana_pyhlml-1.17.1.40.dist-info/RECORD,,