mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
b29849a287
### Description Fix a few warnings in typedoc (for generating JS API): ``` [warning] The signature TrainingSession.loadParametersBuffer has an @param with name "buffer", which was not used. [warning] NonTensorType, defined in ./lib/onnx-value.ts, is referenced by OnnxValue but not included in the documentation. [warning] TensorFactory, defined in ./lib/tensor-factory.ts, is referenced by Tensor but not included in the documentation. [warning] ExternalDataFileType, defined in ./lib/onnx-model.ts, is referenced by InferenceSession.SessionOptions.externalData but not included in the documentation. [warning] TensorToDataUrlOptions, defined in ./lib/tensor-conversion.ts, is referenced by Tensor.toDataURL.toDataURL.options but not included in the documentation. [warning] TensorToImageDataOptions, defined in ./lib/tensor-conversion.ts, is referenced by Tensor.toImageData.toImageData.options but not included in the documentation. [warning] Failed to resolve link to "GpuBufferType" in comment for Env.WebGpuFlags.adapter. [warning] Failed to resolve link to "GpuBufferType" in comment for Env.WebGpuFlags.device. ``` Changes highlighted: - Merge `CoreMlExecutionProviderOption` and `CoreMLExecutionProviderOption`. They expose 2 set of different options for React-native and ORT nodejs binding. This should be fixed in future. - Fix a few inconsistency of names between JSDoc and parameters - Fix broken type links - Exclude trace functions
53 lines
1.3 KiB
TypeScript
53 lines
1.3 KiB
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
import {env} from './env-impl.js';
|
|
|
|
/**
|
|
* @ignore
|
|
*/
|
|
export const TRACE = (deviceType: string, label: string) => {
|
|
if (typeof env.trace === 'undefined' ? !env.wasm.trace : !env.trace) {
|
|
return;
|
|
}
|
|
// eslint-disable-next-line no-console
|
|
console.timeStamp(`${deviceType}::ORT::${label}`);
|
|
};
|
|
|
|
const TRACE_FUNC = (msg: string, extraMsg?: string) => {
|
|
const stack = new Error().stack?.split(/\r\n|\r|\n/g) || [];
|
|
let hasTraceFunc = false;
|
|
for (let i = 0; i < stack.length; i++) {
|
|
if (hasTraceFunc && !stack[i].includes('TRACE_FUNC')) {
|
|
let label = `FUNC_${msg}::${stack[i].trim().split(' ')[1]}`;
|
|
if (extraMsg) {
|
|
label += `::${extraMsg}`;
|
|
}
|
|
TRACE('CPU', label);
|
|
return;
|
|
}
|
|
if (stack[i].includes('TRACE_FUNC')) {
|
|
hasTraceFunc = true;
|
|
}
|
|
}
|
|
};
|
|
|
|
/**
|
|
* @ignore
|
|
*/
|
|
export const TRACE_FUNC_BEGIN = (extraMsg?: string) => {
|
|
if (typeof env.trace === 'undefined' ? !env.wasm.trace : !env.trace) {
|
|
return;
|
|
}
|
|
TRACE_FUNC('BEGIN', extraMsg);
|
|
};
|
|
|
|
/**
|
|
* @ignore
|
|
*/
|
|
export const TRACE_FUNC_END = (extraMsg?: string) => {
|
|
if (typeof env.trace === 'undefined' ? !env.wasm.trace : !env.trace) {
|
|
return;
|
|
}
|
|
TRACE_FUNC('END', extraMsg);
|
|
};
|