diff --git a/prodtests/full-system-test/analyze_gpu_benchmarks.py b/prodtests/full-system-test/analyze_gpu_benchmarks.py new file mode 100644 index 0000000000000..79c8ee43df8ab --- /dev/null +++ b/prodtests/full-system-test/analyze_gpu_benchmarks.py @@ -0,0 +1,464 @@ +#!/usr/bin/env python3 + +import argparse +import re +from pathlib import Path + +import numpy as np +import matplotlib.pyplot as plt + +try: + from scipy.optimize import curve_fit + SCIPY_AVAILABLE = True +except ImportError: + SCIPY_AVAILABLE = False + +CYAN = "\033[96m" +GREEN = "\033[92m" +MAGENTA = "\033[95m" +YELLOW = "\033[93m" +RED = "\033[91m" +BOLD = "\033[1m" +RESET = "\033[0m" + +LINE_RE = re.compile( + r"\[[^\]]*gpu-reconstruction[^\]]*\]:\s*" + r"\[(?P