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

Update DB module for new sysinfo keys

Browse Source 
Signed-off-by: colramos-amd <colramos@amd.com>
This commit is contained in:
colramos-amd
2024-03-12 12:24:49 -05:00
parent 35ed3a4366
commit 706d3bb9f0
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/utils/db_connector.py
+7 -2
View File
@@ -66,8 +66,13 @@ class DatabaseConnector:
sys_info = os.path.join(self.connection_info["workload"], "sysinfo.csv") sys_info = os.path.join(self.connection_info["workload"], "sysinfo.csv")
if os.path.isfile(sys_info): if os.path.isfile(sys_info):
sys_info = pd.read_csv(sys_info) sys_info = pd.read_csv(sys_info)
soc = sys_info["name"][0] try:
name = sys_info["workload_name"][0] soc = sys_info["gpu_model"][0]
name = sys_info["workload_name"][0]
except KeyError as e:
console_error(
f"Outdated workload. Cannot find {e} field. Please reprofile to update."
)
else: else:
console_error( console_error(
"database", "Unable to parse SoC and/or workload name from sysinfo.csv" "database", "Unable to parse SoC and/or workload name from sysinfo.csv"