# RUN: %PYTHON %s

import iree.compiler.ir
from iree.compiler.dialects import iree_input as iree_d
from iree.compiler.dialects import iree_linalg_ext
from iree.compiler.dialects import iree_linalg_transform

with iree.compiler.ir.Context() as ctx:
    iree_d.register_dialect()
    iree_linalg_ext.register_dialect()
    iree_linalg_transform.register_dialect()