Mercurial > hg > CbC > CbC_llvm
diff mlir/test/mlir-cpu-runner/include/cblas.h @ 150:1d019706d866
LLVM10
author | anatofuz |
---|---|
date | Thu, 13 Feb 2020 15:10:13 +0900 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mlir/test/mlir-cpu-runner/include/cblas.h Thu Feb 13 15:10:13 2020 +0900 @@ -0,0 +1,49 @@ +//===- cblas.h - Simple Blas subset ---------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +#ifndef MLIR_CPU_RUNNER_CBLAS_H_ +#define MLIR_CPU_RUNNER_CBLAS_H_ + +#include "mlir_runner_utils.h" + +#ifdef _WIN32 +#ifndef MLIR_CBLAS_EXPORT +#ifdef cblas_EXPORTS +/* We are building this library */ +#define MLIR_CBLAS_EXPORT __declspec(dllexport) +#else +/* We are using this library */ +#define MLIR_CBLAS_EXPORT __declspec(dllimport) +#endif // cblas_EXPORTS +#endif // MLIR_CBLAS_EXPORT +#else +#define MLIR_CBLAS_EXPORT +#endif // _WIN32 + +/// This reproduces a minimal subset of cblas to allow integration testing +/// without explicitly requiring a dependence on an external library. +/// Without loss of generality, various cblas implementations may be swapped in +/// by including the proper headers and linking with the proper library. +enum CBLAS_ORDER { CblasRowMajor = 101, CblasColMajor = 102 }; +enum CBLAS_TRANSPOSE { + CblasNoTrans = 111, + CblasTrans = 112, + CblasConjTrans = 113 +}; + +extern "C" MLIR_CBLAS_EXPORT float cblas_sdot(const int N, const float *X, + const int incX, const float *Y, + const int incY); + +extern "C" MLIR_CBLAS_EXPORT void +cblas_sgemm(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, + const enum CBLAS_TRANSPOSE TransB, const int M, const int N, + const int K, const float alpha, const float *A, const int lda, + const float *B, const int ldb, const float beta, float *C, + const int ldc); + +#endif // MLIR_CPU_RUNNER_CBLAS_H_