huggingface/ratchet

Minimum number of copies from `Uint8Array` to `wgpu::Buffer`

Aperta

#171 aperta il 18 apr 2024

 (4 commenti) (1 reazione) (0 assegnatari)Rust (44 fork)auto 404
help wanted

Metriche repository

Star
 (768 stelle)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

Currently in wgpu, there is no way to create a wgpu::Buffer directly from Uint8Array.

This means that our tensor data does something like this:

IndexedDB -> Uint8Array -> Vec<u8> ~~> Uint8Array -> WebGPU

Judging by this: https://developer.mozilla.org/en-US/docs/Web/API/GPUBuffer

I think the optimal is:

IndexedDB -> Uint8Array -> WebGPU

Reference: https://github.com/gfx-rs/wgpu/blob/trunk/wgpu/src/backend/webgpu.rs#L1821

Guida contributor