Method select_capacity_for_run

src/framework/runtime/session.cpp:389–428

}

int64_t GraphCapacityController::select_capacity_for_run(
    const IGraphCapacityAdapter & adapter,
    int64_t request_size) const {
    if (request_size <= 0) {
        throw std::runtime_error("graph capacity request size must be positive");
    }
    const auto prepared_capacities = sorted_prepared_capacities(adapter);
    switch (mode_) {
    case GraphCapacityMode::Fixed: {
        const int64_t capacity = adapter.fixed_capacity();
        if (capacity <= 0) {
            throw std::runtime_error("fixed graph capacity must be positive");
        }
        if (request_size > capacity) {
            throw std::runtime_error("request exceeds fixed graph capacity");
        }
        return capacity;
    }
    case GraphCapacityMode::Tiered: {
        const int64_t prepared_capacity = select_smallest_prepared_fitting_capacity(prepared_capacities, request_size);
        if (prepared_capacity > 0) {
            return prepared_capacity;
        }
        return adapter.canonical_capacity_for_request(request_size);
    }
    case GraphCapacityMode::Grow:
        if (!prepared_capacities.empty() && prepared_capacities.back() >= request_size) {
            return prepared_capacities.back();
        }
        return adapter.canonical_capacity_for_request(request_size);
    case GraphCapacityMode::Double:
        if (!prepared_capacities.empty() && prepared_capacities.back() >= request_size) {
            return prepared_capacities.back();
        }
        return adapter.canonical_capacity_for_request(checked_double_capacity_request(request_size));
    case GraphCapacityMode::Unsupported:
        throw std::runtime_error("graph capacity mode is unsupported");
    }
    throw std::runtime_error("unknown graph capacity mode");
}

SessionPreparationRequest build_preparation_request(const AudioBuffer & audio) {
    SessionPreparationRequest request;