Function af_example_function

src/api/c/exampleFunction.cpp:52–102

}

af_err af_example_function(af_array* out, const af_array a,
                           const af_someenum_t param) {
    try {
        af_array output = 0;
        const ArrayInfo& info =
            getInfo(a);  // ArrayInfo is the base class which
                         // each backend specific Array inherits
                         // This class stores the basic array meta-data
                         // such as type of data, dimensions,
                         // offsets and strides. This class is declared
                         // in src/backend/common/ArrayInfo.hpp
        af::dim4 dims = info.dims();

        ARG_ASSERT(2, (dims.ndims() >= 0 && dims.ndims() <= 3));
        // defined in err_common.hpp
        // there are other useful Macros
        // for different purposes, feel free
        // to look at the header

        af_dtype type = info.getType();

        switch (type) {  // Based on the data type, call backend specific
                         // implementation
            case f64: output = example<double>(a, a, param); break;
            case f32: output = example<float>(a, a, param); break;
            case s32: output = example<int>(a, a, param); break;
            case u32: output = example<uint>(a, a, param); break;
            case s8: output = example<schar>(a, a, param); break;
            case u8: output = example<uchar>(a, a, param); break;
            case b8: output = example<char>(a, a, param); break;
            case c32: output = example<cfloat>(a, a, param); break;
            case c64: output = example<cdouble>(a, a, param); break;
            default:
                TYPE_ERROR(1,
                           type);  // Another helpful macro from err_common.hpp
                                   // that helps throw type based error messages
        }

        std::swap(*out, output);  // if the function has returned successfully,
                                  // swap the temporary 'output' variable with
                                  // '*out'
    }
    CATCHALL;  // All throws/exceptions from any internal
               // implementations are caught by this CATCHALL
               // macro and handled appropriately.

    return AF_SUCCESS;  // In case of successfull completion, return AF_SUCCESS
                        // There are set of error codes defined in defines.h
                        // which you are used by CATCHALL to return approriate
                        // code
}