add AST dump

tenstorrent · Jan 8, 2025 · 6c2258b · 6c2258b
1 parent 17e52ef
commit 6c2258b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/python/simple_kernel.py b/test/python/simple_kernel.py
@@ -237,6 +237,7 @@ def _wrapper(*args, **kwargs):
         arg_dtypes = [to_data_type(arg.dtype) for arg in args]
         m = ast.parse(inspect.getsource(f))
         b = TTKernelBuilder(f.__name__, arg_shapes, arg_dtypes)
+        # print(ast.dump(m, indent=4))
         b.visit(m)
         print(b.module)
         # return f(*args, **kwargs)