Fix reporting of --roof-only runs in sys_info
Signed-off-by: colramos-amd <colramos@amd.com>
[ROCm/rocprofiler-compute commit: 37bd6b78ae]
This commit is contained in:
@@ -142,4 +142,11 @@ class OmniProfiler_Base():
|
||||
"""Perform any post-processing steps prior to profiling.
|
||||
"""
|
||||
logging.debug("[profiling] performing post-processing using %s profiler" % self.__profiler)
|
||||
gen_sysinfo(self.__args.name, self.get_args().path, self.__args.ipblocks, self.__args.remaining, self.__args.no_roof)
|
||||
gen_sysinfo(
|
||||
workload_name=self.__args.name,
|
||||
workload_dir=self.get_args().path,
|
||||
ip_blocks=self.__args.ipblocks,
|
||||
app_cmd=self.__args.remaining,
|
||||
skip_roof=self.__args.no_roof,
|
||||
roof_only=self.__args.roof_only,
|
||||
)
|
||||
@@ -335,7 +335,14 @@ class Roofline:
|
||||
sysinfo_path = os.path.join(self.__args.path, "sysinfo.csv")
|
||||
if not os.path.isfile(sysinfo_path):
|
||||
logging.info("[roofline] sysinfo.csv not found. Generating...")
|
||||
gen_sysinfo(self.__args.name, self.__workload_dir, self.__args.ipblocks, self.__args.remaining, self.__args.no_roof)
|
||||
gen_sysinfo(
|
||||
workload_name=self.__args.name,
|
||||
workload_dir=self.__workload_dir,
|
||||
ip_blocks=self.__args.ipblocks,
|
||||
app_cmd=self.__args.remaining,
|
||||
skip_roof=self.__args.no_roof,
|
||||
roof_only=self.__args.roof_only
|
||||
)
|
||||
|
||||
@abstractmethod
|
||||
def profile(self):
|
||||
|
||||
@@ -235,7 +235,7 @@ def replace_timestamps(workload_dir):
|
||||
warning = "WARNING: Incomplete profiling data detected. Unable to update timestamps."
|
||||
logging.warning(warning + "\n")
|
||||
|
||||
def gen_sysinfo(workload_name, workload_dir, ip_blocks, app_cmd, skip_roof):
|
||||
def gen_sysinfo(workload_name, workload_dir, ip_blocks, app_cmd, skip_roof, roof_only):
|
||||
# Record system information
|
||||
mspec = specs.get_machine_specs(0)
|
||||
sysinfo = open(workload_dir + "/" + "sysinfo.csv", "w")
|
||||
@@ -295,7 +295,10 @@ def gen_sysinfo(workload_name, workload_dir, ip_blocks, app_cmd, skip_roof):
|
||||
elif mspec.GPU == "gfx90a":
|
||||
param += ["32", "32", "mi200", "56", str(hbmBW)]
|
||||
if not skip_roof:
|
||||
blocks.append("roofline")
|
||||
if roof_only:
|
||||
ip_blocks = ["roofline"]
|
||||
else:
|
||||
blocks.append("roofline")
|
||||
|
||||
# ip block info
|
||||
if ip_blocks == None:
|
||||
|
||||
Reference in New Issue
Block a user