Skip to content

Commit 66590a0

Browse files
committed
Fix typo in blas_impl.h
1 parent ad2e420 commit 66590a0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/fluid/operators/math/blas_impl.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,9 +172,9 @@ void Blas<platform::CPUDeviceContext>::BatchedGEMM(
172172
c_array.data(), &ldc, 1 /* group_count */, &batchCount);
173173
#else
174174
for (int k = 0; k < batchCount; ++k) {
175-
const float *Ak = &A[k * strideA];
176-
const float *Bk = &B[k * strideB];
177-
float *Ck = &C[k * M * N];
175+
auto *Ak = &A[k * strideA];
176+
auto *Bk = &B[k * strideB];
177+
auto *Ck = &C[k * M * N];
178178
this->template GEMM<T>(transA, transB, M, N, K, alpha, Ak, Bk, beta, Ck);
179179
}
180180
#endif

0 commit comments

Comments
 (0)