tilelang.language.v2.utils

Attributes

CompileMethod

Functions

disk_compile(source, name)
get_func_nonlocals(func)	A modified version of inspect.getclosurevars
inspect_function_capture(func)	Capture function non-locals and global variables.
get_ast(func)
get_compiled_object(source, name[, filename, globals])

Module Contents

tilelang.language.v2.utils.disk_compile(source, name)

tilelang.language.v2.utils.get_func_nonlocals(func)

A modified version of inspect.getclosurevars

tilelang.language.v2.utils.inspect_function_capture(func)

Capture function non-locals and global variables.

Parameters:

func (Callable) – The function to inspect.

Returns:

res – The function variables map with non-local or global variables.

Return type:

Dict[str, Any]

tilelang.language.v2.utils.get_ast(func)

Parameters:

func (Callable)

tilelang.language.v2.utils.CompileMethod

tilelang.language.v2.utils.get_compiled_object(source, name, filename=None, globals=None)

Parameters:

• source (str | ast.AST)

• name (str)

• filename (str)

• globals (dict[str, Any])