-
Notifications
You must be signed in to change notification settings - Fork 4
Open
Description
Lets make it possible to test WebGPU accelerated code via scripts, e.g. when running: https://github.com/PAIR-code/tiny-transformers/blob/main/animated-transformer/src/lib/seqtasks/tiny_worlds_train_self_att.script.ts
This may help us debug memory leaks too.
Breadcrumbs:
General TFJS platform guide: https://www.tensorflow.org/js/guide/platform_environment
WebGPU support for tfjs:
TFJS Node support (references GPU support too, but for linux and windows, not yet mac)
Key issues for Deno (looks resolved... uses tfjs-backend-webgpu and seems to use Chrome GPU backend support):
- WebGPU adapter.info call fails after device creation denoland/deno#25874 : WebGPU adapter.info call fails after device creation [closed]
- GPUAdapter invalidation causes TensorFlow.js WebGPU backend to fail — Follow-up to #22029 denoland/deno#28135 : GPUAdapter invalidation causes TensorFlow.js WebGPU backend to fail — Follow-up to [closed, but some discussion on if it can already be used...]
Possible alternative paths... webgpu for node and then make tfjs-node-gpu
- https://www.npmjs.com/package/webgpu & discussion about it in nodejs forum: https://github.com/orgs/nodejs/discussions/41994
Metadata
Metadata
Assignees
Labels
No labels