Print message to inform user an error is handled gracefully.

tracel-ai · Aug 17, 2024 · 76df6e4 · 76df6e4
1 parent 3fec002
commit 76df6e4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/crates/cubecl-runtime/src/tune/tuner.rs b/crates/cubecl-runtime/src/tune/tuner.rs
@@ -122,7 +122,10 @@ impl<K: AutotuneKey> Tuner<K> {
             catch_unwind(AssertUnwindSafe(|| {
                 TuneBenchmark::new(operation, client.clone()).run()
             }))
-            .map_err(ManuallyDrop::new)
+            .map_err(|e| {
+                println!("Caught error while benchmarking, falling back to next operation.");
+                ManuallyDrop::new(e)
+            })
         }
         #[cfg(not(feature = "std"))]
         Ok(TuneBenchmark::new(operation, client.clone()).run())