From 146e1caea314f4db24f7d2541eb3e70fdeb674b5 Mon Sep 17 00:00:00 2001 From: Ahmed Harmouche Date: Mon, 2 Dec 2024 15:48:44 +0100 Subject: [PATCH] Downgrade wgpu to prevent sd segfault (#7969) --- setup.py | 2 +- tinygrad/runtime/ops_webgpu.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index a9ec0db3ff..d7e2ac6b16 100644 --- a/setup.py +++ b/setup.py @@ -59,7 +59,7 @@ setup(name='tinygrad', "bottle", "ggml-python" ], - 'webgpu': ["wgpu>=v0.19.0"], + 'webgpu': ["wgpu==v0.18.1"], 'docs': [ "mkdocs", "mkdocs-material", diff --git a/tinygrad/runtime/ops_webgpu.py b/tinygrad/runtime/ops_webgpu.py index 0e61c2553a..5a2e33c014 100644 --- a/tinygrad/runtime/ops_webgpu.py +++ b/tinygrad/runtime/ops_webgpu.py @@ -58,8 +58,8 @@ class WebGpuAllocator(Allocator): class WebGpuDevice(Compiled): def __init__(self, device:str): - adapter = wgpu.gpu.request_adapter_sync(power_preference="high-performance") + adapter = wgpu.gpu.request_adapter(power_preference="high-performance") timestamp_supported = wgpu.FeatureName.timestamp_query in adapter.features - wgpu_device = adapter.request_device_sync(required_features=[wgpu.FeatureName.timestamp_query] if timestamp_supported else []) + wgpu_device = adapter.request_device(required_features=[wgpu.FeatureName.timestamp_query] if timestamp_supported else []) super().__init__(device, WebGpuAllocator(wgpu_device), WGSLRenderer(), Compiler(), functools.partial(WebGPUProgram, (wgpu_device, timestamp_supported)))