diff --git a/dsv4/layers/mhc.py b/dsv4/layers/mhc.py index 88486589..eb95a2a9 100644 --- a/dsv4/layers/mhc.py +++ b/dsv4/layers/mhc.py @@ -99,7 +99,8 @@ def sinkhorn_knopp( from dsv4.kernels.cuda.loader import get_cuda_module mod = get_cuda_module("mhc_sinkhorn", ["mhc_sinkhorn.cu"]) return mod.mhc_sinkhorn(logits.float(), t_max, eps) - except Exception: + except Exception as e: + import sys; print(f"mhc_sinkhorn CUDA kernel failed: {e}, falling back to Python", file=sys.stderr, flush=True) pass # Fall back to Python # Python fallback