1 files changed, 6 insertions, 1 deletions

diff --git a/benchmark/bench.sh b/benchmark/bench.sh
index 839515a..882c836 100755
--- a/benchmark/bench.sh
+++ b/benchmark/bench.sh
@@ -1,10 +1,15 @@
 #!/bin/sh
 
+CC="cc -std=c99 -pthread -O3 -D$CUBETYPE"
+if [ "$CUBETYPE" = "CUBE_AVX2" ]; then
+        CC="$CC -mavx2"
+fi
+
 BENCHBIN="benchmark/run"
 BENCHDIR="benchmark/results"
 CUBEOBJ="cube.o"
 
-cc -std=c99 -pthread -O3 -o $BENCHBIN benchmark/bench.c $CUBEOBJ || exit 1;
+$CC -o $BENCHBIN benchmark/bench.c $CUBEOBJ || exit 1;
 
 d="$(date +'%Y-%m-%d-%H-%M-%S')" 
 mkdir -p "$BENCHDIR"