Function getOffset

src/backend/oneapi/kernel/select.hpp:31–38

constexpr int REPEAT = 64;

int getOffset(const dim_t *dims, const dim_t *strides, const dim_t *refdims,
              int ids[4]) {
    int off = 0;
    off += ids[3] * (dims[3] == refdims[3]) * strides[3];
    off += ids[2] * (dims[2] == refdims[2]) * strides[2];
    off += ids[1] * (dims[1] == refdims[1]) * strides[1];
    return off;
}

template<typename T>
class selectKernelCreateKernel {