Force CPU

tracel-ai · Jan 5, 2024 · 01a71ae · 01a71ae
1 parent 3907ea6
commit 01a71ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/burn-wgpu/src/compute/base.rs b/burn-wgpu/src/compute/base.rs
@@ -242,7 +242,7 @@ fn select_adapter<G: GraphicsApi>(device: &WgpuDevice) -> wgpu::Adapter {
                         // often a Discrete Gpu.
                         DeviceType::IntegratedGpu => 3,
                         DeviceType::VirtualGpu => 2,
-                        DeviceType::Cpu => 1,
+                        DeviceType::Cpu => 6,
                     };
                     println!("dbg 8_15");
                     if score > current_score {