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

This commit is contained in:
chenyu
2025-02-15 09:12:51 -05:00
committed by GitHub
parent 2e19976d03
commit d129ccda4c

View File

@@ -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)