Can I get webgpu capabilities through nodejs?

[wewindy]

As the title.

// main.js
const entryFunc = async () => {
  const adapter = await navigator.gpu.requestAdapter()
  const device = await adapter.requestDevice()

  console.log(device.limits)
  // ...
}

node ./main.js

[targos]

WebGPU is not implemented in Node.js.

Maybe we could integrate it using the C++ library that is being made for Chromium: https://dawn.googlesource.com/dawn

[wewindy]

Well so, is there any plan to implement webgpu in nodejs?

[targos]

Not for now.

[wewindy]

Thank you!

[richardlau]

You could try Electron electron/electron#26944

[robertleeplummerjr]

https://github.com/maierfelix/webgpu could be of use. Perhaps we could ask to turn into organization and continue support?:

[axinging]

Based on dawn's nodejs, I sucessfully run some examples on Windows/Linux:
Example: webgpu on node
https://github.com/axinging/webgpu-node-demo/tree/main/external-node

Example: tfjs-webgpu on node
https://github.com/axinging/webgpu-node-demo/tree/main/tfjs-on-external-node

[ovflowd]

@targos could we add these bindings to the feature requests board of Node.js? Feels like something that could be added im the future of Node.js

[bnoordhuis]

There's #42896 already but it was closed due to inactivity.

WebGPU is still niche enough at this point that we can punt on it for another year and see what the uptake is.

[loretoparisi]

Why wait considering WebGPU is now in beta in Chrome and the raise of recent Chat LLM that can run directly in JavaScript thanks to the WebLLM project?
Let's also consider the recent interest in a TypeScript/JavaScript pure Tensor library.

[earonesty]

no longer beta. it's live in chrome.

[benjamingr]

There are official Node.js bindinging by the Chromium team IIRC

[benjamingr]

https://dawn.googlesource.com/dawn/+/refs/heads/main/src/dawn/node/ found it, looks like it's under active development

[benjamingr]

Also cc @Kangz

[Kangz]

The Dawn team uses these node bindings for its own testing so we maintain them. They aren't exactly production ready yet, but should be pretty decent. The goal is to eventually have automated builds for them that are published on NPM but we didn't do that yet.

[kungfooman]

Any update for 2024? This is highly interesting for neural network libraries (currently testing diffusion.js in browser). Would be lovely to have the same API and have it working in nodejs aswell.

[greggman]

There's a build of dawn.node here

https://npmjs.org/webgpu