mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
0f01de3b0b
* [js/web] Add wasm SIMD backend to onnxruntime-web * Import SIMD wasm artifacts enabled by PR #7839 * Detect SIMD capability of web engine * Use SIMD wasm backend in both single-thread and multi-thread cases * update optimized SIMD loading from ort web * code lint and format * fix WasmFileName in CI * replace deprecated wasm SIMD functions * fix unittest for simd * optimize CI pipeline to merge build matrix * make clean build for each config * fix simd wasm to enable it. * update script/pull-prebuilt-wasm-artifacts.ts Co-authored-by: Yulong Wang <yulongw@microsoft.com> Co-authored-by: Lei Zhang <zhang.huanning@hotmail.com>
75 lines
2 KiB
TypeScript
75 lines
2 KiB
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
import {EnvImpl} from './env-impl';
|
|
export declare namespace Env {
|
|
export interface WebAssemblyFlags {
|
|
/**
|
|
* set or get number of thread(s). If omitted or set to 0, number of thread(s) will be determined by system. If set
|
|
* to 1, no worker thread will be spawned.
|
|
*
|
|
* This setting is available only when WebAssembly multithread feature is available in current context.
|
|
*/
|
|
numThreads?: number;
|
|
|
|
/**
|
|
* set or get a boolean value indicating whether to enable SIMD. If set to false, SIMD will be forcely disabled.
|
|
*
|
|
* This setting is available only when WebAssembly SIMD feature is available in current context.
|
|
*/
|
|
simd?: boolean;
|
|
|
|
/**
|
|
* Set or get a number specifying the timeout for initialization of WebAssembly backend, in milliseconds. A zero
|
|
* value indicates no timeout is set. (default is 0)
|
|
*/
|
|
initTimeout?: number;
|
|
}
|
|
|
|
export interface WebGLFlags {
|
|
/**
|
|
* Set or get the WebGL Context ID (webgl or webgl2). Default value is webgl2.
|
|
*/
|
|
contextId?: 'webgl'|'webgl2';
|
|
/**
|
|
* Set or get the maximum batch size for matmul. 0 means to disable batching.
|
|
*/
|
|
matmulMaxBatchSize?: number;
|
|
/**
|
|
* Set or get the texture cache mode. Default value is full.
|
|
*/
|
|
textureCacheMode?: 'initializerOnly'|'full';
|
|
/**
|
|
* Set or get the packed texture mode
|
|
*/
|
|
pack?: boolean;
|
|
}
|
|
}
|
|
|
|
export interface Env {
|
|
/**
|
|
* set the severity level for logging. If omitted, default is 'warning'
|
|
*/
|
|
logLevel?: 'verbose'|'info'|'warning'|'error'|'fatal';
|
|
/**
|
|
* Indicate whether run in debug mode.
|
|
*/
|
|
debug?: boolean;
|
|
|
|
/**
|
|
* Represent a set of flags for WebAssembly
|
|
*/
|
|
wasm: Env.WebAssemblyFlags;
|
|
|
|
/**
|
|
* Represent a set of flags for WebGL
|
|
*/
|
|
webgl: Env.WebGLFlags;
|
|
|
|
[name: string]: unknown;
|
|
}
|
|
|
|
/**
|
|
* Represent a set of flags as a global singleton.
|
|
*/
|
|
export const env: Env = new EnvImpl();
|