Function af_svd_inplace

src/api/c/svd.cpp:100–127

}

af_err af_svd_inplace(af_array *u, af_array *s, af_array *vt, af_array in) {
    try {
        const ArrayInfo &info = getInfo(in);
        dim4 dims             = info.dims();

        ARG_ASSERT(3, (dims.ndims() >= 0 && dims.ndims() <= 2));
        af_dtype type = info.getType();

        if (dims.ndims() == 0) {
            AF_CHECK(af_create_handle(u, 0, nullptr, type));
            AF_CHECK(af_create_handle(s, 0, nullptr, type));
            AF_CHECK(af_create_handle(vt, 0, nullptr, type));
            return AF_SUCCESS;
        }

        DIM_ASSERT(3, dims[0] >= dims[1]);

        switch (type) {
            case f64: svdInPlace<double>(s, u, vt, in); break;
            case f32: svdInPlace<float>(s, u, vt, in); break;
            case c64: svdInPlace<cdouble>(s, u, vt, in); break;
            case c32: svdInPlace<cfloat>(s, u, vt, in); break;
            default: TYPE_ERROR(1, type);
        }
    }
    CATCHALL;
    return AF_SUCCESS;
}