Include omp headers from compiler installation

microkernels/sbatch/heuristic_search_conv_daint.sh

@@ -3,7 +3,7 @@
 #SBATCH --job-name=heuristic_search_conv
 #SBATCH --output=logs/heuristic_search_conv_daint.log
 #SBATCH --account=g34
-#SBATCH --time=1:00:00
+#SBATCH --time=4:00:00
 #SBATCH --constraint=mc
 #SBATCH --partition=normal
 #SBATCH --nodes=1
@@ -12,7 +12,7 @@ source ~/.bashrc
 conda activate main
 export CC=clang
 export LDFLAGS="-lm"
-export CFLAGS="-O3 -I . -fopenmp -march=native"
+export CFLAGS="-O3 -I . -fopenmp -march=native -I$(dirname $(which clang))/../include/"
 export LD_LIBRARY_PATH="$(dirname $(which clang))/../lib/"
 
 srun python heuristic_search.py --kernel conv --min_repeats 3
\ No newline at end of file

microkernels/sbatch/heuristic_search_layernorm_daint.sh

@@ -3,7 +3,7 @@
 #SBATCH --job-name=heuristic_search_layernorm
 #SBATCH --output=logs/heuristic_search_layernorm_daint.log
 #SBATCH --account=g34
-#SBATCH --time=1:00:00
+#SBATCH --time=4:00:00
 #SBATCH --constraint=mc
 #SBATCH --partition=normal
 #SBATCH --nodes=1
@@ -12,7 +12,7 @@ source ~/.bashrc
 conda activate main
 export CC=clang
 export LDFLAGS="-lm"
-export CFLAGS="-O3 -I . -fopenmp -march=native"
+export CFLAGS="-O3 -I . -fopenmp -march=native -I$(dirname $(which clang))/../include/"
 export LD_LIBRARY_PATH="$(dirname $(which clang))/../lib/"
 
 srun python heuristic_search.py --kernel layernorm --min_repeats 3
\ No newline at end of file

microkernels/sbatch/heuristic_search_log_softmax_daint.sh

@@ -3,7 +3,7 @@
 #SBATCH --job-name=heuristic_search_log_softmax
 #SBATCH --output=logs/heuristic_search_log_softmax_daint.log
 #SBATCH --account=g34
-#SBATCH --time=1:00:00
+#SBATCH --time=4:00:00
 #SBATCH --constraint=mc
 #SBATCH --partition=normal
 #SBATCH --nodes=1
@@ -12,7 +12,7 @@ source ~/.bashrc
 conda activate main
 export CC=clang
 export LDFLAGS="-lm"
-export CFLAGS="-O3 -I . -fopenmp -march=native"
+export CFLAGS="-O3 -I . -fopenmp -march=native -I$(dirname $(which clang))/../include/"
 export LD_LIBRARY_PATH="$(dirname $(which clang))/../lib/"
 
 srun python heuristic_search.py --kernel log_softmax --min_repeats 3
\ No newline at end of file

microkernels/sbatch/heuristic_search_matmul_daint.sh

@@ -3,7 +3,7 @@
 #SBATCH --job-name=heuristic_search_matmul
 #SBATCH --output=logs/heuristic_search_matmul_daint.log
 #SBATCH --account=g34
-#SBATCH --time=1:00:00
+#SBATCH --time=4:00:00
 #SBATCH --constraint=mc
 #SBATCH --partition=normal
 #SBATCH --nodes=1
@@ -12,7 +12,7 @@ source ~/.bashrc
 conda activate main
 export CC=clang
 export LDFLAGS="-lm"
-export CFLAGS="-O3 -I . -fopenmp -march=native"
+export CFLAGS="-O3 -I . -fopenmp -march=native -I$(dirname $(which clang))/../include/"
 export LD_LIBRARY_PATH="$(dirname $(which clang))/../lib/"
 
 srun python heuristic_search.py --kernel matmul --min_repeats 1
\ No newline at end of file

microkernels/sbatch/heuristic_search_reducemean_daint.sh

@@ -3,7 +3,7 @@
 #SBATCH --job-name=heuristic_search_reducemean
 #SBATCH --output=logs/heuristic_search_reducemean_daint.log
 #SBATCH --account=g34
-#SBATCH --time=1:00:00
+#SBATCH --time=4:00:00
 #SBATCH --constraint=mc
 #SBATCH --partition=normal
 #SBATCH --nodes=1
@@ -12,7 +12,7 @@ source ~/.bashrc
 conda activate main
 export CC=clang
 export LDFLAGS="-lm"
-export CFLAGS="-O3 -I . -fopenmp -march=native"
+export CFLAGS="-O3 -I . -fopenmp -march=native -I$(dirname $(which clang))/../include/"
 export LD_LIBRARY_PATH="$(dirname $(which clang))/../lib/"
 
 srun python heuristic_search.py --kernel reducemean --min_repeats 3
\ No newline at end of file