Function run_cpp_test_runner

backends/mlx/test/test_utils.py:601–656

Run the C++ op_test_runner binary. Args: pte_path: Path to the .pte model file. input_path: Path to input .bin file. output_path: Path to write output .bin file. verbose: Whether to print verbose output. timeout: Timeout in seconds. None means use DE

(
    pte_path: Path,
    input_path: Path,
    output_path: Path,
    verbose: bool = False,
    timeout: Optional[int] = None,
)

def run_cpp_test_runner(
    pte_path: Path,
    input_path: Path,
    output_path: Path,
    verbose: bool = False,
    timeout: Optional[int] = None,
) -> bool:
    """
    Run the C++ op_test_runner binary.

    Args:
        pte_path: Path to the .pte model file.
        input_path: Path to input .bin file.
        output_path: Path to write output .bin file.
        verbose: Whether to print verbose output.
        timeout: Timeout in seconds. None means use DEFAULT_TEST_TIMEOUT.

    Returns:
        True if execution succeeded, False otherwise.
    """
    if timeout is None:
        timeout = DEFAULT_TEST_TIMEOUT

    runner = find_op_test_runner()

    cmd = [
        str(runner),
        "--pte",
        str(pte_path),
        "--input",
        str(input_path),
        "--output",
        str(output_path),
    ]
    if verbose:
        cmd.append("--verbose")

    print(f"Running: {' '.join(cmd)}")
    try:
        result = subprocess.run(
            cmd,
            capture_output=True,
            text=True,
            timeout=timeout,
        )
    except subprocess.TimeoutExpired:
        print(f"TIMEOUT: C++ runner exceeded {timeout}s timeout")
        return False

    if result.returncode != 0:
        print(f"FAILED: {result.stderr}")
        print(f"stdout: {result.stdout}")
        return False

    print(f"C++ binary output: {result.stdout.strip()}")
    return True