rocm-systems/examples/python/external.py

#!@PYTHON_EXECUTABLE@

import os
import sys
import random

_prefix = ""


def fib(n):
    return n if n < 2 else (fib(n - 1) + fib(n - 2))


def inefficient(n):
    print(f"[{_prefix}] ... running inefficient({n})")
    a = 0
    for i in range(n):
        a += i
        for j in range(n):
            a += j
    _len = a * n * n * n
    _arr = [random.random() for _ in range(_len)]
    _sum = sum(_arr)
    print(f"[{_prefix}] ... sum of {_len} random elements: {_sum}")
    return _sum


def run(n):
    _ret = 0
    _ret += fib(n)
    _ret += inefficient(n)
    return _ret


if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument("-n", "--num-iterations", help="Number", type=int, default=3)
    parser.add_argument("-v", "--value", help="Starting value", type=int, default=20)
    args = parser.parse_args()

    _prefix = os.path.basename(__file__)
    print(f"[{_prefix}] Executing {args.num_iterations} iterations...\n")
    for i in range(args.num_iterations):
        ans = run(args.value)
        print(f"[{_prefix}] [{i}] result of run({args.value}) = {ans}\n")