Skip to content

Commit

Permalink
fix: interOpNumThreads for onnx reactnative package
Browse files Browse the repository at this point in the history
  • Loading branch information
pavangoyal42 committed Jul 5, 2024
1 parent 3f6b743 commit 9d42a50
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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);
}
}

Expand Down

0 comments on commit 9d42a50

Please sign in to comment.