Improve kokkos trace handling. (#573)

* Improve kokkos trace is handling.

* Fix formatting.

* Syntax fix and adding comments.

[ROCm/rocprofiler-compute commit: 10e1d57bfb]
Этот коммит содержится в:
xuchen-amd
2025-02-22 14:07:14 -05:00
коммит произвёл GitHub
родитель 8c43eec752
Коммит 69f0312b7b
2 изменённых файлов: 129 добавлений и 80 удалений
+20
Просмотреть файл
@@ -77,6 +77,26 @@ class RocProfCompute_Base:
out = self.__args.path + "/pmc_perf.csv"
files = glob.glob(self.__args.path + "/" + "pmc_perf_*.csv")
files.extend(glob.glob(self.__args.path + "/" + "SQ_*.csv"))
if self.get_args().hip_trace:
# remove hip api trace ouputs from this list
files = [
f
for f in files
if not re.compile(r"^.*_hip_api_trace\.csv$").match(
os.path.basename(f)
)
]
if self.get_args().kokkos_trace:
# remove marker api trace ouputs from this list
files = [
f
for f in files
if not re.compile(r"^.*_marker_api_trace\.csv$").match(
os.path.basename(f)
)
]
elif type(self.__args.path) == list:
files = self.__args.path
else: