From 5d522ece898d4b010c862238479388720be7223c Mon Sep 17 00:00:00 2001
From: Ard <88337265+aw471@users.noreply.github.com>
Date: Thu, 22 Oct 2026 16:09:06 -0700
Subject: [PATCH] Avoid symbolic contiguity checks on compiled output buffers

---
 sdm/nn/attention.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/sdm/nn/attention.py b/sdm/nn/attention.py
index 3d0275119..199d8376f 100644
--- a/sdm/nn/attention.py
+++ b/sdm/nn/attention.py
@@ -812,11 +812,7 @@ def _forward(
         if self.post_attn_norm is not None:
             attn_out = self.post_attn_norm(attn_out)
 
-        if (
-            out is not None
-            and torch.compiler.is_compiling()
-            and not out.is_contiguous()
-        ):
+        if out is not None and torch.compiler.is_compiling():
             tmp = (attn_out + query).to(out.dtype)
             if self.mlp is not None:
                 tmp += self.mlp(tmp)