def METHOD_NAME(
    module, num_vitis_ai_modules=1, params=None, target="llvm", tvm_ops=0, dpu_target="DPUCADX8G"
):
    """Check Vitis-AI codegen against a known good output."""
    module = build_module(
        module,
        target,
        params=params,
        dpu_target=dpu_target,
        tvm_ops=tvm_ops,
        vitis_ai_partitions=num_vitis_ai_modules,
    )
    vitis_ai_modules = extract_vitis_ai_modules(module)
    assert len(vitis_ai_modules) == num_vitis_ai_modules, (
        f"The number of Vitis-AI modules produced ({len(vitis_ai_modules)}) does not "
        f"match the expected value ({num_vitis_ai_modules})."
    )