Add OpenMP usage

xiph · Dec 5, 2018 · 7ec7072 · 7ec7072
1 parent b85dda6
commit 7ec7072
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/compile.sh b/src/compile.sh
@@ -1,4 +1,4 @@
 #!/bin/sh
 
 gcc -DTRAINING=1 -Wall -W -O3 -g -I../include denoise.c kiss_fft.c pitch.c celt_lpc.c -o dump_data -lm
-gcc -o test_lpcnet -mavx2 -mfma -g -O3 -Wall -W -Wextra lpcnet.c nnet.c nnet_data.c -lm
+gcc -fopenmp -o test_lpcnet -mavx2 -mfma -g -O3 -Wall -W -Wextra lpcnet.c nnet.c nnet_data.c -lm
diff --git a/src/lpcnet.c b/src/lpcnet.c
@@ -31,6 +31,7 @@
 #include "common.h"
 #include "arch.h"
 #include "lpcnet.h"
+#include "omp.h"
 
 #define NB_FEATURES 38
 #define NB_TOTAL_FEATURES 55
@@ -166,6 +167,7 @@ void lpcnet_synthesize(LPCNetState *lpcnet, short *output, const float *features
         RNN_CLEAR(output, N);
         return;
     }
+    #pragma omp parallel for
     for (i=start;i<N;i++)
     {
         int j;
@@ -195,6 +197,7 @@ void lpcnet_synthesize(LPCNetState *lpcnet, short *output, const float *features
 int main(int argc, char **argv) {
     FILE *fin, *fout;
     LPCNetState *net;
+
     net = lpcnet_create();
     if (argc != 3)
     {