Function evalMultiple

src/backend/cpu/Array.cpp:162–200

template<typename T>
void evalMultiple(vector<Array<T> *> array_ptrs) {
    vector<Array<T> *> outputs;
    vector<common::Node_ptr> nodes;
    vector<Param<T>> params;
    if (getQueue().is_worker()) {
        AF_ERROR("Array not evaluated", AF_ERR_INTERNAL);
    }

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

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

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

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

        outputs.push_back(array);
        params.emplace_back(array->getData().get(), array->dims(),
                            array->strides());
        nodes.push_back(array->node);
    }

    if (params.empty()) return;

    getQueue().enqueue(cpu::kernel::evalMultiple<T>, params, nodes);

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

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