Function _parse_and_assert_equal

python/tests/kernel/test_kernel_signature.py:58–77

Parse the given callable via both AST and MLIR routes, assert that the resulting signatures are identical, and return the signature for further assertions.

(fn)

def _parse_and_assert_equal(fn):
    """
    Parse the given callable via both AST and MLIR routes,
    assert that the resulting signatures are identical,
    and return the signature for further assertions.
    """
    # Parse via AST
    sig_a = KernelSignature.parse_from_ast(_get_ast_from_callable(fn),
                                           fn.__name__)
    # replace None with empty list for comparision with MLIR
    sig_a.captured_args = []

    # Parse via MLIR
    decorated = cudaq.kernel(fn)
    sig_m = KernelSignature.parse_from_mlir(decorated.qkeModule,
                                            decorated.uniqName)

    assert sig_a == sig_m

    return sig_m


def _make_kernel_ast(arg_types=None, return_type=None) -> ast.Module: