Class AParam

src/backend/oneapi/Param.hpp:49–90

template<typename T, sycl::access_mode AM>
struct AParam {
    sycl::accessor<T, 1, AM, sycl::target::device,
                   sycl::access::placeholder::true_t>
        data;
    af::dim4 dims;
    af::dim4 strides;
    dim_t offset;
    AParam& operator=(const AParam& other) = default;
    AParam(const AParam& other)            = default;
    AParam(AParam&& other)                 = default;

    dim_t* dims_ptr() { return dims.get(); }
    dim_t* strides_ptr() { return strides.get(); }

    // AF_DEPRECATED("Use Array<T>")
    AParam() : data(), dims{0, 0, 0, 0}, strides{0, 0, 0, 0}, offset(0) {}

    AParam(sycl::buffer<T, 1>& data_, const dim_t dims_[4],
           const dim_t strides_[4], dim_t offset_)
        : data(data_), dims(4, dims_), strides(4, strides_), offset(offset_) {}
    // AF_DEPRECATED("Use Array<T>")
    AParam(sycl::handler& h, sycl::buffer<T, 1>& data_, const dim_t dims_[4],
           const dim_t strides_[4], dim_t offset_)
        : data(data_), dims(4, dims_), strides(4, strides_), offset(offset_) {
        require(h);
    }

    template<sycl::access::mode MODE>
    sycl::accessor<data_t<T>, 1, MODE> get_accessor(sycl::handler& h) const {
        return *data;
    }

    void require(sycl::handler& h) const { h.require(data); }

    operator KParam() const {
        return KParam{{dims[0], dims[1], dims[2], dims[3]},
                      {strides[0], strides[1], strides[2], strides[3]},
                      offset};
    }

    ~AParam() = default;
};

// AF_DEPRECATED("Use Array<T>")
template<typename T>