[js/web] only apply max thread number when it's omitted (microsoft#7834)

shaoboyan091 · May 26, 2021 · 331f204 · 331f204
1 parent c5ea590
commit 331f204
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/js/web/lib/backend-wasm.ts b/js/web/lib/backend-wasm.ts
@@ -20,11 +20,10 @@ export const initializeFlags = (): void => {
     env.wasm.initTimeout = 0;
   }
 
-  if (typeof env.wasm.numThreads !== 'number' || !Number.isInteger(env.wasm.numThreads) || env.wasm.numThreads < 0) {
+  if (typeof env.wasm.numThreads !== 'number' || !Number.isInteger(env.wasm.numThreads) || env.wasm.numThreads <= 0) {
     const numCpuLogicalCores = typeof navigator === 'undefined' ? cpus().length : navigator.hardwareConcurrency;
-    env.wasm.numThreads = Math.ceil((numCpuLogicalCores || 1) / 2);
+    env.wasm.numThreads = Math.min(4, Math.ceil((numCpuLogicalCores || 1) / 2));
   }
-  env.wasm.numThreads = Math.min(4, env.wasm.numThreads);
 };
 
 class OnnxruntimeWebAssemblyBackend implements Backend {