saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-02 03:55:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[js/web] remove webgl from default fallback list (#9374)

Browse source
This commit is contained in:
Yulong Wang 2021-10-14 21:46:22 -07:00 committed by GitHub
parent f05c285a58
commit 901c7de918
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
js/common/lib/backend-impl.ts
View file

@ -20,7 +20,8 @@ const backendsSortedByPriority: string[] = [];
*
* @param name - the name as a key to lookup as an execution provider.
* @param backend - the backend object.
* @param priority - an integer indicating the priority of the backend. Higher number means higher priority.
* @param priority - an integer indicating the priority of the backend. Higher number means higher priority. if priority
* < 0, it will be considered as a 'beta' version and will not be used as a fallback backend by default.
*
* @internal
*/
@ -35,13 +36,15 @@ export const registerBackend = (name: string, backend: Backend, priority: number
throw new Error(`backend "${name}" is already registered`);
}
for (let i = 0; i < backendsSortedByPriority.length; i++) {
if (backends[backendsSortedByPriority[i]].priority <= priority) {
backendsSortedByPriority.splice(i, 0, name);
return;
if (priority >= 0) {
for (let i = 0; i < backendsSortedByPriority.length; i++) {
if (backends[backendsSortedByPriority[i]].priority <= priority) {
backendsSortedByPriority.splice(i, 0, name);
return;
}
}
backendsSortedByPriority.push(name);
}
backendsSortedByPriority.push(name);
return;
}

4
js/web/lib/index.ts
View file

@ -6,5 +6,5 @@ import {registerBackend} from 'onnxruntime-common';
import {onnxjsBackend} from './backend-onnxjs';
import {wasmBackend} from './backend-wasm';
registerBackend('webgl', onnxjsBackend, 1);
registerBackend('wasm', wasmBackend, 2);
registerBackend('webgl', onnxjsBackend, -1);
registerBackend('wasm', wasmBackend, 0);