Add NEON simulator and statespace headers

Author: xinpin

lib/BUILD

@@ -55,11 +55,13 @@ cc_library(
         "simulator_avx.h",
         "simulator_avx512.h",
         "simulator_basic.h",
+        "simulator_neon.h",
         "simulator_sse.h",
         "statespace.h",
         "statespace_avx.h",
         "statespace_avx512.h",
         "statespace_basic.h",
+        "statespace_neon.h",
         "statespace_sse.h",
         "umux.h",
         "unitary_calculator_avx.h",
@@ -130,13 +132,15 @@ cuda_library(
         "simulator_basic.h",
         "simulator_cuda.h",
         "simulator_cuda_kernels.h",
+        "simulator_neon.h",
         "simulator_sse.h",
         "statespace.h",
         "statespace_avx.h",
         "statespace_avx512.h",
         "statespace_basic.h",
         "statespace_cuda.h",
         "statespace_cuda_kernels.h",
+        "statespace_neon.h",
         "statespace_sse.h",
         "umux.h",
         "unitary_calculator_avx.h",
@@ -201,13 +205,15 @@ cuda_library(
         "simulator_basic.h",
         "simulator_custatevec.h",
         "simulator_custatevecex.h",
+        "simulator_neon.h",
         "simulator_sse.h",
         "statespace.h",
         "statespace_avx.h",
         "statespace_avx512.h",
         "statespace_basic.h",
         "statespace_custatevec.h",
         "statespace_custatevecex.h",
+        "statespace_neon.h",
         "statespace_sse.h",
         "umux.h",
         "unitary_calculator_avx.h",
@@ -602,6 +608,16 @@ cc_library(
     ],
 )
 
+cc_library(
+    name = "statespace_neon",
+    hdrs = ["statespace_neon.h"],
+    deps = [
+        ":statespace",
+        ":util",
+        ":vectorspace",
+    ],
+)
+
 cc_library(
     name = "statespace_sse",
     hdrs = ["statespace_sse.h"],
@@ -686,6 +702,17 @@ cc_library(
     ],
 )
 
+cc_library(
+    name = "simulator_neon",
+    hdrs = ["simulator_neon.h"],
+    deps = [
+        ":simulator_base",
+        ":simulator_basic",
+        ":statespace_basic",
+        ":statespace_neon",
+    ],
+)
+
 cc_library(
     name = "simulator_sse",
     hdrs = ["simulator_sse.h"],
@@ -741,6 +768,7 @@ cc_library(
         ":simulator_avx",
         ":simulator_avx512",
         ":simulator_basic",
+        ":simulator_neon",
         ":simulator_sse",
     ],
 )

lib/parfor.h

@@ -44,7 +44,7 @@ struct ParallelForT {
   template <typename Function, typename... Args>
   void Run(uint64_t size, Function&& func, Args&&... args) const {
     if (num_threads > 1 && size >= MIN_SIZE) {
-      #pragma omp parallel num_threads(num_threads)
+      #pragma omp parallel num_threads(num_threads) proc_bind(close)
       {
         unsigned n = omp_get_num_threads();
         unsigned m = omp_get_thread_num();
@@ -71,7 +71,7 @@ struct ParallelForT {
     if (num_threads > 1 && size >= MIN_SIZE) {
       partial_results.resize(num_threads, 0);
 
-      #pragma omp parallel num_threads(num_threads)
+      #pragma omp parallel num_threads(num_threads) proc_bind(close)
       {
         unsigned n = omp_get_num_threads();
         unsigned m = omp_get_thread_num();

lib/simmux.h

@@ -15,7 +15,13 @@
 #ifndef SIMMUX_H_
 #define SIMMUX_H_
 
-#ifdef __AVX512F__
+#if defined(__ARM_NEON__) || defined(__ARM_NEON)
+# include "simulator_neon.h"
+  namespace qsim {
+    template <typename For>
+    using Simulator = SimulatorNEON<For>;
+  }
+#elif defined(__AVX512F__)
 # include "simulator_avx512.h"
   namespace qsim {
     template <typename For>