From 2dc0412c11a33e33e2ae0a6b8f0aadef0645c459 Mon Sep 17 00:00:00 2001
From: Marco Barbone <mbarbone@flatironinstitute.org>
Date: Wed, 28 Aug 2024 16:05:07 -0400
Subject: [PATCH] added temperature check

---
 perftest/bench.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/perftest/bench.py b/perftest/bench.py
index 9b25d9ddd..45dfe4bc0 100644
--- a/perftest/bench.py
+++ b/perftest/bench.py
@@ -33,6 +33,20 @@ def build_args(args):
 # clone the repository
 run_command('git', ['clone', 'https://github.com/DiamonDinoia/finufft.git'])
 
+def get_cpu_temperature():
+    try:
+        # Run the sensors command
+        output, _ = run_command('sensors', [])
+        # Parse the output to find the CPU temperature
+        for line in output.split('\n'):
+            if 'Core 0' in line:
+                # Extract the temperature value
+                temp_str = line.split()[2]
+                return temp_str
+    except subprocess.CalledProcessError as e:
+        print('Error executing sensors command:', e)
+        return None
+
 all_data = pd.DataFrame()
 
 args = {
@@ -148,6 +162,9 @@ def plot_stacked_bar_chart(pivot_data, speedup_data, args, figname):
                 args['--' + 'type'] = type
                 for upsampfac in upsamp:
                     args['--upsampfac'] = upsampfac
+                    while (cpu_temp := float(get_cpu_temperature())) > 42.0:
+                        print(f'CPU temperature is {cpu_temp}°C, waiting for it to cool down...')
+                        sleep(30)
                     if param.thread == 1:
                          out, _ = run_command('taskset', ['-c', '0', 'build/perftest/perftest'] + build_args(args))
                     else: