From 5396a04c28b74025897113eedd5db7ba2c8fa298 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sun, 31 May 2026 23:54:16 +0000
Subject: [PATCH] router: broaden except to catch all CuTeDSL errors, fall
 through to cuBLAS+activation_topk path

---
 dsv4/kernels/router/dense_router_decode.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dsv4/kernels/router/dense_router_decode.py b/dsv4/kernels/router/dense_router_decode.py
index a24b581b..dda5ae88 100644
--- a/dsv4/kernels/router/dense_router_decode.py
+++ b/dsv4/kernels/router/dense_router_decode.py
@@ -33,7 +33,7 @@ def dense_router_dispatch(
                 out_weights, out_ids,
             )
             return
-        except (ImportError, NotImplementedError):
+        except Exception:
             pass  # fall through to prefill path
 
     _run_prefill_path(