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
gaoqiong
composable_kernel
Commits
04ba72c7
Commit
04ba72c7
authored
Jun 01, 2022
by
Jing Zhang
Browse files
formatting
parent
5a3c2297
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
4 deletions
+7
-4
library/include/ck/library/reference_tensor_operation/cpu/reference_gemm_transpose.hpp
...ference_tensor_operation/cpu/reference_gemm_transpose.hpp
+7
-4
No files found.
library/include/ck/library/reference_tensor_operation/cpu/reference_gemm_transpose.hpp
View file @
04ba72c7
...
...
@@ -53,8 +53,8 @@ struct ReferenceGemmTranspose : public device::BaseOperator
auto
f_mk_kn_m0m1n0n1
=
[
&
](
auto
m0
,
auto
m1
,
auto
n0
,
auto
n1
)
{
const
int
K
=
arg
.
a_m_k_
.
mDesc
.
GetLengths
()[
1
];
const
int
m
=
m0
*
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
1
]
+
m1
;
const
int
n
=
n0
*
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
3
]
+
n1
;
const
int
m
=
m0
*
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
1
]
+
m1
;
const
int
n
=
n0
*
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
3
]
+
n1
;
float
v_acc
=
0
;
...
...
@@ -76,8 +76,11 @@ struct ReferenceGemmTranspose : public device::BaseOperator
arg
.
c_m0_m1_n0_n1_
(
m0
,
m1
,
n0
,
n1
)
=
v_c
;
};
make_ParallelTensorFunctor
(
f_mk_kn_m0m1n0n1
,
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
0
],
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
1
],
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
2
],
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
3
])(
make_ParallelTensorFunctor
(
f_mk_kn_m0m1n0n1
,
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
0
],
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
1
],
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
2
],
arg
.
c_m0_m1_n0_n1_
.
mDesc
.
GetLengths
()[
3
])(
std
::
thread
::
hardware_concurrency
());
return
0
;
...
...
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