From 9d42a5009b238f288aa8e7ca8cc488661d33e9c8 Mon Sep 17 00:00:00 2001 From: Pavan Goyal Date: Fri, 5 Jul 2024 21:14:30 +0530 Subject: [PATCH] fix: interOpNumThreads for onnx reactnative package --- .../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 3915292648aee..1be8c22b40da8 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 @@ private SessionOptions parseSessionOptions(ReadableMap options) throws OrtExcept if (options.hasKey("interOpNumThreads")) { int interOpNumThreads = options.getInt("interOpNumThreads"); if (interOpNumThreads > 0 && interOpNumThreads < Integer.MAX_VALUE) { - sessionOptions.setIntraOpNumThreads(interOpNumThreads); + sessionOptions.setInterOpNumThreads(interOpNumThreads); } }