Skip to content

vulkan support for nodejs #1309

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ali-ghamdan opened this issue May 5, 2025 · 2 comments
Open

vulkan support for nodejs #1309

ali-ghamdan opened this issue May 5, 2025 · 2 comments
Labels
enhancement New feature or request

Comments

@ali-ghamdan
Copy link

Feature request

it looks like that transformers.js only can run on cuda if linux on nvidia and if there is an internal gpu, the cpu will only be used.

Motivation

Vulcan will allow to a lot of devices to leave the CPU speed, we all know that Vulkan will make it faster.

Your contribution

sorry but no.

@ali-ghamdan ali-ghamdan added the enhancement New feature or request label May 5, 2025
@xenova
Copy link
Collaborator

xenova commented May 5, 2025

Hi there 👋 Vulkan support for Node.js may be available with the Native WebGPU EP, which will be released with Transformers.js soon (PR for tracking: #1306).

cc @fs-eire to confirm?

@fs-eire
Copy link
Contributor

fs-eire commented May 5, 2025

no. currently WebGPU EP is only enabled in Windows and macOS in Node.js binding. There are multiple reasons:

  • the current release pipeline does not yet support CUDA + WebGPU in the same build
  • WebGPU EP is not well tested under linux and/or vulkan. For example, f16 feature is disabled for all NVidia GPUs under vulkan, because of CTS tests are not passing.

WebGPU EP still works on linux with vulkan but the current status is it's not ready out of the box for linux. you can build it from source.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

3 participants