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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: habana-pyhlml
-Version: 1.18.0.524
+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:
@@ -202,7 +218,13 @@ class c_hlml_pci_info(_PrintS):
                ]
 
 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

@@ -228,7 +228,7 @@ def hlmlDeviceGetClockLimitInfo(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 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.
@@ -242,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
@@ -343,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:

habana_pyhlml-1.18.0.524.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=agWxLywPOHqmkWuUJqFd5raRtPNG5qvqOorbiXE90bw,10738
-pyhlml/main.py,sha256=TifTYyaDoLiBs8qZESFpnVdFt9QB4kgRS-9uMBOWqPo,37907
-habana_pyhlml-1.18.0.524.dist-info/LICENSE,sha256=_J8fTEtZF8y69onNU5_EAdOn0AC9RM2uHPh_q36-5bc,1065
-habana_pyhlml-1.18.0.524.dist-info/METADATA,sha256=CrVsiG5Ax8x6nNXhtRbm0hUu_dsSVf-7Q38q_FAjWsE,1457
-habana_pyhlml-1.18.0.524.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-habana_pyhlml-1.18.0.524.dist-info/top_level.txt,sha256=0jXJHTTotoeuyzpI1dcVr0mzCVB0NKY7LyCF4OwoMZU,7
-habana_pyhlml-1.18.0.524.dist-info/RECORD,,