Method device

src/backend/opencl/api.cpp:16–25

namespace af {
template<>
AFAPI cl_mem *array::device() const {
    auto *mem_ptr = new cl_mem;
    void *dptr    = nullptr;
    af_err err    = af_get_device_ptr(&dptr, get());
    memcpy(mem_ptr, &dptr, sizeof(void *));
    if (err != AF_SUCCESS) {
        throw af::exception("Failed to get cl_mem from array object");
    }
    return mem_ptr;
}
}  // namespace af