From 9bb5c7d3dfe6ad825cc6fdac1a62bdcd4ad1d39f Mon Sep 17 00:00:00 2001
From: blepping <blepping@tuta.io>
Date: Wed, 17 Sep 2025 13:17:46 -0600
Subject: [PATCH] Ensure the attention override returns a contiguous tensor.

---
 py/nodes/sageAttention.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/py/nodes/sageAttention.py b/py/nodes/sageAttention.py
index 755e0fd..af1aa46 100644
--- a/py/nodes/sageAttention.py
+++ b/py/nodes/sageAttention.py
@@ -155,7 +155,7 @@ def attention_bleh(  # noqa: PLR0914
         result = result.transpose(1, 2)
     if not skip_output_reshape:
         result = result.reshape(batch, -1, heads * dim_head)
-    return result
+    return result.contiguous()
 
 
 def copy_funattrs(fun, dest=None):