iree.turbine.tools

class iree.turbine.tools.Interpreter(workgroup_ids: list[int], thread_ids: list[int])

Python interpreter for MLIR. Uses torch for tensor operations.

callback(op: Operation) → None

get_dtype(dtype)

interpret(asm: str) → None

static interpret_ndrange(asm: str, workgroup_count: list[int], workgroup_size: list[int])

walk_operations(operation: Operation, callback: Callable) → None