Skip to content

Commit a6668dd

Browse files
committed
Stop use gemm_mt globally
1 parent c4fc6aa commit a6668dd

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

source/module_hsolver/kernels/math_kernel_op.cpp

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -275,10 +275,16 @@ struct gemm_op<T, base_device::DEVICE_CPU>
275275
const int& ldb,
276276
const T* beta,
277277
T* c,
278-
const int& ldc)
278+
const int& ldc,
279+
bool use_dsp)
279280
{
280281
#ifdef __DSP
281-
BlasConnector::gemm(transb, transa, n, m, k, *alpha, b, ldb, a, lda, *beta, c, ldc, base_device::AbacusDevice_t::DspDevice);
282+
if (use_dsp){
283+
BlasConnector::gemm(transb, transa, n, m, k, *alpha, b, ldb, a, lda, *beta, c, ldc, base_device::AbacusDevice_t::DspDevice);
284+
}
285+
else{
286+
BlasConnector::gemm(transb, transa, n, m, k, *alpha, b, ldb, a, lda, *beta, c, ldc);
287+
}
282288
#else
283289
BlasConnector::gemm(transb, transa, n, m, k, *alpha, b, ldb, a, lda, *beta, c, ldc);
284290
#endif

source/module_hsolver/kernels/math_kernel_op.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ template <typename T, typename Device> struct gemm_op {
261261
void operator()(const Device *d, const char &transa, const char &transb,
262262
const int &m, const int &n, const int &k, const T *alpha,
263263
const T *a, const int &lda, const T *b, const int &ldb,
264-
const T *beta, T *c, const int &ldc);
264+
const T *beta, T *c, const int &ldc, bool usd_dsp = false);
265265
};
266266

267267
template <typename T, typename Device> struct matrixTranspose_op {

0 commit comments

Comments
 (0)