Set DEFAULT_MAX_TASKS to 1 when running tests

tracel-ai · Jul 5, 2024 · d6efb3c · d6efb3c
1 parent 882a27c
commit d6efb3c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/burn-wgpu/src/runtime.rs b/crates/burn-wgpu/src/runtime.rs
@@ -79,6 +79,9 @@ pub struct RuntimeOptions {
 
 impl Default for RuntimeOptions {
     fn default() -> Self {
+        #[cfg(test)]
+        const DEFAULT_MAX_TASKS: usize = 1;
+        #[cfg(not(test))]
         const DEFAULT_MAX_TASKS: usize = 16;
 
         let tasks_max = match std::env::var("BURN_WGPU_MAX_TASKS") {