AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added amd-smi partition as preliminary command.

Browse Source 
new command includes following arguments:
  - current - display the current partition information for the selected
    gpu(s)
  - memory - display memory partition information for the selected
    gpu(s)
  - accelerator - display accelerator partition information for the
    selected gpu(s)
additional functionality will be added as more partition APIs are added.

Signed-off-by: gabrpham <Gabriel.Pham@amd.com>
Change-Id: Ica86160139002ef5213d6d4b0e390670aeef01c8
This commit is contained in:
gabrpham
2024-09-25 22:44:38 -05:00
parent 2c8e2060cb
commit 4e2fc2d604
6 changed files with 294 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
amdsmi_cli/amdsmi_cli.py
+2 -2
View File
@@ -94,7 +94,8 @@ if __name__ == "__main__":
amd_smi_commands.reset,
amd_smi_commands.monitor,
amd_smi_commands.rocm_smi,
amd_smi_commands.xgmi)
amd_smi_commands.xgmi,
amd_smi_commands.partition)
try:
try:
argcomplete.autocomplete(amd_smi_parser)
@@ -128,7 +129,6 @@ if __name__ == "__main__":
sys.tracebacklimit = 10
else:
sys.tracebacklimit = -1
# Execute subcommands
args.func(args)
except amdsmi_cli_exceptions.AmdSmiException as e: