Function evalMultiple

src/backend/cuda/Array.cpp:202–237

template<typename T>
void evalMultiple(std::vector<Array<T> *> arrays) {
    vector<Param<T>> output_params;
    vector<Array<T> *> output_arrays;
    vector<Node *> nodes;

    // Check if all the arrays have the same dimension
    auto it = std::adjacent_find(begin(arrays), end(arrays),
                                 [](const Array<T> *l, const Array<T> *r) {
                                     return l->dims() != r->dims();
                                 });

    // If they are not the same. eval individually
    if (it != end(arrays)) {
        for (auto ptr : arrays) { ptr->eval(); }
        return;
    }

    for (Array<T> *array : arrays) {
        if (array->isReady()) { continue; }

        array->setId(getActiveDeviceId());
        array->data =
            shared_ptr<T>(memAlloc<T>(array->elements()).release(), memFree);

        output_params.emplace_back(array->getData().get(), array->dims().get(),
                                   array->strides().get());
        output_arrays.push_back(array);
        nodes.push_back(array->getNode().get());
    }

    if (output_params.empty()) return;

    evalNodes(output_params, nodes);

    for (Array<T> *array : output_arrays) { array->node.reset(); }
}

template<typename T>
Node_ptr Array<T>::getNode() {