Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
8e2192f4
Commit
8e2192f4
authored
Apr 11, 2025
by
xgqdut2016
Browse files
issue/130: modified for
parent
bc981f39
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
7 deletions
+7
-7
src/infiniop/ops/gemm/cpu/gemm_cpu.cc
src/infiniop/ops/gemm/cpu/gemm_cpu.cc
+7
-7
No files found.
src/infiniop/ops/gemm/cpu/gemm_cpu.cc
View file @
8e2192f4
...
...
@@ -42,15 +42,15 @@ void calculate(
if
(
info
.
is_transed
)
{
std
::
swap
(
a
,
b
);
}
const
size_t
m_n
=
info
.
m
*
info
.
n
;
const
size_t
n
=
info
.
n
;
#pragma omp parallel for
for
(
ptrdiff_t
index
=
0
;
index
<
ptrdiff_t
(
info
.
batch
*
info
.
m
*
info
.
n
);
++
index
)
{
size_t
i
,
m_
,
n_
;
i
=
index
/
m_n
;
size_t
rem
=
index
-
i
*
m_n
;
// 替代 `%` 用减法
m_
=
rem
/
n
;
n_
=
rem
-
m_
*
n
;
// 替代 `%` 用减法
size_t
ind
=
index
;
size_t
n_
=
ind
%
info
.
n
;
ind
/=
info
.
n
;
size_t
m_
=
ind
%
info
.
m
;
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
;
float
sum
=
0
;
for
(
int
k_
=
0
;
k_
<
static_cast
<
int
>
(
info
.
k
);
++
k_
)
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment