Maisam Arif
fab2184e71
Signed-off-by: Maisam Arif <maisarif@amd.com> Change-Id: Ib712a182e839343472f4c6fd87a34d045b63f619
93 Zeilen
4.2 KiB
Python
Ausführbare Datei
93 Zeilen
4.2 KiB
Python
Ausführbare Datei
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (C) 2023 Advanced Micro Devices. All rights reserved.
|
|
#
|
|
# 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.
|
|
#
|
|
|
|
import logging
|
|
import sys
|
|
|
|
from amdsmi_commands import AMDSMICommands
|
|
from amdsmi_parser import AMDSMIParser
|
|
from amdsmi_logger import AMDSMILogger
|
|
import amdsmi_cli_exceptions
|
|
from amdsmi import amdsmi_interface
|
|
from amdsmi import amdsmi_exception
|
|
|
|
|
|
def _print_error(e, destination):
|
|
if destination in ['stdout', 'json', 'csv']:
|
|
print(e)
|
|
else:
|
|
f = open(destination, "w")
|
|
f.write(e)
|
|
f.close()
|
|
print("Error occured. Result written to " + str(destination) + " file")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Set compatability mode based on which cli mapping user selects
|
|
if 'gpuv-smi' in sys.argv[0]:
|
|
compatibility = AMDSMILogger.LoggerCompatibility.gpuvsmi.value
|
|
elif 'rocm-smi' in sys.argv[0]:
|
|
compatibility = AMDSMILogger.LoggerCompatibility.rocmsmi.value
|
|
else:
|
|
compatibility = AMDSMILogger.LoggerCompatibility.amdsmi.value
|
|
|
|
amd_smi_commands = AMDSMICommands(compatibility=compatibility)
|
|
amd_smi_parser = AMDSMIParser(amd_smi_commands.version,
|
|
amd_smi_commands.list,
|
|
amd_smi_commands.static,
|
|
amd_smi_commands.firmware,
|
|
amd_smi_commands.bad_pages,
|
|
amd_smi_commands.metric,
|
|
amd_smi_commands.process,
|
|
amd_smi_commands.profile,
|
|
amd_smi_commands.event,
|
|
amd_smi_commands.topology,
|
|
amd_smi_commands.set_value,
|
|
amd_smi_commands.reset,
|
|
amd_smi_commands.rocm_smi)
|
|
try:
|
|
args = amd_smi_parser.parse_args(args=None if sys.argv[1:] else ['--help'])
|
|
|
|
# Handle command modifiers before subcommand execution
|
|
if args.json:
|
|
amd_smi_commands.logger.format = amd_smi_commands.logger.LoggerFormat.json.value
|
|
if args.csv:
|
|
amd_smi_commands.logger.format = amd_smi_commands.logger.LoggerFormat.csv.value
|
|
if args.file:
|
|
amd_smi_commands.logger.destination = args.file
|
|
if args.loglevel:
|
|
logging_dict = {'DEBUG' : logging.DEBUG,
|
|
'INFO' : logging.INFO,
|
|
'WARNING': logging.WARNING,
|
|
'ERROR': logging.ERROR,
|
|
'CRITICAL': logging.CRITICAL}
|
|
# Enable debug logs on amdsmi library ie. RSMI_LOGGING = 1 in environment or otherwise
|
|
logging.basicConfig(format='%(levelname)s: %(message)s', level=logging_dict[args.loglevel])
|
|
|
|
# Execute subcommands
|
|
args.func(args)
|
|
except amdsmi_cli_exceptions.AmdSmiException as e:
|
|
_print_error(str(e), amd_smi_commands.logger.destination)
|
|
except amdsmi_exception.AmdSmiLibraryException as e:
|
|
exc = amdsmi_cli_exceptions.AmdSmiAMDSMIErrorException(amd_smi_commands.logger.format, e.get_error_code())
|
|
_print_error(str(exc), amd_smi_commands.logger.destination)
|