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