Function af_cast

src/api/c/cast.cpp:71–97

}

af_err af_cast(af_array* out, const af_array in, const af_dtype type) {
    try {
        const ArrayInfo& info = getInfo(in, false);

        af_dtype inType = info.getType();
        if ((inType == c32 || inType == c64) &&
            (type == f32 || type == f64 || type == f16)) {
            AF_ERROR(
                "Casting is not allowed from complex (c32/c64) to real "
                "(f16/f32/f64) types.\n"
                "Use abs, real, imag etc to convert complex to floating type.",
                AF_ERR_TYPE);
        }

        dim4 idims = info.dims();
        if (idims.elements() == 0) {
            return af_create_handle(out, 0, nullptr, type);
        }

        af_array res = cast(in, type);

        std::swap(*out, res);
    }
    CATCHALL;

    return AF_SUCCESS;
}