Skip to content
16 changes: 8 additions & 8 deletions paddle/phi/kernels/funcs/blas/blas.h
Original file line number Diff line number Diff line change
Expand Up @@ -256,14 +256,14 @@ class Blas {
void VPOW(int n, const T* x, T alpha, T* y) const;

template <typename T>
void GEMV(bool trans_a,
int M,
int N,
T alpha,
const T* A,
const T* B,
T beta,
T* C) const;
TEST_API void GEMV(bool trans_a,
int M,
int N,
T alpha,
const T* A,
const T* B,
T beta,
T* C) const;

template <typename T>
T DOT(int n, const T* x, const T* y) const;
Expand Down
16 changes: 8 additions & 8 deletions paddle/phi/kernels/funcs/blas/blas_impl.h
Original file line number Diff line number Diff line change
Expand Up @@ -1338,14 +1338,14 @@ T Blas<phi::CPUContext>::ASUM(int n, T *x, int inc) const {

template <>
template <typename T>
void Blas<phi::CPUContext>::GEMV(bool trans_a,
int M,
int N,
T alpha,
const T *A,
const T *B,
T beta,
T *C) const {
TEST_API void Blas<phi::CPUContext>::GEMV(bool trans_a,
int M,
int N,
T alpha,
const T *A,
const T *B,
T beta,
T *C) const {
CBLAS_TRANSPOSE transA = !trans_a ? CblasNoTrans : CblasTrans;
CBlas<T>::GEMV(CblasRowMajor, transA, M, N, alpha, A, N, B, 1, beta, C, 1);
}
Expand Down
5 changes: 1 addition & 4 deletions test/cpp/phi/kernels/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,4 @@
cc_test(
test_math_function
SRCS test_math_function.cc
DEPS phi common)
paddle_test(test_math_function SRCS test_math_function.cc DEPS common)
if(WITH_GPU)
nv_test(
test_math_function_gpu
Expand Down