Cleanup error handling

tracel-ai · Nov 27, 2024 · 0903201 · 0903201
1 parent 341963a
commit 0903201
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/crates/cubecl-linalg/src/matmul/base.rs b/crates/cubecl-linalg/src/matmul/base.rs
@@ -59,8 +59,19 @@ pub fn launch_ref<R: Runtime, EG: Float>(
         }
         Strategy::Simple => simple::launch_ref::<R, EG>(client, lhs, rhs, out),
         Strategy::Auto => {
-            if matmul::launch_ref::<R, EG>(client, lhs, rhs, out, false).is_err() {
-                tiling2d::launch_ref::<R, EG>(client, lhs, rhs, out, Tiling2dConfig::default())
+            if let Err(err) = matmul::launch_ref::<R, EG>(client, lhs, rhs, out, false) {
+                match err {
+                    super::kernels::MatmulLaunchError::Unavailable(_) => {
+                        tiling2d::launch_ref::<R, EG>(
+                            client,
+                            lhs,
+                            rhs,
+                            out,
+                            Tiling2dConfig::default(),
+                        )
+                    }
+                    _ => panic!("{err:?}"),
+                }
             }
         }
     };