mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
74eaaad768
* Support opset-13 for squeeze, unsqueeze, maxpool, pad, cast, clip * merge master and update a operators.md * resolve comment. revise pool and cast kernel implementation. * skip fusion when clip min and max is not in initializer
80 lines
3.3 KiB
TypeScript
80 lines
3.3 KiB
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
import {spawnSync} from 'child_process';
|
|
import {compareSync} from 'dir-compare';
|
|
import * as fs from 'fs-extra';
|
|
import minimist from 'minimist';
|
|
import npmlog from 'npmlog';
|
|
import * as path from 'path';
|
|
|
|
const ROOT = path.join(__dirname, '..', '..', '..');
|
|
const DEPS_ONNX = path.join(ROOT, 'cmake', 'external', 'onnx');
|
|
const TEST_DATA_ROOT = path.join(__dirname, '..', 'test', 'data');
|
|
const TEST_DATA_NODE = path.join(TEST_DATA_ROOT, 'node');
|
|
|
|
const TEST_DATA_OPSET_VERSIONS = [
|
|
['v13', 'rel-1.8.1'],
|
|
['v12', 'rel-1.7.0'],
|
|
['v11', 'rel-1.6.1'],
|
|
['v10', 'rel-1.5.0'],
|
|
['v9', 'rel-1.4.1'],
|
|
['v8', 'rel-1.3.0'],
|
|
['v7', 'rel-1.2.3'],
|
|
];
|
|
|
|
const args = minimist(process.argv);
|
|
// prepare test data only when eiter flag '-f' or '--force' is specified, or the folder does not exist.
|
|
if (args.f || args.force || !fs.existsSync(TEST_DATA_NODE)) {
|
|
npmlog.info('PrepareTestData', 'Preparing node tests ...');
|
|
fs.removeSync(TEST_DATA_NODE);
|
|
TEST_DATA_OPSET_VERSIONS.forEach(v => {
|
|
const version = v[0];
|
|
const commit = v[1];
|
|
npmlog.info('PrepareTestData', `Checking out deps/onnx ${commit}...`);
|
|
const checkout = spawnSync(`git checkout -q -f ${commit}`, {shell: true, stdio: 'inherit', cwd: DEPS_ONNX});
|
|
if (checkout.status !== 0) {
|
|
if (checkout.error) {
|
|
console.error(checkout.error);
|
|
}
|
|
process.exit(checkout.status === null ? undefined : checkout.status);
|
|
}
|
|
const from = path.join(DEPS_ONNX, 'onnx/backend/test/data/node');
|
|
const to = path.join(TEST_DATA_NODE, version);
|
|
npmlog.info('PrepareTestData', `Copying folders from "${from}" to "${to}"...`);
|
|
fs.copySync(from, to);
|
|
});
|
|
|
|
npmlog.info('PrepareTestData', 'Revert git index...');
|
|
const update = spawnSync(`git submodule update ${DEPS_ONNX}`, {shell: true, stdio: 'inherit', cwd: ROOT});
|
|
if (update.status !== 0) {
|
|
if (update.error) {
|
|
console.error(update.error);
|
|
}
|
|
process.exit(update.status === null ? undefined : update.status);
|
|
}
|
|
npmlog.info('PrepareTestData', 'Revert git index... DONE');
|
|
|
|
npmlog.info('PrepareTestData', 'Deduplicating test cases...');
|
|
for (let i = 0; i < TEST_DATA_OPSET_VERSIONS.length - 1; i++) {
|
|
const currentVersion = TEST_DATA_OPSET_VERSIONS[i][0];
|
|
const currentFolder = path.join(TEST_DATA_NODE, currentVersion);
|
|
const previousVersion = TEST_DATA_OPSET_VERSIONS[i + 1][0];
|
|
const previousFolder = path.join(TEST_DATA_NODE, previousVersion);
|
|
|
|
// compare each subfolder to its previous version. If they are same, remove the one in current version.
|
|
let count = 0;
|
|
fs.readdirSync(currentFolder, {withFileTypes: true}).forEach(dir => {
|
|
const currentDir = path.join(currentFolder, dir.name);
|
|
const previousDir = path.join(previousFolder, dir.name);
|
|
if (dir.isDirectory() && fs.existsSync(previousDir) && fs.statSync(previousDir).isDirectory()) {
|
|
if (compareSync(currentDir, previousDir, {compareContent: true}).differences === 0) {
|
|
fs.removeSync(currentDir);
|
|
count++;
|
|
}
|
|
}
|
|
});
|
|
npmlog.info('PrepareTestData', `Deduplicated ${count} test case(s) in folder ${currentVersion}.`);
|
|
}
|
|
npmlog.info('PrepareTestData', 'Deduplicating test cases... DONE');
|
|
}
|