fix cuda error

This commit is contained in:
zhang hui 2026-01-15 16:27:19 +08:00
parent a31225cc03
commit 8c875b23cb
1 changed files with 1 additions and 1 deletions

View File

@ -809,7 +809,7 @@ static void mul_mat_f_switch_cols_per_block(
}
#define DECL_MMF_CASE_HELPER(T, ncols_dst) \
template void mul_mat_f_cuda<T, ncols_dst>( \
template void mul_mat_f_cuda<T, MMF_ROWS_PER_BLOCK, ncols_dst>( \
const T * x, const float * y, const int32_t * ids, float * dst, \
const int64_t ncols_x, const int64_t nrows_x, int64_t ncols_dst_total, const int64_t stride_row, const int64_t stride_col_y, const int64_t stride_col_dst, \
const int64_t stride_col_id, const int64_t stride_row_id, \