From 1b82d835d8465e772c4bf3f6b314f9d7395d6f50 Mon Sep 17 00:00:00 2001 From: Pavan Goyal Date: Wed, 10 Jul 2024 19:30:18 +0530 Subject: [PATCH] [Fix] InterOpNumThreads Session Option for ONNX ReactNative Package (#21263) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit ### Description This PR resolves a bug related to setting the **interOpNumThreads** session option when creating an **ORTSession**. Currently, when the **interOpNumThreads** option is passed from React Native, the native module incorrectly sets **intraOpNumThreads** instead of **interOpNumThreads**. ### Motivation and Context Since this is a bug, users of the Onnx React Native package may believe that they are setting **interOpNumThreads** correctly, So this change is required. Refer to the code snippet below for details Screenshot 2024-07-05 at 9 28 58 PM --- .../main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/js/react_native/android/src/main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java b/js/react_native/android/src/main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java index 3915292648..1be8c22b40 100644 --- a/js/react_native/android/src/main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java +++ b/js/react_native/android/src/main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java @@ -349,7 +349,7 @@ public class OnnxruntimeModule extends ReactContextBaseJavaModule implements Lif if (options.hasKey("interOpNumThreads")) { int interOpNumThreads = options.getInt("interOpNumThreads"); if (interOpNumThreads > 0 && interOpNumThreads < Integer.MAX_VALUE) { - sessionOptions.setIntraOpNumThreads(interOpNumThreads); + sessionOptions.setInterOpNumThreads(interOpNumThreads); } }