Add benchmarks for some of the basic operations on some of the basic types

benchmarks/simple_benchmarks.py

@@ -0,0 +1,126 @@
+"""Benchmarks for basic operations of a number of flint types."""
+
+import timeit
+from flint import arb, fmpz, fmpq, acb
+from utils import run_all_benchmarks, print_benchmark_results
+
+NUM_EXECUTIONS = 10000000
+
+
+def benchmark_arb_addition(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for adding two arbs."""
+    a = arb(1, 2)
+    b = arb.pi()
+
+    return timeit.timeit(lambda: a + b, number=num_executions)
+
+
+def benchmark_arb_multiplication(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for multiplying two arbs."""
+    a = arb(1, 2)
+    b = arb.pi()
+
+    return timeit.timeit(lambda: a * b, number=num_executions)
+
+
+def benchmark_arb_contains(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for comparing two arbs."""
+    a = arb(1, 2)
+    b = arb.pi()
+
+    def to_time():
+        first = a in b
+        second = a in a
+        return first, second
+
+    return timeit.timeit(to_time, number=num_executions)
+
+
+def benchmark_fmpz_addition(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for adding two fmpzs."""
+    a = fmpz(1)
+    b = fmpz(6)
+
+    return timeit.timeit(lambda: a + b, number=num_executions)
+
+
+def benchmark_fmpz_multiplication(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for multiplying two fmpzs."""
+    a = fmpz(1)
+    b = fmpz(6)
+
+    return timeit.timeit(lambda: a * b, number=num_executions)
+
+
+def benchmark_fmpz_eq(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for equality on two fmpzs."""
+    a = fmpz(1)
+    b = fmpz(6)
+
+    def to_time():
+        first = a == b
+        second = a == a
+        return first, second
+
+    return timeit.timeit(to_time, number=num_executions)
+
+
+def benchmark_fmpq_addition(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for adding two fmpqs."""
+    a = fmpq(1, 2)
+    b = fmpq(15, 7)
+
+    return timeit.timeit(lambda: a + b, number=num_executions)
+
+
+def benchmark_fmpq_multiplication(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for multiplying two fmpqs."""
+    a = fmpq(1, 2)
+    b = fmpq(15, 7)
+
+    return timeit.timeit(lambda: a * b, number=num_executions)
+
+
+def benchmark_fmpq_eq(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for equality on two fmpqs."""
+    a = fmpq(1, 2)
+    b = fmpq(15, 7)
+
+    def to_time():
+        first = a == b
+        second = a == a
+        return first, second
+
+    return timeit.timeit(to_time, number=num_executions)
+
+
+def benchmark_acb_addition(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for adding two acbs."""
+    a = acb(1 + 3j)
+    b = acb.pi()
+
+    return timeit.timeit(lambda: a + b, number=num_executions)
+
+
+def benchmark_acb_multiplication(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for multiplying two acbs."""
+    a = acb(1 + 3j)
+    b = acb.pi()
+
+    return timeit.timeit(lambda: a * b, number=num_executions)
+
+
+def benchmark_acb_eq(num_executions = NUM_EXECUTIONS):
+    """Simple benchmark for equality on two acbs."""
+    a = acb(1 + 3j)
+    b = acb.pi()
+
+    def to_time():
+        first = a in b
+        second = a in a
+        return first, second
+
+    return timeit.timeit(to_time, number=num_executions)
+
+if __name__ == "__main__":
+    print_benchmark_results(run_all_benchmarks(__name__))

benchmarks/utils.py

@@ -0,0 +1,28 @@
+"""Utility functions for benchmarking."""
+
+import inspect
+import sys
+
+
+def get_all_benchmarks(modulename):
+    """Returns a list of all functions in the current module named 'benchmark_*'."""
+    return [
+        obj
+        for name, obj in inspect.getmembers(sys.modules[modulename])
+        if inspect.isfunction(obj) and name.startswith("benchmark_")
+    ]
+
+
+def run_all_benchmarks(modulename):
+    """Runs all benchmarks in the current file and returns their times."""
+    return {
+        benchmark.__name__[len("benchmark_"):]: benchmark()
+        for benchmark in get_all_benchmarks(modulename)
+    }
+
+def print_benchmark_results(results):
+    """Pretty-prints a set of benchmark results."""
+    print(results)
+    longest_name = max(len(name) for name in results.keys())
+    for name, time in results.items():
+        print(f"{name:<{longest_name+2}} | {time:>10.5f}")