Function removeEmptyDimensions

src/backend/common/jit/Node.hpp:377–414

/// Removes empty columns from output and the other node pointers in \p nodes
template<typename ParamT, typename BufferNodeT, typename ShiftNodeT>
void removeEmptyDimensions(nonstd::span<ParamT> outputs,
                           nonstd::span<Node_ptr> nodes) {
    dim_t *outDims{outputs[0].dims_ptr()};
    dim_t *outStrides{outputs[0].strides_ptr()};
    auto shifts = compressArray(outDims);
    applyShifts<dim_t>(shifts, {outStrides, AF_MAX_DIMS});
    for (auto nodeIt{begin(nodes)}, endIt{end(nodes)};
         (nodeIt = find_if(nodeIt, endIt, isBufferOrShift)) != endIt;
         ++nodeIt) {
        switch ((*nodeIt)->getNodeType()) {
            case kNodeType::Buffer: {
                BufferNodeT *buf{static_cast<BufferNodeT *>(nodeIt->get())};
                applyShifts<dim_t>(shifts,
                                   {buf->m_param.dims_ptr(), AF_MAX_DIMS});
                applyShifts<dim_t>(shifts,
                                   {buf->m_param.strides_ptr(), AF_MAX_DIMS});
            } break;
            case kNodeType::Shift: {
                ShiftNodeT &shiftNode{
                    *static_cast<ShiftNodeT *>(nodeIt->get())};
                BufferNodeT &buf{shiftNode.getBufferNode()};
                applyShifts<dim_t>(shifts,
                                   {buf.m_param.dims_ptr(), AF_MAX_DIMS});
                applyShifts<dim_t>(shifts,
                                   {buf.m_param.strides_ptr(), AF_MAX_DIMS});

                auto &node_shifts = shiftNode.getShifts();
                applyShifts<int>(shifts, node_shifts);
            } break;
            default: break;
        }
    }
    std::for_each(
        std::begin(outputs) + 1, std::end(outputs), [&shifts](ParamT &output) {
            applyShifts<dim_t>(shifts, {output.dims_ptr(), AF_MAX_DIMS});
            applyShifts<dim_t>(shifts, {output.strides_ptr(), AF_MAX_DIMS});
        });
}

}  // namespace common
}  // namespace arrayfire