TAU: Added new test for other variants. (#42503)
Now it tests all GPUs, syscall and python.
This commit is contained in:
parent
5efa723289
commit
ff1e700b56
@ -408,6 +408,10 @@ def setup_run_environment(self, env):
|
|||||||
makefile_inc_test = join_path("include", "Makefile")
|
makefile_inc_test = join_path("include", "Makefile")
|
||||||
cuda_test = join_path("examples", "gpu", "cuda", "dataElem_um")
|
cuda_test = join_path("examples", "gpu", "cuda", "dataElem_um")
|
||||||
level_zero_test = join_path("examples", "gpu", "oneapi", "complex_mult")
|
level_zero_test = join_path("examples", "gpu", "oneapi", "complex_mult")
|
||||||
|
rocm_test = join_path("examples", "gpu", "hip", "vectorAdd")
|
||||||
|
syscall_test = join_path("examples", "syscall")
|
||||||
|
ompt_test = join_path("examples", "openmp", "c++")
|
||||||
|
python_test = join_path("examples", "python")
|
||||||
|
|
||||||
@run_after("install")
|
@run_after("install")
|
||||||
def setup_build_tests(self):
|
def setup_build_tests(self):
|
||||||
@ -422,6 +426,14 @@ def setup_build_tests(self):
|
|||||||
self.cache_extra_test_sources(self.cuda_test)
|
self.cache_extra_test_sources(self.cuda_test)
|
||||||
if "+level_zero" in self.spec:
|
if "+level_zero" in self.spec:
|
||||||
self.cache_extra_test_sources(self.level_zero_test)
|
self.cache_extra_test_sources(self.level_zero_test)
|
||||||
|
if "+rocm" in self.spec:
|
||||||
|
self.cache_extra_test_sources(self.rocm_test)
|
||||||
|
if "+syscall" in self.spec:
|
||||||
|
self.cache_extra_test_sources(self.syscall_test)
|
||||||
|
if "+ompt" in self.spec:
|
||||||
|
self.cache_extra_test_sources(self.ompt_test)
|
||||||
|
if "+python" in self.spec:
|
||||||
|
self.cache_extra_test_sources(self.python_test)
|
||||||
|
|
||||||
def _run_dyninst_test(self, test_dir):
|
def _run_dyninst_test(self, test_dir):
|
||||||
dyn_dir = join_path(test_dir, self.dyninst_test)
|
dyn_dir = join_path(test_dir, self.dyninst_test)
|
||||||
@ -457,6 +469,7 @@ def _run_tau_test(
|
|||||||
self, main_test_dir, test_dir, test_name, test_exe, tau_exec_flags=[], use_tau_exec=False
|
self, main_test_dir, test_dir, test_name, test_exe, tau_exec_flags=[], use_tau_exec=False
|
||||||
):
|
):
|
||||||
inst_test_dir = join_path(main_test_dir, test_dir)
|
inst_test_dir = join_path(main_test_dir, test_dir)
|
||||||
|
print(inst_test_dir)
|
||||||
test_description = "Build {} test code".format(test_name)
|
test_description = "Build {} test code".format(test_name)
|
||||||
self.run_test("make", ["all"], [], 0, False, test_description, False, inst_test_dir)
|
self.run_test("make", ["all"], [], 0, False, test_description, False, inst_test_dir)
|
||||||
if "+mpi" in self.spec:
|
if "+mpi" in self.spec:
|
||||||
@ -496,12 +509,44 @@ def _run_tau_test(
|
|||||||
inst_test_dir,
|
inst_test_dir,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def _run_python_test(self, test_dir):
|
||||||
|
python_dir = join_path(test_dir, self.python_test)
|
||||||
|
flags = "serial"
|
||||||
|
if "+mpi" in self.spec:
|
||||||
|
flags = "mpi"
|
||||||
|
self.run_test(
|
||||||
|
"tau_python",
|
||||||
|
["-T", flags, "firstprime.py"],
|
||||||
|
[],
|
||||||
|
0,
|
||||||
|
False,
|
||||||
|
"Pyhon example",
|
||||||
|
False,
|
||||||
|
python_dir,
|
||||||
|
)
|
||||||
|
self.run_test(
|
||||||
|
"pprof",
|
||||||
|
[],
|
||||||
|
[],
|
||||||
|
0,
|
||||||
|
False,
|
||||||
|
"Run pprof profile analysis tool on profile output",
|
||||||
|
False,
|
||||||
|
python_dir,
|
||||||
|
)
|
||||||
|
|
||||||
def test(self):
|
def test(self):
|
||||||
test_dir = self.test_suite.current_test_cache_dir
|
test_dir = self.test_suite.current_test_cache_dir
|
||||||
# Run mm test program pulled from the build
|
# Run mm test program pulled from the build
|
||||||
self._run_tau_test(test_dir, self.matmult_test, "matrix multiplication", "matmult")
|
if "+ompt" in self.spec:
|
||||||
|
tau_exec_flags = ["-ompt"]
|
||||||
|
self._run_tau_test(test_dir, self.ompt_test, "OMPT example", "mandel", tau_exec_flags)
|
||||||
|
else:
|
||||||
|
self._run_tau_test(test_dir, self.matmult_test, "matrix multiplication", "matmult")
|
||||||
if "+dyninst" in self.spec:
|
if "+dyninst" in self.spec:
|
||||||
self._run_dyninst_test(test_dir)
|
self._run_dyninst_test(test_dir)
|
||||||
|
if "+python" in self.spec:
|
||||||
|
self._run_python_test(test_dir)
|
||||||
if "+cuda" in self.spec:
|
if "+cuda" in self.spec:
|
||||||
tau_exec_flags = ["-cupti"]
|
tau_exec_flags = ["-cupti"]
|
||||||
self._run_tau_test(
|
self._run_tau_test(
|
||||||
@ -522,3 +567,23 @@ def test(self):
|
|||||||
tau_exec_flags,
|
tau_exec_flags,
|
||||||
use_tau_exec=True,
|
use_tau_exec=True,
|
||||||
)
|
)
|
||||||
|
if "+rocm" in self.spec and ("+rocprofiler" in self.spec or "+roctracer" in self.spec):
|
||||||
|
tau_exec_flags = ["-rocm"]
|
||||||
|
self._run_tau_test(
|
||||||
|
test_dir,
|
||||||
|
self.rocm_test,
|
||||||
|
"Rocm example",
|
||||||
|
"vectoradd_hip.exe",
|
||||||
|
tau_exec_flags,
|
||||||
|
use_tau_exec=True,
|
||||||
|
)
|
||||||
|
if "+syscall" in self.spec:
|
||||||
|
tau_exec_flags = ["-syscall"]
|
||||||
|
self._run_tau_test(
|
||||||
|
test_dir,
|
||||||
|
self.syscall_test,
|
||||||
|
"Syscall example",
|
||||||
|
"syscall_test",
|
||||||
|
tau_exec_flags,
|
||||||
|
use_tau_exec=True,
|
||||||
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user