Function dry_run

kleidiai-examples/audiogen-et/app/main.cpp:201–229

}

static void dry_run(std::unique_ptr<executorch::extension::Module>& module) {

    // Dummy run for a module
    // -------------------------
    auto module_forward_meta_res = module->method_meta("forward");
    AUDIOGEN_CHECK(module_forward_meta_res.ok());

    auto module_forward_meta = module_forward_meta_res.get();
    size_t module_num_inputs = module_forward_meta.num_inputs();

    // To keep the created tensors inside the loop alive
    std::vector<std::shared_ptr<Tensor>> allocated_tensors;

    std::vector<executorch::runtime::EValue> module_inputs(module_num_inputs);
    for (size_t i = 0; i < module_num_inputs; ++i) {
        auto input_tensor_meta = module_forward_meta.input_tensor_meta(i).get();
        auto input_tensor_dims = get_tensor_dims(input_tensor_meta);
        auto input_tensor_scalar_type = input_tensor_meta.scalar_type();
        auto input_tensor = input_tensor_scalar_type == ScalarType::Float ? randn(input_tensor_dims, input_tensor_scalar_type) : randint(1, 100, input_tensor_dims, input_tensor_scalar_type);
        allocated_tensors.push_back(input_tensor);
        module_inputs[i] = input_tensor;
    }

    auto module_output = module->forward(module_inputs);
    if(!module_output.ok()) {
        ET_LOG(Error, "Failed to run module forward");
        exit(EXIT_FAILURE);
    }
}

int main(int32_t argc, char** argv) {