Allow disabling of mutlithreading in simd op check (halide#8096)

simd_op_check_xtensa is not threadsafe at present
ardier · Mar 3, 2024 · a205550 · a205550
1 parent 90fb5f8
commit a205550
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/test/correctness/simd_op_check.h b/test/correctness/simd_op_check.h
@@ -450,6 +450,10 @@ class SimdOpCheckTest {
         return 16;
     }
 
+    virtual bool use_multiple_threads() const {
+        return true;
+    }
+
     virtual bool test_all() {
         /* First add some tests based on the target */
         add_tests();
@@ -460,7 +464,10 @@ class SimdOpCheckTest {
 
         Sharder sharder;
 
-        Halide::Tools::ThreadPool<TestResult> pool;
+        Halide::Tools::ThreadPool<TestResult> pool(
+            use_multiple_threads() ?
+                Halide::Tools::ThreadPool<TestResult>::num_processors_online() :
+                1);
         std::vector<std::future<TestResult>> futures;
 
         for (size_t t = 0; t < tasks.size(); t++) {