Fixed wasm32 non atomic

Author: SarthakSingh31

crates/cubecl-wgpu/src/compute/server.rs

@@ -25,6 +25,7 @@ pub struct WgpuServer<C: WgpuCompiler> {
     memory_management: MemoryManagement<WgpuStorage>,
     pub(crate) device: Pdrc<wgpu::Device>,
     queue: Pdrc<wgpu::Queue>,
+    #[allow(unused)]
     pub(crate) adapter: Pdrc<wgpu::Adapter>,
     encoder: CommandEncoder,
     current_pass: Option<ComputePass<'static>>,

crates/cubecl-wgpu/src/runtime.rs

@@ -7,7 +7,11 @@ use crate::{
     compute::{WgpuServer, WgpuStorage},
     AutoGraphicsApi, GraphicsApi, Pdrc, WgpuDevice,
 };
-use cubecl_core::{channel::ComputeChannel, server::ComputeServer, Feature, Runtime};
+#[cfg(not(all(target_arch = "wasm32", target_feature = "atomics")))]
+use cubecl_core::future;
+#[cfg(all(target_arch = "wasm32", target_feature = "atomics"))]
+use cubecl_core::{channel::ComputeChannel, server::ComputeServer};
+use cubecl_core::{Feature, Runtime};
 #[cfg(not(all(target_arch = "wasm32", target_feature = "atomics")))]
 use cubecl_runtime::channel::MutexComputeChannel;
 use cubecl_runtime::{
@@ -31,6 +35,7 @@ thread_local! {
     static LOCAL_DEVICE: RefCell<hashbrown::HashMap<WgpuDevice, Rc<RefCell<Server>>>> = RefCell::new(hashbrown::HashMap::default());
 }
 
+#[cfg(all(target_arch = "wasm32", target_feature = "atomics"))]
 static RUNTIME: ComputeRuntime<WgpuDevice, Server, ThreadLocalChannel> = ComputeRuntime::new();
 
 /// The compute instance is shared across all [wgpu runtimes](WgpuRuntime).
@@ -214,6 +219,7 @@ pub(crate) fn create_client_on_setup<C: WgpuCompiler>(
         memory_management,
         setup.device.clone(),
         setup.queue,
+        setup.adapter.clone(),
         options.tasks_max,
     );
     let channel = MutexComputeChannel::new(server);