From 4abe6de52651149b85c28268fa3a0c0407de3c4f Mon Sep 17 00:00:00 2001 From: Stefano Bosisio <75615539+Steboss89@users.noreply.github.com> Date: Fri, 10 Feb 2023 14:15:40 +0000 Subject: [PATCH] Update run_gpu_benchmark.py --- benchmarking/run_gpu_benchmark.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmarking/run_gpu_benchmark.py b/benchmarking/run_gpu_benchmark.py index c6bbfaa..707491d 100644 --- a/benchmarking/run_gpu_benchmark.py +++ b/benchmarking/run_gpu_benchmark.py @@ -80,6 +80,7 @@ def main(): for algorithm_name, factorization in factorized_algorithms: if algorithm_name == 'AlphaTensor TPU-optimized' and s > 19000: continue # This TPU-optimized algorithm runs OOM on a V100 GPU. + print(algorithm_name) results_algorithm = utils.benchmark_factorized_algorithm( factorization, (s, s, s), num_trials=num_trials) ratio = np.median(results_dot / results_algorithm)