Function padDataHelper

src/backend/oneapi/kernel/fftconvolve_pad.hpp:88–119

template<typename convT, typename T>
void padDataHelper(Param<convT> packed, Param<T> sig, Param<T> filter,
                   const int rank, AF_BATCH_KIND kind) {
    Param<T> sig_tmp, filter_tmp;
    calcParamSizes(sig_tmp, filter_tmp, packed, sig, filter, rank, kind);

    int filter_packed_elem =
        filter_tmp.info.strides[3] * filter_tmp.info.dims[3];

    int blocks = divup(filter_packed_elem, THREADS);

    // Locate features kernel sizes
    auto local  = sycl::range(THREADS);
    auto global = sycl::range(blocks * THREADS);

    // Treat complex output as an array of scalars
    using convScalarT      = typename convT::value_type;
    auto packed_num_elem   = (*packed.data).get_range().size();
    auto filter_tmp_buffer = (*packed.data)
                                 .template reinterpret<convScalarT>(
                                     sycl::range<1>{packed_num_elem * 2});

    getQueue().submit([&](auto &h) {
        read_accessor<T> d_filter = {*filter.data, h, sycl::read_only};
        write_accessor<convScalarT> d_filter_tmp = {filter_tmp_buffer, h};
        h.parallel_for(
            sycl::nd_range{global, local},
            fftconvolve_padCreateKernel<T, convScalarT>(
                d_filter_tmp, filter_tmp.info, d_filter, filter.info));
    });

    ONEAPI_DEBUG_FINISH(getQueue());
}
}  // namespace kernel
}  // namespace oneapi
}  // namespace arrayfire