[SWDEV-469278] - Lowered PyYAML dependency

Signed-off-by: Maisam Arif <Maisam.Arif@amd.com>
Change-Id: Icfee09b84cf1071ec82b65fc2877be69e0283489


[ROCm/amdsmi commit: 09c9574454]
This commit is contained in:
Maisam Arif
2024-09-20 15:55:32 -05:00
parent 2aa8e8c755
commit 457619bfad
6 changed files with 31 additions and 15 deletions
+2
View File
@@ -305,6 +305,8 @@ GPU: 1
### Resolved issues ### Resolved issues
- **Improved Offline install process & lowered dependency for PyYAML**.
- **Fixed CPX not showing total number of logical GPUs**. - **Fixed CPX not showing total number of logical GPUs**.
- Updates were made to `amdsmi_init()` and `amdsmi_get_gpu_bdf_id(..)`. In order to display all logical devices, we needed a way to provide order to GPU's enumerated. This was done by adding a partition_id within the BDF optional pci_id bits. - Updates were made to `amdsmi_init()` and `amdsmi_get_gpu_bdf_id(..)`. In order to display all logical devices, we needed a way to provide order to GPU's enumerated. This was done by adding a partition_id within the BDF optional pci_id bits.
- Due to driver changes in KFD, some devices may report bits [31:28] or [2:0]. With the newly added `amdsmi_get_gpu_bdf_id(..)`, we provided this fallback to properly retreive partition ID. We - Due to driver changes in KFD, some devices may report bits [31:28] or [2:0]. With the newly added `amdsmi_get_gpu_bdf_id(..)`, we provided this fallback to properly retreive partition ID. We
+5 -6
View File
@@ -149,7 +149,7 @@ do_install_amdsmi_python_lib() {
return return
fi fi
# upgrade pip if it's an ancient version # upgrade pip if it's an ancient version, typically should not execute
# otherwise the amdsmi install will fail # otherwise the amdsmi install will fail
local pip_version local pip_version
pip_version=$(python3 -m pip --version | grep -Eo '^[^\ ]+ ([0-9]+)' | grep -Eo '[0-9]+$') pip_version=$(python3 -m pip --version | grep -Eo '^[^\ ]+ ([0-9]+)' | grep -Eo '[0-9]+$')
@@ -159,10 +159,9 @@ do_install_amdsmi_python_lib() {
fi fi
unset pip_version unset pip_version
# Check PyYAML dependency # Check PyYAML dependency, typically should not execute
local pyyaml_version pyyaml_version=$(python3 -m pip show pyyaml | grep -Po '(?<=Version: )[0-9]')
pyyaml_version=$(pip show pyyaml | grep -Po '(?<=Version: )[0-9]') if [[ "$pyyaml_version" -lt 3 ]]; then
if [[ "$pyyaml_version" -lt 5 ]]; then
echo "Detected ancient pyyaml version ($pyyaml_version)... Upgrading..." echo "Detected ancient pyyaml version ($pyyaml_version)... Upgrading..."
python3 -m pip install 'PyYAML>=5.1' --quiet --disable-pip-version-check --ignore-installed python3 -m pip install 'PyYAML>=5.1' --quiet --disable-pip-version-check --ignore-installed
fi fi
@@ -170,7 +169,7 @@ do_install_amdsmi_python_lib() {
# install python library at @CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@/amdsmi # install python library at @CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@/amdsmi
local python_lib_path=@CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@ local python_lib_path=@CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@
python3 -m pip install "$python_lib_path" --quiet --disable-pip-version-check python3 -m pip install "$python_lib_path" --quiet --disable-pip-version-check --no-build-isolation
export PIP_ROOT_USER_ACTION="$PREVIOUS_PIP_ROOT_USER_ACTION" export PIP_ROOT_USER_ACTION="$PREVIOUS_PIP_ROOT_USER_ACTION"
export PIP_BREAK_SYSTEM_PACKAGES="$PREVIOUS_PIP_BREAK_SYSTEM_PACKAGES" export PIP_BREAK_SYSTEM_PACKAGES="$PREVIOUS_PIP_BREAK_SYSTEM_PACKAGES"
+5 -5
View File
@@ -149,7 +149,7 @@ do_install_amdsmi_python_lib() {
fi fi
# upgrade pip if it's an ancient version # upgrade pip if it's an ancient version
# otherwise the amdsmi install will fail # otherwise amdsmi install will fail
local pip_version local pip_version
pip_version=$(python3 -m pip --version | grep -Eo '^[^\ ]+ ([0-9]+)' | grep -Eo '[0-9]+$') pip_version=$(python3 -m pip --version | grep -Eo '^[^\ ]+ ([0-9]+)' | grep -Eo '[0-9]+$')
if [[ "$pip_version" -lt 19 ]]; then if [[ "$pip_version" -lt 19 ]]; then
@@ -158,10 +158,10 @@ do_install_amdsmi_python_lib() {
fi fi
unset pip_version unset pip_version
# Check PyYAML dependency # Check PyYAML dependency, typically should not execute
local pyyaml_version local pyyaml_version
pyyaml_version=$(pip show pyyaml | grep -Po '(?<=Version: )[0-9]') pyyaml_version=$(python3 -m pip show pyyaml | grep -Po '(?<=Version: )[0-9]')
if [[ "$pyyaml_version" -lt 5 ]]; then if [[ "$pyyaml_version" -lt 3 ]]; then
echo "Detected ancient pyyaml version ($pyyaml_version)... Upgrading..." echo "Detected ancient pyyaml version ($pyyaml_version)... Upgrading..."
python3 -m pip install 'PyYAML>=5.1' --quiet --disable-pip-version-check --ignore-installed python3 -m pip install 'PyYAML>=5.1' --quiet --disable-pip-version-check --ignore-installed
fi fi
@@ -169,7 +169,7 @@ do_install_amdsmi_python_lib() {
# install python library at @CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@/amdsmi # install python library at @CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@/amdsmi
local python_lib_path=@CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@ local python_lib_path=@CPACK_PACKAGING_INSTALL_PREFIX@/@SHARE_INSTALL_PREFIX@
python3 -m pip install "$python_lib_path" --quiet --disable-pip-version-check python3 -m pip install "$python_lib_path" --quiet --disable-pip-version-check --no-build-isolation
export PIP_ROOT_USER_ACTION="$PREVIOUS_PIP_ROOT_USER_ACTION" export PIP_ROOT_USER_ACTION="$PREVIOUS_PIP_ROOT_USER_ACTION"
export PIP_BREAK_SYSTEM_PACKAGES="$PREVIOUS_PIP_BREAK_SYSTEM_PACKAGES" export PIP_BREAK_SYSTEM_PACKAGES="$PREVIOUS_PIP_BREAK_SYSTEM_PACKAGES"
+17 -2
View File
@@ -27,10 +27,19 @@ import time
from typing import Dict from typing import Dict
from enum import Enum from enum import Enum
import yaml import yaml
import inspect
from amdsmi_helpers import AMDSMIHelpers from amdsmi_helpers import AMDSMIHelpers
import amdsmi_cli_exceptions import amdsmi_cli_exceptions
### Custom YAML Functions
# Dumper class to preserve order of yaml.dump
class CustomDumper(yaml.Dumper):
def represent_dict_preserve_order(self, data):
return self.represent_dict(data.items())
def has_sort_keys_option(): # to check if sort_keys is available
return 'sort_keys' in inspect.signature(yaml.dump).parameters
class AMDSMILogger(): class AMDSMILogger():
def __init__(self, format='human_readable', destination='stdout') -> None: def __init__(self, format='human_readable', destination='stdout') -> None:
self.output = {} self.output = {}
@@ -202,8 +211,14 @@ class AMDSMILogger():
capitalized_json["AMDSMI_SPACING_REMOVAL"] = tabbed_dictionary capitalized_json["AMDSMI_SPACING_REMOVAL"] = tabbed_dictionary
json_string = json.dumps(capitalized_json, indent=4) json_string = json.dumps(capitalized_json, indent=4)
yaml_data = yaml.safe_load(json_string)
yaml_output = yaml.dump(yaml_data, sort_keys=False, allow_unicode=True) if has_sort_keys_option():
yaml_data = yaml.safe_load(json_string)
yaml_output = yaml.dump(yaml_data, sort_keys=False, allow_unicode=True)
else:
CustomDumper.add_representer(dict, CustomDumper.represent_dict_preserve_order)
yaml_data = yaml.safe_load(json_string)
yaml_output = yaml.dump(yaml_data, Dumper=CustomDumper, allow_unicode=True, default_flow_style=False)
# Remove a key line if it is a spacer # Remove a key line if it is a spacer
yaml_output = yaml_output.replace("AMDSMI_SPACING_REMOVAL:\n", "") yaml_output = yaml_output.replace("AMDSMI_SPACING_REMOVAL:\n", "")
@@ -16,7 +16,7 @@ readme = {file = "amdsmi/README.md", content-type = "text/markdown"}
description = "AMDSMI Python LIB - AMD GPU Monitoring Library" description = "AMDSMI Python LIB - AMD GPU Monitoring Library"
requires-python = ">=3.6" requires-python = ">=3.6"
dependencies = [ dependencies = [
"PyYAML >= 5.0", "PyYAML >= 3.12",
] ]
[project.urls] [project.urls]
+1 -1
View File
@@ -18,7 +18,7 @@ include_package_data = True
packages = find: packages = find:
python_requires = >=3.6 python_requires = >=3.6
install_requires= install_requires=
PyYAML >= 5.0 PyYAML >= 3.12
[options.package_data] [options.package_data]
* = *.so * = *.so