xuchen-amd
591632dd69
* Add new sample applications. * Generalize py test launcher for additional apps. * Add TCP pytest, and add to ctest. * Update licensing. * Disable for non-mi300 machines.
85 خطوط
2.5 KiB
Python
85 خطوط
2.5 KiB
Python
import subprocess
|
|
from importlib.machinery import SourceFileLoader
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
rocprof_compute = SourceFileLoader("rocprof-compute", "src/rocprof-compute").load_module()
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption(
|
|
"--call-binary",
|
|
action="store_true",
|
|
default=False,
|
|
help="Call standalone binary instead of main function during tests",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def binary_handler_profile_rocprof_compute(request):
|
|
def _handler(
|
|
config, workload_dir, options=[], check_success=True, roof=False, app_name="app_1"
|
|
):
|
|
if request.config.getoption("--call-binary"):
|
|
baseline_opts = [
|
|
"build/rocprof-compute.bin",
|
|
"profile",
|
|
"-n",
|
|
app_name,
|
|
"-VVV",
|
|
]
|
|
if not roof:
|
|
baseline_opts.append("--no-roof")
|
|
process = subprocess.run(
|
|
baseline_opts
|
|
+ options
|
|
+ ["--path", workload_dir, "--"]
|
|
+ config[app_name],
|
|
text=True,
|
|
)
|
|
# verify run status
|
|
if check_success:
|
|
assert process.returncode == 0
|
|
return process.returncode
|
|
else:
|
|
baseline_opts = ["rocprof-compute", "profile", "-n", app_name, "-VVV"]
|
|
if not roof:
|
|
baseline_opts.append("--no-roof")
|
|
with pytest.raises(SystemExit) as e:
|
|
with patch(
|
|
"sys.argv",
|
|
baseline_opts
|
|
+ options
|
|
+ ["--path", workload_dir, "--"]
|
|
+ config[app_name],
|
|
):
|
|
rocprof_compute.main()
|
|
# verify run status
|
|
if check_success:
|
|
assert e.value.code == 0
|
|
return e.value.code
|
|
|
|
return _handler
|
|
|
|
|
|
@pytest.fixture
|
|
def binary_handler_analyze_rocprof_compute(request):
|
|
def _handler(arguments):
|
|
if request.config.getoption("--call-binary"):
|
|
process = subprocess.run(
|
|
["build/rocprof-compute.bin", *arguments],
|
|
text=True,
|
|
)
|
|
return process.returncode
|
|
else:
|
|
with pytest.raises(SystemExit) as e:
|
|
with patch(
|
|
"sys.argv",
|
|
["rocprof-compute", *arguments],
|
|
):
|
|
rocprof_compute.main()
|
|
return e.value.code
|
|
|
|
return _handler
|