mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-29 23:06:41 +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 --------- |
||
|---|---|---|
| .. | ||
| backend-impl.ts | ||
| backend.ts | ||
| env-impl.ts | ||
| env.ts | ||
| index.ts | ||
| inference-session-impl.ts | ||
| inference-session.ts | ||
| onnx-value.ts | ||
| tensor-conversion-impl.ts | ||
| tensor-conversion.ts | ||
| tensor-factory-impl.ts | ||
| tensor-factory.ts | ||
| tensor-impl.ts | ||
| tensor-utils-impl.ts | ||
| tensor-utils.ts | ||
| tensor.ts | ||
| version.ts | ||