-
Notifications
You must be signed in to change notification settings - Fork 127
feature: Add support for sysman python bindings #420
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
Merged
Changes from 2 commits
Commits
Show all changes
14 commits
Select commit
Hold shift + click to select a range
82d04e5
feature: Add support for sysman python bindings
shubskmr 3ef4c34
feature: Add support for sysman python bindings
shubskmr 9086388
feature: Add support for sysman python bindings
shubskmr 244e5f5
feature: Add support for sysman python bindings
shubskmr 9be93d5
Merge branch 'master' into python_bindings
shubskmr 37fca5a
Merge branch 'master' into python_bindings
shubskmr 51bc498
feature: Add support for sysman python bindings
shubskmr 3faa738
feature: Add support for sysman python bindings
shubskmr aede7a8
fix: Correct awk comparison syntax in coverage summary step
shubskmr 5a15047
fix: Correct awk comparison syntax in coverage summary step
shubskmr f430f1a
fix: Correct awk comparison syntax in coverage summary step
shubskmr bb0935a
fix: Correct awk comparison syntax in coverage summary step
shubskmr 6208ada
fix: Correct awk comparison syntax in coverage summary step
shubskmr 594c78a
feature: Add support for sysman python bindings
shubskmr File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,55 @@ | ||
| # Python cache files | ||
| __pycache__/ | ||
| *.py[cod] | ||
| *$py.class | ||
|
|
||
| # Coverage reports | ||
| .coverage | ||
| .coverage.* | ||
| htmlcov/ | ||
| .tox/ | ||
| .pytest_cache/ | ||
|
|
||
| # Distribution / packaging | ||
| .Python | ||
| build/ | ||
| develop-eggs/ | ||
| dist/ | ||
| downloads/ | ||
| eggs/ | ||
| .eggs/ | ||
| lib/ | ||
| lib64/ | ||
| parts/ | ||
| sdist/ | ||
| var/ | ||
| wheels/ | ||
| *.egg-info/ | ||
| .installed.cfg | ||
| *.egg | ||
| MANIFEST | ||
|
|
||
| # Virtual environments | ||
| .env | ||
| .venv | ||
| env/ | ||
| venv/ | ||
| ENV/ | ||
| env.bak/ | ||
| venv.bak/ | ||
|
|
||
| # IDE | ||
| .vscode/ | ||
| .idea/ | ||
| *.swp | ||
| *.swo | ||
| *~ | ||
|
|
||
| # OS | ||
| .DS_Store | ||
| .DS_Store? | ||
| ._* | ||
| .Spotlight-V100 | ||
| .Trashes | ||
| ehthumbs.db | ||
| Thumbs.db |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,9 @@ | ||
| MIT License | ||
|
|
||
| Copyright (C) 2025 Intel Corporation | ||
|
|
||
| Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: | ||
|
|
||
| The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. | ||
|
|
||
| THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,208 @@ | ||
| # drivers.gpu.compute.pyzes | ||
| pyzes | ||
| ====== | ||
|
|
||
| Python bindings to the Intel Level-Zero-Driver Library | ||
| ------------------------------------------------ | ||
|
|
||
| Provides a Python interface to GPU management and monitoring functions. | ||
|
|
||
| This is a wrapper around the Level-Zero-Driver library. | ||
| For information about the Level-Zero-Driver library, see the spec document | ||
| https://oneapi-src.github.io/level-zero-spec/level-zero/latest/index.html | ||
|
|
||
| Download the latest package from: https://github.com/oneapi-src/level-zero | ||
|
|
||
| The level-zero header file contains function documentation that is relevant | ||
| to this wrapper. The header file is distributed with driver (ze_api.h and zes_api.h) | ||
|
|
||
| This module does not handles allocating structs before returning the desired value. | ||
| Non-success codes are returned and respective error is printed. | ||
|
|
||
| REQUIREMENTS | ||
| ------------ | ||
| - **Python 3.10** (required) | ||
| - ctypes module (included in standard Python library) | ||
| - Level Zero driver installed on the system | ||
|
|
||
| INSTALLATION | ||
| ------------ | ||
| ```bash | ||
| # Ensure you have Python 3.10 installed | ||
| python3.10 --version | ||
|
|
||
| # Install the package (when available) | ||
| pip install pyzes | ||
| ``` | ||
|
|
||
| USAGE | ||
| ----- | ||
| ``` | ||
| >>> from pyzes import * | ||
| >>> rc = zesInit(0) | ||
| >>> driver_count = c_uint32(0) | ||
| >>> rc = pyzes.zesDriverGet(byref(driver_count), None) | ||
| >>> print(f"Driver Count: {driver_count.value}") | ||
|
|
||
| ``` | ||
|
|
||
| ## C Structure and its python module class ## | ||
| struct zes_process_state_t { | ||
|
|
||
| zes_structure_type_t stype | ||
| [in] type of this structure | ||
|
|
||
| const void *pNext | ||
| [in][optional] must be null or a pointer to an extension-specific structure (i.e. contains stype and pNext). | ||
|
|
||
| uint32_t processId | ||
| [out] Host OS process ID. | ||
|
|
||
| uint64_t memSize | ||
| [out] Device memory size in bytes allocated by this process (may not necessarily be resident on the device at the time of reading). | ||
|
|
||
| uint64_t sharedSize | ||
| [out] The size of shared device memory mapped into this process (may not necessarily be resident on the device at the time of reading). | ||
|
|
||
| zes_engine_type_flags_t engines | ||
| [out] Bitfield of accelerator engine types being used by this process. | ||
| } | ||
|
|
||
| Python Class | ||
| class zes_process_state_t(Structure): | ||
| _fields_ = [ | ||
| ("pid", c_uint32), | ||
| ("command", c_char * ZES_STRING_PROPERTY_SIZE), | ||
| ("memSize", c_uint64), # in bytes | ||
| ("sharedMemSize", c_uint64),# in bytes | ||
| ("engineType", zes_engine_type_flags_t), | ||
| ("subdeviceId", c_uint32), | ||
| ] | ||
|
|
||
|
|
||
| FUNCTIONS | ||
| --------- | ||
| Python methods wrap Level-Zero-Driver functions, implemented in a C shared library. | ||
| Each function's use is the same: | ||
| - C function output parameters are filled in with values, and return codes are returned. | ||
|
|
||
| ``` | ||
| ze_result_t zesDeviceGetProperties( | ||
| zes_device_handle_t hDevice, | ||
| zes_device_properties_t* pProperties); | ||
|
|
||
| >>> props = zes_device_properties_t() | ||
| >>> props.stype = ZES_STRUCTURE_TYPE_DEVICE_PROPERTIES | ||
| >>> props.pNext = None | ||
| >>> pyzes.zesDeviceGetProperties(devices[i], byref(props)) | ||
|
|
||
| ``` | ||
|
|
||
| - C structs are converted into Python classes. | ||
| ``` | ||
| // C Function and typedef struct | ||
|
|
||
| ze_result_t zesDeviceGetProperties( | ||
| zes_device_handle_t hDevice, | ||
| zes_device_properties_t* pProperties); | ||
|
|
||
| typedef struct _zes_device_properties_t | ||
| { | ||
| zes_structure_type_t stype; | ||
| void* pNext; | ||
| ze_device_properties_t core; | ||
| uint32_t numSubdevices; | ||
| char serialNumber[ZES_STRING_PROPERTY_SIZE]; | ||
| char boardNumber[ZES_STRING_PROPERTY_SIZE]; | ||
| char brandName[ZES_STRING_PROPERTY_SIZE]; | ||
| char modelName[ZES_STRING_PROPERTY_SIZE]; | ||
| char vendorName[ZES_STRING_PROPERTY_SIZE]; | ||
| char driverVersion[ZES_STRING_PROPERTY_SIZE] | ||
| } zes_device_properties_t; | ||
|
|
||
| >>>print(f"numSubdevices: {props.numSubdevices}") | ||
| >>>print(f"serialNumber: {props.serialNumber}") | ||
| >>>print(f"boardNumber: {props.boardNumber}") | ||
| >>>print(f"brandName: {props.brandName}") | ||
| >>>print(f"modelName: {props.modelName}") | ||
| >>>print(f"driverVersion: {props.driverVersion}") | ||
| >>>print(f"coreClockMHz: {props.core.coreClockRate}") | ||
| ``` | ||
|
|
||
| HOW TO USE STRUCTURE CHAINING | ||
| ``` | ||
| >>> props = zes_device_properties_t() | ||
| >>> props.stype = ZES_STRUCTURE_TYPE_DEVICE_PROPERTIES | ||
| >>> ext = zes_device_ext_properties_t() | ||
| >>> ext.stype = ZES_STRUCTURE_TYPE_DEVICE_EXT_PROPERTIES | ||
| >>> ext.pNext = None | ||
| >>> base.pNext = cast(pointer(ext), c_void_p) | ||
| >>> pyzes.zesDeviceGetProperties(devices[i], byref(props)) | ||
| >>> print(f"Extension properties flags: {ext.flags}") | ||
| ``` | ||
|
|
||
| For more information see the Level-Zero-Driver documentation. | ||
|
|
||
| VARIABLES | ||
| --------- | ||
| All meaningful constants and enums are exposed in Python module. | ||
|
|
||
| SUPPORTED APIs | ||
| -------------- | ||
|
|
||
| | API Function | Module | Since Version | Limitations | | ||
| |--------------|--------|---------------|-------------| | ||
| | `zesInit` | Device | 0.0.0 | None | | ||
| | `zesDriverGet` | Device | 0.0.0 | None | | ||
| | `zesDeviceGet` | Device | 0.0.0 | None | | ||
| | `zesDeviceGetProperties` | Device | 0.0.0 | None | | ||
| | `zesDriverGetDeviceByUuidExp` | Device | 0.0.0 | Experimental API | | ||
| | `zesDeviceProcessesGetState` | Device | 0.0.0 | None | | ||
| | **Memory Management** |-|-|-| | ||
| | `zesDeviceEnumMemoryModules` | Memory | 0.0.0 | None | | ||
| | `zesMemoryGetProperties` | Memory | 0.0.0 | None | | ||
| | `zesMemoryGetState` | Memory | 0.0.0 | None | | ||
| | `zesMemoryGetBandwidth` | Memory | 0.0.0 | Linux: Requires superuser or read permissions for telem nodes | | ||
| | **Power Management** |-|-|-| | ||
| | `zesDeviceEnumPowerDomains` | Power | 0.0.0 | None | | ||
| | `zesPowerGetEnergyCounter` | Power | 0.0.0 | Linux: Requires superuser or read permissions for telem nodes | | ||
| | **Frequency Management** |-|-|-| | ||
| | `zesDeviceEnumFrequencyDomains` | Frequency | 0.0.0 | None | | ||
| | `zesFrequencyGetState` | Frequency | 0.0.0 | None | | ||
| | **Temperature Monitoring** |-|-|-| | ||
| | `zesDeviceEnumTemperatureSensors` | Temperature | 0.0.0 | None | | ||
| | `zesTemperatureGetProperties` | Temperature | 0.0.0 | None | | ||
| | `zesTemperatureGetConfig` | Temperature | 0.0.0 | None | | ||
| | `zesTemperatureGetState` | Temperature | 0.0.0 | Linux: Requires superuser or read permissions for telem nodes | | ||
| | **Engine Management** |-|-|-| | ||
| | `zesDeviceEnumEngineGroups` | Engine | 0.1.0 | Linux: Shows "no handles found" error when not in superuser mode | | ||
| | `zesEngineGetProperties` | Engine | 0.1.0 | None | | ||
| | `zesEngineGetActivity` | Engine | 0.1.0 | None | | ||
|
|
||
| RELEASE NOTES | ||
| ------------- | ||
| Version 0.0.0 | ||
| - Added pyzes.py module that has python binding wrapper functions. | ||
| - Added pyzes_example.py tool as a sample app. | ||
|
|
||
| Version 0.1.0 | ||
| - Added support for engine module APIs. | ||
| - Added APIs: | ||
| zesDeviceEnumEngineGroups | ||
| zesEngineGetProperties | ||
| zesEngineGetActivity | ||
|
|
||
| Notes: | ||
| Linux: | ||
| zesPowerGetEnergyCounter | ||
| zesTemperatureGetState | ||
| zesMemoryGetBandwidth | ||
|
|
||
| The above APIs needs user to be in superuser/root mode or have read permissions for telem nodes | ||
| Telem Node Directory: /sys/class/intel_pmt/telem(1/2/3/4)/telem | ||
|
|
||
| zesDeviceEnumEngineGroups shows no handles found error when not in super user mode. | ||
|
|
||
| COPYRIGHT | ||
|
rwmcguir marked this conversation as resolved.
Outdated
|
||
| --------- | ||
| Copyright (c) 2025, INTEL Corporation. All rights reserved. | ||
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.