quick fix how to call v3 with pc sampling

[ROCm/rocprofiler-compute commit: 3a40e4b58c]
This commit is contained in:
Fei Zheng
2025-07-31 08:17:06 -06:00
committed by GitHub
parent deefddf2ff
commit 94ce792f39
2 changed files with 6 additions and 2 deletions
@@ -27,6 +27,7 @@ import glob
import logging import logging
import os import os
import re import re
import shlex
import shutil import shutil
import time import time
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
@@ -462,7 +463,9 @@ class RocProfCompute_Base:
method=self.get_args().pc_sampling_method, method=self.get_args().pc_sampling_method,
interval=self.get_args().pc_sampling_interval, interval=self.get_args().pc_sampling_interval,
workload_dir=self.get_args().path, workload_dir=self.get_args().path,
appcmd=self.get_args().remaining, appcmd=shlex.split(
self.get_args().remaining
), # FIXME: the right solution is applying it when argparsing once!
rocprofiler_sdk_library_path=self.get_args().rocprofiler_sdk_library_path, rocprofiler_sdk_library_path=self.get_args().rocprofiler_sdk_library_path,
) )
end_run_prof = time.time() end_run_prof = time.time()
@@ -1018,8 +1018,9 @@ def pc_sampling_prof(
"-o", "-o",
"ps_file", # todo: sync up with the name from source in 2100_.yaml "ps_file", # todo: sync up with the name from source in 2100_.yaml
"--", "--",
appcmd,
] ]
options.extend(appcmd)
success, output = capture_subprocess_output( success, output = capture_subprocess_output(
[rocprof_cmd] + options, new_env=os.environ.copy(), profileMode=True [rocprof_cmd] + options, new_env=os.environ.copy(), profileMode=True
) )