add RAWAST back to DEBUG=3 [pr] (#9107)

2025-02-15 09:12:51 -05:00
parent 2e19976d03
commit d129ccda4c
1 changed files with 1 additions and 0 deletions
@@ -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)