Merge pull request #85 from etaf/master

Parallize data initailization.
baidu-research · Apr 4, 2018 · 9ab7a74 · 9ab7a74
2 parents 6a50a01 + 9fe447f
commit 9ab7a74
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/code/intel/convolution/mkl_conv/Makefile b/code/intel/convolution/mkl_conv/Makefile
@@ -53,7 +53,7 @@ OPTFLAGS = -O3
 endif
 
 CXX      = icpc
-CXXFLAGS = -Wall -std=c++11 $(OPTFLAGS) $(EXTRACXXFLAGS) -I../../../kernels
+CXXFLAGS = -Wall -std=c++11 $(OPTFLAGS) $(EXTRACXXFLAGS) -fopenmp -I../../../kernels
 LFLAGS   = -lrt
 OBJS     = std_conv_bench.o
 

diff --git a/code/intel/convolution/mkl_conv/std_conv_bench.cpp b/code/intel/convolution/mkl_conv/std_conv_bench.cpp
@@ -127,6 +127,7 @@ static inline void rand_fill(T *data, size_t size)
         srand48(1);
         initialized = true;
     }
+#pragma omp parallel for
     for (size_t i = 0; i < size / sizeof(T); i++)
         data[i] = static_cast<T>(drand48());
 }