github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-16 01:26:29 -05:00
Code Issues Packages Projects Releases Wiki Activity

wgsl and ops_webgpu simplifications [pr] (#8182)

Browse Source 
Simplify wgsl and ops_webgpu
This commit is contained in:
Ahmed Harmouche
2024-12-12 14:21:58 +01:00
committed by GitHub
parent d9a0880d33
commit 2f2b1e792c
2 changed files with 12 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/runtime/ops_webgpu.py
View File

@@ -7,8 +7,7 @@ import struct
def create_uniform(wgpu_device, val) -> wgpu.GPUBuffer:
buf = wgpu_device.create_buffer(size=4, usage=wgpu.BufferUsage.UNIFORM | wgpu.BufferUsage.COPY_DST)
if isinstance(val, int): wgpu_device.queue.write_buffer(buf, 0, val.to_bytes(4, "little"))
else: wgpu_device.queue.write_buffer(buf, 0, struct.pack('<f', val))
wgpu_device.queue.write_buffer(buf, 0, val.to_bytes(4, "little") if isinstance(val, int) else struct.pack('<f', val))
return buf
class WebGPUProgram: