diff --git a/src/omniperf_analyze/analysis_base.py b/src/omniperf_analyze/analysis_base.py index 37c1278e4b..a67cb044e2 100644 --- a/src/omniperf_analyze/analysis_base.py +++ b/src/omniperf_analyze/analysis_base.py @@ -126,7 +126,7 @@ class OmniAnalyze_Base(): sys_info = file_io.load_sys_info(Path(d[0], "sysinfo.csv")) arch = sys_info.iloc[0]["gpu_soc"] args = self.__args - self.generate_configs(arch, args.config_dir, args.list_kernels, args.filter_metrics, sys_info) + self.generate_configs(arch, args.config_dir, args.list_kernels, args.filter_metrics, sys_info.iloc[0]) self.load_options(normalization_filter) diff --git a/src/utils/parser.py b/src/utils/parser.py index c4f01dca59..dbb645d3f5 100644 --- a/src/utils/parser.py +++ b/src/utils/parser.py @@ -30,7 +30,7 @@ import os import pandas as pd import numpy as np from utils import schema -from utils.utils import error +from utils.utils import error, get_hbm_stack_num from pathlib import Path import logging