diff --git a/paddle/fluid/operators/math/blas_impl.h b/paddle/fluid/operators/math/blas_impl.h index 577cbe3beb806ffcb2f1a7d7a469402be9b69224..14b3624b420cb883b36268c0a5a9e8692dbb5b43 100644 --- a/paddle/fluid/operators/math/blas_impl.h +++ b/paddle/fluid/operators/math/blas_impl.h @@ -172,9 +172,9 @@ void Blas::BatchedGEMM( c_array.data(), &ldc, 1 /* group_count */, &batchCount); #else for (int k = 0; k < batchCount; ++k) { - const float *Ak = &A[k * strideA]; - const float *Bk = &B[k * strideB]; - float *Ck = &C[k * M * N]; + auto *Ak = &A[k * strideA]; + auto *Bk = &B[k * strideB]; + auto *Ck = &C[k * M * N]; this->template GEMM(transA, transB, M, N, K, alpha, Ak, Bk, beta, Ck); } #endif