Function lu_inplace

src/backend/opencl/lu.cpp:61–79

template<typename T>
Array<int> lu_inplace(Array<T> &in, const bool convert_pivot) {
    if (OpenCLCPUOffload()) { return cpu::lu_inplace(in, convert_pivot); }

    dim4 iDims = in.dims();
    int M      = iDims[0];
    int N      = iDims[1];
    int MN     = std::min(M, N);
    std::vector<int> ipiv(MN);

    cl::Buffer *in_buf = in.get();
    int info           = 0;
    magma_getrf_gpu<T>(M, N, (*in_buf)(), in.getOffset(), in.strides()[1],
                       &ipiv[0], getQueue()(), &info);

    if (!convert_pivot) { return createHostDataArray<int>(dim4(MN), &ipiv[0]); }

    Array<int> pivot = convertPivot(&ipiv[0], MN, M);
    return pivot;
}

bool isLAPACKAvailable() { return true; }