saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Fix] InterOpNumThreads Session Option for ONNX ReactNative Package (#21263)

Browse source 
### 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
<img width="634" alt="Screenshot 2024-07-05 at 9 28 58 PM"
src="https://github.com/microsoft/onnxruntime/assets/88655321/70a8f216-553a-4f4c-9481-e6871f0e37e6">
This commit is contained in:
Pavan Goyal 2024-07-10 19:30:18 +05:30 committed by GitHub
parent 1b19045afa
commit 1b82d835d8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
js/react_native/android/src/main/java/ai/onnxruntime/reactnative/OnnxruntimeModule.java
View file

@ -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);
}
}