Commit 8e2192f4 authored by xgqdut2016's avatar xgqdut2016
Browse files

issue/130: modified for

parent bc981f39
...@@ -42,15 +42,15 @@ void calculate( ...@@ -42,15 +42,15 @@ void calculate(
if (info.is_transed) { if (info.is_transed) {
std::swap(a, b); std::swap(a, b);
} }
const size_t m_n = info.m * info.n;
const size_t n = info.n;
#pragma omp parallel for #pragma omp parallel for
for (ptrdiff_t index = 0; index < ptrdiff_t(info.batch * info.m * info.n); ++index) { for (ptrdiff_t index = 0; index < ptrdiff_t(info.batch * info.m * info.n); ++index) {
size_t i, m_, n_; size_t ind = index;
i = index / m_n; size_t n_ = ind % info.n;
size_t rem = index - i * m_n; // 替代 `%` 用减法 ind /= info.n;
m_ = rem / n; size_t m_ = ind % info.m;
n_ = rem - m_ * n; // 替代 `%` 用减法 ind /= info.m;
size_t i = ind;
auto c_ = reinterpret_cast<Tdata *>(c) + i * info.c_matrix.stride + m_ * info.c_matrix.row_stride + n_ * info.c_matrix.col_stride; auto c_ = reinterpret_cast<Tdata *>(c) + i * info.c_matrix.stride + m_ * info.c_matrix.row_stride + n_ * info.c_matrix.col_stride;
float sum = 0; float sum = 0;
for (int k_ = 0; k_ < static_cast<int>(info.k); ++k_) { for (int k_ = 0; k_ < static_cast<int>(info.k); ++k_) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment