From 732e17296172e1d6dfcfe343e053b271fa820472 Mon Sep 17 00:00:00 2001 From: George Hotz <72895+geohot@users.noreply.github.com> Date: Sun, 27 Apr 2025 13:17:22 -0400 Subject: [PATCH] don't require contiguous after fuse (#10074) --- tinygrad/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index a508dd3944..e392f5ab49 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -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: """