Function create_bundled_program

backends/vulkan/test/utils.py:674–714

Create a bundled program containing the model and test cases for correctness testing. Args: executorch_program: The ExecutorchProgramManager to bundle sample_inputs: Sample inputs for the model expected_outputs: Expected outputs from running the model with sample_in

(
    executorch_program: ExecutorchProgramManager,
    sample_inputs: Tuple[torch.Tensor, ...],
    expected_outputs: List[Any],
    method_name: str = "forward",
)

def create_bundled_program(
    executorch_program: ExecutorchProgramManager,
    sample_inputs: Tuple[torch.Tensor, ...],
    expected_outputs: List[Any],
    method_name: str = "forward",
) -> bytes:
    """
    Create a bundled program containing the model and test cases for correctness testing.

    Args:
        executorch_program: The ExecutorchProgramManager to bundle
        sample_inputs: Sample inputs for the model
        expected_outputs: Expected outputs from running the model with sample_inputs
        method_name: Name of the method to test (default: "forward")

    Returns:
        Serialized bundled program as bytes
    """
    # Flatten sample inputs to match expected format
    inputs_flattened, _ = tree_flatten(sample_inputs)

    # Create test suite with the sample inputs and expected outputs
    test_suites = [
        MethodTestSuite(
            method_name=method_name,
            test_cases=[
                MethodTestCase(
                    inputs=inputs_flattened,
                    expected_outputs=expected_outputs,
                )
            ],
        )
    ]

    # Create bundled program
    bundled_program = BundledProgram(executorch_program, test_suites)

    # Serialize to flatbuffer
    bundled_buffer = serialize_from_bundled_program_to_flatbuffer(bundled_program)

    return bundled_buffer


def save_bundled_program(