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
54b47924
Commit
54b47924
authored
Apr 07, 2025
by
xgqdut2016
Browse files
issue/130: use ptrdiff_t
parent
4f1244be
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
src/infiniop/ops/gemm/cpu/gemm_cpu.cc
src/infiniop/ops/gemm/cpu/gemm_cpu.cc
+3
-3
No files found.
src/infiniop/ops/gemm/cpu/gemm_cpu.cc
View file @
54b47924
...
...
@@ -43,9 +43,9 @@ void calculate(
std
::
swap
(
a
,
b
);
}
#pragma omp parallel for collapse(3)
for
(
size
_t
i
=
0
;
i
<
info
.
batch
;
++
i
)
{
for
(
size
_t
m_
=
0
;
m_
<
info
.
m
;
++
m_
)
{
for
(
size
_t
n_
=
0
;
n_
<
info
.
n
;
++
n_
)
{
for
(
ptrdiff
_t
i
=
0
;
i
<
ptrdiff_t
(
info
.
batch
)
;
++
i
)
{
for
(
ptrdiff
_t
m_
=
0
;
m_
<
ptrdiff_t
(
info
.
m
)
;
++
m_
)
{
for
(
ptrdiff
_t
n_
=
0
;
n_
<
ptrdiff_t
(
info
.
n
)
;
++
n_
)
{
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
(
size_t
k_
=
0
;
k_
<
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