mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
56bced0581
### Description enable webgpu in browser unit test. The CI pipeline uses Edge v113+ which enables WebGPU. === **UPDATE on 08/07/2023:** - add flags to Edge browser launch commandline so that Edge on CI agents can initialize WebGPU correctly. - ONLY enable webgpu on web release build. Other pipelines are using flag `-b=wasm,webgl,xnnpack` to specify the other 3 backends explicitly. - disable "Resize" related test failures. Once they are fixed the tests can be re-enabled. --------- Co-authored-by: Satya Jandhyala <satya.k.jandhyala@gmail.com>
55 lines
2 KiB
JavaScript
55 lines
2 KiB
JavaScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
'use strict';
|
|
|
|
const args = require('minimist')(process.argv.slice(2));
|
|
const SELF_HOST = !!args['self-host'];
|
|
const ORT_MAIN = args['ort-main'];
|
|
const TEST_MAIN = args['test-main'];
|
|
if (typeof TEST_MAIN !== 'string') {
|
|
throw new Error('flag --test-main=<TEST_MAIN_JS_FILE> is required');
|
|
}
|
|
const USER_DATA = args['user-data'];
|
|
if (typeof USER_DATA !== 'string') {
|
|
throw new Error('flag --user-data=<CHROME_USER_DATA_FOLDER> is required');
|
|
}
|
|
|
|
module.exports = function(config) {
|
|
const distPrefix = SELF_HOST ? './node_modules/onnxruntime-web/dist/' : 'http://localhost:8081/dist/';
|
|
config.set({
|
|
frameworks: ['mocha'],
|
|
files: [
|
|
{pattern: distPrefix + ORT_MAIN},
|
|
{pattern: './common.js'},
|
|
{pattern: TEST_MAIN},
|
|
{pattern: './node_modules/onnxruntime-web/dist/*.wasm', included: false, nocache: true},
|
|
{pattern: './model.onnx', included: false},
|
|
],
|
|
plugins: [require('@chiragrupani/karma-chromium-edge-launcher'), ...config.plugins],
|
|
proxies: {
|
|
'/model.onnx': '/base/model.onnx',
|
|
'/test-wasm-path-override/ort-wasm.wasm': '/base/node_modules/onnxruntime-web/dist/ort-wasm.wasm',
|
|
'/test-wasm-path-override/renamed.wasm': '/base/node_modules/onnxruntime-web/dist/ort-wasm.wasm',
|
|
},
|
|
client: {captureConsole: true, mocha: {expose: ['body'], timeout: 60000}},
|
|
reporters: ['mocha'],
|
|
captureTimeout: 120000,
|
|
reportSlowerThan: 100,
|
|
browserDisconnectTimeout: 600000,
|
|
browserNoActivityTimeout: 300000,
|
|
browserDisconnectTolerance: 0,
|
|
browserSocketTimeout: 60000,
|
|
hostname: 'localhost',
|
|
browsers: [],
|
|
customLaunchers: {
|
|
Chrome_default: {base: 'ChromeHeadless', chromeDataDir: USER_DATA},
|
|
Chrome_no_threads: {
|
|
base: 'ChromeHeadless',
|
|
chromeDataDir: USER_DATA,
|
|
// TODO: no-thread flags
|
|
},
|
|
Edge_default: {base: 'Edge', edgeDataDir: USER_DATA}
|
|
}
|
|
});
|
|
};
|