fix handcode_opt.py with DEBUG=2 (#5530)

only one ast per kernel now
This commit is contained in:
chenyu 2024-07-17 14:50:47 -04:00 committed by GitHub
commit 4193095f67
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -82,7 +82,7 @@ if __name__ == "__main__":
ops = get_lazyop_info(si.ast.src[0]).flops
if DEBUG >= 2:
for ast in si.ast: print_tree(ast)
print_tree(si.ast)
rawbufs = bufs_from_lin(Kernel(si.ast))