apache
diff --git a/‎src/operator/mxnet_op.h‎
Lines changed: 6 additions & 0 deletions b/‎src/operator/mxnet_op.h‎
Lines changed: 6 additions & 0 deletions
@@ -244,6 +244,7 @@ template<typename OP>
 struct Kernel<OP, cpu> {
   template<typename ...Args>
   inline static void Launch(mshadow::Stream<cpu> *s, const int N, Args... args) {
+#ifdef _OPENMP
     const int omp_cores = Engine::Get()->num_omp_threads_per_worker();
     if (omp_cores <= 1) {
       // Zero means not to use OMP, but don't interfere with external OMP behavior
@@ -257,6 +258,11 @@ struct Kernel<OP, cpu> {
       }
     }
   }
+#else
+  for (int i = 0; i < N; ++i) {
+        OP::Map(i, args...);
+  }
+#endif
 };
Original file line number	Diff line number	Diff line change
`@@ -244,6 +244,7 @@ template<typename OP>`
`244`	`244`	`struct Kernel<OP, cpu> {`
`245`	`245`	`template<typename ...Args>`
`246`	`246`	`inline static void Launch(mshadow::Stream<cpu> *s, const int N, Args... args) {`
	`247`	`+#ifdef _OPENMP`
`247`	`248`	`const int omp_cores = Engine::Get()->num_omp_threads_per_worker();`
`248`	`249`	`if (omp_cores <= 1) {`
`249`	`250`	`// Zero means not to use OMP, but don't interfere with external OMP behavior`
`@@ -257,6 +258,11 @@ struct Kernel<OP, cpu> {`
`257`	`258`	`}`
`258`	`259`	`}`
`259`	`260`	`}`
	`261`	`+#else`
	`262`	`+ for (int i = 0; i < N; ++i) {`
	`263`	`+ OP::Map(i, args...);`
	`264`	`+ }`
	`265`	`+#endif`
`260`	`266`	`};`
`261`	`267`
`262`	`268`