mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-30 03:37:44 +00:00
c373a808a2
### Description
* follows the packaging approach according to the design document
* adds `ENABLE_TRAINING` boolean flag to `BUILD_DEFS`
* modifies `package.json` to include training submodule
* modifies build script to handle, validate, and minimize training WASM
artifacts
* adds the binding for the new backend with training enabled & the new
training artifacts
* adds training backend
* edits `index.ts` to use training backend depending on `BUILD_DEFS`
* edits `wasm-factory.ts` to use the training artifacts if necessary
### Motivation and Context
* we are in the process of adding web bindings to enable training.
* Adding the "glue" to allow onnxruntime-web to use the training WASM
artifacts is required for this work.
* Since BUILD_DEFS is defined and used at build time, I thought that it
made sense to bundle the changes to building in the same PR.
#### Related work
* #16521 allowed for training artifacts to be built
* #17333 must be merged in before this one
---------
Co-authored-by: Yulong Wang <7679871+fs-eire@users.noreply.github.com>
39 lines
1.2 KiB
TypeScript
39 lines
1.2 KiB
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
/* eslint-disable @typescript-eslint/no-unused-vars, @typescript-eslint/naming-convention */
|
|
|
|
/**
|
|
* The interface BuildDefinitions contains a set of flags which are defined at build time.
|
|
*
|
|
* Those flags are processed in bundler for tree shaking to remove unused code.
|
|
* No flags in this file should present in production build.
|
|
*/
|
|
interface BuildDefinitions {
|
|
/**
|
|
* defines whether to disable the whole WebGL backend in the build.
|
|
*/
|
|
readonly DISABLE_WEBGL: boolean;
|
|
/**
|
|
* defines whether to disable the whole WebGpu backend in the build.
|
|
*/
|
|
readonly DISABLE_WEBGPU: boolean;
|
|
/**
|
|
* defines whether to disable the whole WebAssembly backend in the build.
|
|
*/
|
|
readonly DISABLE_WASM: boolean;
|
|
/**
|
|
* defines whether to disable proxy feature in WebAssembly backend in the build.
|
|
*/
|
|
readonly DISABLE_WASM_PROXY: boolean;
|
|
/**
|
|
* defines whether to disable multi-threading feature in WebAssembly backend in the build.
|
|
*/
|
|
readonly DISABLE_WASM_THREAD: boolean;
|
|
/**
|
|
* defines whether to disable training APIs in WebAssembly backend.
|
|
*/
|
|
readonly DISABLE_TRAINING: boolean;
|
|
}
|
|
|
|
declare const BUILD_DEFS: BuildDefinitions;
|