mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-25 22:26:24 +00:00
c262879214
### Description I've added changes to support CUDA and DML (only on Windows, on other platforms it will throw an error) ### Motivation and Context It fixes this feature request https://github.com/microsoft/onnxruntime/issues/14127 which is tracked here https://github.com/microsoft/onnxruntime/issues/14529 I was working on StableDiffusion implementation for node.js and it is very slow on CPU, so GPU support is essential. Here is a working demo with a patched and precompiled version https://github.com/dakenf/stable-diffusion-nodejs ---------
15 lines
510 B
TypeScript
15 lines
510 B
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
export * from 'onnxruntime-common';
|
|
export {listSupportedBackends} from './backend';
|
|
import {registerBackend, env} from 'onnxruntime-common';
|
|
import {version} from './version';
|
|
import {onnxruntimeBackend, listSupportedBackends} from './backend';
|
|
|
|
const backends = listSupportedBackends();
|
|
for (const backend of backends) {
|
|
registerBackend(backend.name, onnxruntimeBackend, 100);
|
|
}
|
|
|
|
env.versions.node = version;
|