router: broaden except to catch all CuTeDSL errors, fall through to cuBLAS+activation_topk path

2026-05-31 23:54:16 +00:00
parent 3b5b9f487c
commit 5396a04c28
1 changed files with 1 additions and 1 deletions
--- 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(