From aa654175f45feec7aece1c307c856d4f486c4c47 Mon Sep 17 00:00:00 2001 From: Maisam Arif Date: Wed, 13 Dec 2023 23:52:54 -0600 Subject: [PATCH] Fix imports for partition API's Change-Id: Ic3bc0230405ee5e662bfd2d5c6d0ed5bca42a671 [ROCm/amdsmi commit: e9a615383622eafd6795ba5209788ca0186b08d0] --- projects/amdsmi/py-interface/__init__.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/projects/amdsmi/py-interface/__init__.py b/projects/amdsmi/py-interface/__init__.py index 4d98d7fddd..2c54cfff33 100644 --- a/projects/amdsmi/py-interface/__init__.py +++ b/projects/amdsmi/py-interface/__init__.py @@ -194,6 +194,14 @@ from .amdsmi_interface import amdsmi_topo_get_link_type from .amdsmi_interface import amdsmi_is_P2P_accessible from .amdsmi_interface import amdsmi_get_xgmi_info +# # Partition Functions +from .amdsmi_interface import amdsmi_get_gpu_compute_partition +from .amdsmi_interface import amdsmi_set_gpu_compute_partition +from .amdsmi_interface import amdsmi_reset_gpu_compute_partition +from .amdsmi_interface import amdsmi_get_gpu_memory_partition +from .amdsmi_interface import amdsmi_set_gpu_memory_partition +from .amdsmi_interface import amdsmi_reset_gpu_memory_partition + # # Individual GPU Metrics Functions from .amdsmi_interface import amdsmi_get_gpu_metrics_temp_hotspot from .amdsmi_interface import amdsmi_get_gpu_metrics_temp_mem @@ -263,6 +271,8 @@ from .amdsmi_interface import AmdSmiEvtNotificationType from .amdsmi_interface import AmdSmiTemperatureMetric from .amdsmi_interface import AmdSmiVoltageMetric from .amdsmi_interface import AmdSmiVoltageType +from .amdsmi_interface import AmdSmiComputePartitionType +from .amdsmi_interface import AmdSmiMemoryPartitionType from .amdsmi_interface import AmdSmiPowerProfilePresetMasks from .amdsmi_interface import AmdSmiGpuBlock from .amdsmi_interface import AmdSmiRasErrState