From 2a40e38b13b0ada94dcc516bd11b3a107e3f1729 Mon Sep 17 00:00:00 2001
From: Comma Device <device@comma.ai>
Date: Thu, 19 Nov 2020 01:30:01 +0000
Subject: [PATCH] update waste for variable cores

---
 scripts/waste.c | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/scripts/waste.c b/scripts/waste.c
index a16eec49..0ddb0971 100644
--- a/scripts/waste.c
+++ b/scripts/waste.c
@@ -1,12 +1,16 @@
+// gcc -O2 waste.c -lpthread -owaste
+
 #define _GNU_SOURCE
 #include <stdio.h>
 #include <sched.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <pthread.h>
 #include <arm_neon.h>
+#include <sys/sysinfo.h>
 #include "../selfdrive/common/timing.h"
 
-#define CORES 4
-double ttime[CORES];
-double oout[CORES];
+double *ttime, *oout;
 
 void waste(int pid) {
   cpu_set_t my_set;
@@ -37,6 +41,10 @@ void waste(int pid) {
 }
 
 int main() {
+  int CORES = get_nprocs();
+  ttime = (double *)malloc(CORES*sizeof(double));
+  oout = (double *)malloc(CORES*sizeof(double));
+
   pthread_t waster[CORES];
   for (int i = 0 ; i < CORES; i++) {
     pthread_create(&waster[i], NULL, waste, (void*)i);