quick fix how to call v3 with pc sampling

This commit is contained in:
Fei Zheng
2025-07-31 08:17:06 -06:00
committed by GitHub
parent c7b5a0f43a
commit 3a40e4b58c
2 changed files with 6 additions and 2 deletions
+4 -1
View File
@@ -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()
+2 -1
View File
@@ -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
) )