Splitting unary ops

crazy-cat · Sep 25, 2017 · 0218c49 · 0218c49
1 parent 9abbba1
commit 0218c49
Show file tree

Hide file tree

Showing 6 changed files with 871 additions and 834 deletions.
diff --git a/src/operator/mxnet_op.h b/src/operator/mxnet_op.h
@@ -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
 };