diff --git a/tinygrad/codegen/kernel.py b/tinygrad/codegen/kernel.py index 64ad7226a0..c1718811ee 100644 --- a/tinygrad/codegen/kernel.py +++ b/tinygrad/codegen/kernel.py @@ -669,6 +669,7 @@ class Kernel: if DEBUG >= 3: print(self.name) + if getenv("RAWAST"): print(self.ast) for i,(buf,st) in enumerate([(buf,st) for buf,st in zip(self.bufs, self.sts) if buf.op not in {Ops.CONST, Ops.VALID}]): print(f"{i:2d}: {str(st.shape):25s} {str(buf.src[0].dtype).replace('dtypes.',''):20s}", st.real_strides()) print(self.applied_opts)