From 65af5d68970518e02cb1fdc753429d9f66b3efe6 Mon Sep 17 00:00:00 2001 From: Bill Teng Date: Wed, 8 Jan 2025 22:21:41 +0000 Subject: [PATCH] add AST dump --- test/python/simple_kernel.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/python/simple_kernel.py b/test/python/simple_kernel.py index f8ca2260ef..82c334e9bc 100644 --- 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)