Function af_write_array

src/api/c/array.cpp:286–363

}

af_err af_write_array(af_array arr, const void *data, const size_t bytes,
                      af_source src) {
    if (bytes == 0) { return AF_SUCCESS; }
    try {
        af_dtype type = getInfo(arr).getType();
        ARG_ASSERT(1, (data != nullptr));
        ARG_ASSERT(3, (src == afHost || src == afDevice));
        // FIXME ArrayInfo class no bytes method, hence commented
        // DIM_ASSERT(2, bytes <= getInfo(arr).bytes());

        switch (type) {
            case f32:
                write_array(arr, static_cast<const float *>(data), bytes, src);
                break;
            case c32:
                write_array(arr, static_cast<const cfloat *>(data), bytes, src);
                break;
            case f64:
                write_array(arr, static_cast<const double *>(data), bytes, src);
                break;
            case c64:
                write_array(arr, static_cast<const cdouble *>(data), bytes,
                            src);
                break;
            case b8:
                write_array(arr, static_cast<const char *>(data), bytes, src);
                break;
            case s32:
                write_array(arr, static_cast<const int *>(data), bytes, src);
                break;
            case u32:
                write_array(arr, static_cast<const uint *>(data), bytes, src);
                break;
            case s8:
                write_array(arr, static_cast<const schar *>(data), bytes, src);
                break;
            case u8:
                write_array(arr, static_cast<const uchar *>(data), bytes, src);
                break;
            case s64:
                write_array(arr, static_cast<const intl *>(data), bytes, src);
                break;
            case u64:
                write_array(arr, static_cast<const uintl *>(data), bytes, src);
                break;
            case s16:
                write_array(arr, static_cast<const short *>(data), bytes, src);
                break;
            case u16:
                write_array(arr, static_cast<const ushort *>(data), bytes, src);
                break;
            case f16:
                write_array(arr, static_cast<const half *>(data), bytes, src);
                break;
            default: TYPE_ERROR(4, type);
        }
    }
    CATCHALL