Files
rocm-systems/projects/rocprofiler-systems/examples/python/fill.py
T
David Galiffi 489eda995d Rename Omnitrace to ROCm Systems Profiler (#4)
The Omnitrace program is being renamed. 

Full name: "ROCm Systems Profiler"
Package name: "rocprofiler-systems"
Binary / Library names: "rocprof-sys-*"

---------
Co-authored-by: Xuan Chen <xuchen@amd.com>
Signed-off-by: David Galiffi <David.Galiffi@amd.com>

[ROCm/rocprofiler-systems commit: d07bf508a9]
2024-10-15 11:20:40 -04:00

40 rader
967 B
Python
Executable File

#!@PYTHON_EXECUTABLE@
import os
import sys
import time
import rocprofsys
from rocprofsys.user import region as omni_user_region
from rocprofsys.profiler import config as omni_config
_prefix = ""
def loop(n):
pass
@rocprofsys.profile()
def run(i, n, v):
for l in range(n * n):
loop(v + l)
return v + (n * n)
if __name__ == "__main__":
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-n", "--num-iterations", help="Number", type=int, default=100)
parser.add_argument("-v", "--value", help="Starting value", type=int, default=10)
args = parser.parse_args()
omni_config.include_args = True
_prefix = os.path.basename(__file__)
print(f"[{_prefix}] Executing {args.num_iterations} iterations...\n")
ans = 0
for i in range(args.num_iterations):
beg = ans
ans = run(i, args.value, beg)
print(f"[{_prefix}] [{i}] result of run({args.value}, {beg}) = {ans}")