Method configure

src/gpu/cl/kernels/ClScatterKernel.cpp:114–210

}

void ClScatterKernel::configure(const ClCompileContext &compile_context,
                                const ITensorInfo      *updates,
                                const ITensorInfo      *indices,
                                ITensorInfo            *dst,
                                const ScatterInfo      &info)
{
    ARM_COMPUTE_ERROR_ON_NULLPTR(updates, dst, indices);
    ARM_COMPUTE_LOG_PARAMS(updates, indices, dst, info);

    const TensorShape &dst_shape = dst->tensor_shape();
    const int          index_len = indices->dimension(0);

    // Check for single element data block
    const bool is_scalar_block = (dst->num_dimensions() == static_cast<uint32_t>(index_len));

    const int n0         = adjust_vec_size(16 / updates->element_size(), is_scalar_block ? 1 : updates->dimension(0));
    const int partial_n0 = updates->dimension(0) % n0;

    // The GWS will be 2D [x, y]
    //  x-dimension refers to the x coordinate of the dst tensor
    //  y-dimension refers to the collapsed y-coordinate of the data part of the dst tensor
    Window win;

    if (!is_scalar_block)
    {
        win = calculate_max_window(dst_shape, Steps(n0));

        // Collapse the dimensions corresponding to indices in the execution window
        for (int i = 0; i < index_len; ++i)
        {
            win.set(dst->num_dimensions() - (i + 1), Window::Dimension(0, 1, 1));
        }

        win = win.collapse(win, 1);
    }

    // Set build options
    CLBuildOptions build_opts;
    build_opts.add_option("-DDATA_TYPE=" + get_cl_type_from_data_type(dst->data_type()));
    build_opts.add_option_if(is_data_type_float(dst->data_type()), "-DIS_FLOAT");

    const int   num_dims      = dst->num_dimensions();
    TensorShape ind_collapsed = indices->tensor_shape().collapsed_from(1);
    build_opts.add_option("-DNUM_INDICES=" + support::cpp11::to_string(ind_collapsed[1]));
    build_opts.add_option("-DINDEX_LENGTH=" + support::cpp11::to_string(index_len));

    // We provide 5 variables to use in a constant array
    for (int i = 1; i <= max_index_length; i++)
    {
        build_opts.add_option("-DOUT_SHAPE_N_MINUS_" + support::cpp11::to_string(i) + "=" +
                              support::cpp11::to_string(dst_shape[std::max(num_dims - i, 0)]));
    }

    build_opts.add_option("-DN0=" + support::cpp11::to_string(n0));
    build_opts.add_option("-DPARTIAL_N0=" + support::cpp11::to_string(partial_n0));

    switch (info.func)
    {