don't require contiguous after fuse (#10074)

2025-04-27 13:17:22 -04:00
parent 1aed04ec12
commit 732e172961
1 changed files with 1 additions and 1 deletions
@@ -2750,7 +2750,7 @@ class Tensor(SimpleMathTrait):
    Useful for single kernel softmax and flash attention.
    Careful, this can break codegen or make kernels really slow.
    """
-    return self._apply_uop(UOp.fuse).contiguous()
+    return self._apply_uop(UOp.fuse)

  def contiguous_backward(self) -> Tensor:
    """