blas.h

Go to the documentation of this file.

/********************************************************
 * Copyright (c) 2014, ArrayFire
 * All rights reserved.
 *
 * This file is distributed under 3-clause BSD license.
 * The complete license agreement can be obtained at:
 * http://arrayfire.com/licenses/BSD-3-Clause
 ********************************************************/

#pragma once
#include <af/defines.h>

#ifdef __cplusplus
namespace af
{
    class array;
    AFAPI array matmul(const array &lhs, const array &rhs,
                       const matProp optLhs = AF_MAT_NONE,
                       const matProp optRhs = AF_MAT_NONE);

    AFAPI array matmulNT(const array &lhs, const array &rhs);

    AFAPI array matmulTN(const array &lhs, const array &rhs);

    AFAPI array matmulTT(const array &lhs, const array &rhs);

    AFAPI array matmul(const array &a, const array &b, const array &c);


    AFAPI array matmul(const array &a, const array &b, const array &c, const array &d);

#if AF_API_VERSION >= 35

    template <typename T>
    T dot(const array &lhs, const array &rhs,
          const matProp optLhs = AF_MAT_NONE,
          const matProp optRhs = AF_MAT_NONE);
#endif

    AFAPI array dot(const array &lhs, const array &rhs,
                    const matProp optLhs = AF_MAT_NONE,
                    const matProp optRhs = AF_MAT_NONE);

    AFAPI array transpose(const array &in, const bool conjugate = false);

    AFAPI void transposeInPlace(array &in, const bool conjugate = false);
}
#endif

#ifdef __cplusplus
extern "C" {
#endif

#if AF_API_VERSION >= 37

    AFAPI af_err af_gemm(af_array *C, const af_mat_prop opA, const af_mat_prop opB,
                         const void *alpha, const af_array A, const af_array B,
                         const void *beta);
#endif

    AFAPI af_err af_matmul( af_array *out ,
                            const af_array lhs, const af_array rhs,
                            const af_mat_prop optLhs, const af_mat_prop optRhs);

    AFAPI af_err af_dot(af_array *out,
                        const af_array lhs, const af_array rhs,
                        const af_mat_prop optLhs, const af_mat_prop optRhs);

#if AF_API_VERSION >= 35

    AFAPI af_err af_dot_all(double *real, double *imag,
                            const af_array lhs, const af_array rhs,
                            const af_mat_prop optLhs, const af_mat_prop optRhs);
#endif

    AFAPI af_err af_transpose(af_array *out, af_array in, const bool conjugate);

    AFAPI af_err af_transpose_inplace(af_array in, const bool conjugate);


#ifdef __cplusplus
}
#endif