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
cec3e696
Unverified
Commit
cec3e696
authored
Sep 08, 2023
by
Bartlomiej Wroblewski
Committed by
GitHub
Sep 08, 2023
Browse files
Merge branch 'develop' into bwroblew/dpp8_small_cases
parents
9e41b563
8f84a012
Changes
21
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
19 additions
and
0 deletions
+19
-0
profiler/src/profile_gemm_bilinear.cpp
profiler/src/profile_gemm_bilinear.cpp
+19
-0
No files found.
profiler/src/profile_gemm_bilinear.cpp
View file @
cec3e696
...
...
@@ -71,6 +71,9 @@ int profile_gemm_bilinear(int argc, char* argv[])
using
F16
=
ck
::
half_t
;
using
F32
=
float
;
using
I8
=
std
::
int8_t
;
using
I32
=
std
::
int32_t
;
using
Row
=
ck
::
tensor_layout
::
gemm
::
RowMajor
;
using
Col
=
ck
::
tensor_layout
::
gemm
::
ColumnMajor
;
...
...
@@ -141,6 +144,22 @@ int profile_gemm_bilinear(int argc, char* argv[])
{
return
profile
(
F16
{},
F16
{},
F32
{},
F16
{},
F16
{},
Col
{},
Col
{},
Row
{},
Row
{});
}
else
if
(
data_type
==
MatrixDataType
::
INT8_INT8_INT8_INT8
&&
layout
==
MatrixLayout
::
MK_KN_MN_MN
)
{
return
profile
(
I8
{},
I8
{},
I32
{},
I8
{},
I8
{},
Row
{},
Row
{},
Row
{},
Row
{});
}
else
if
(
data_type
==
MatrixDataType
::
INT8_INT8_INT8_INT8
&&
layout
==
MatrixLayout
::
MK_NK_MN_MN
)
{
return
profile
(
I8
{},
I8
{},
I32
{},
I8
{},
I8
{},
Row
{},
Col
{},
Row
{},
Row
{});
}
else
if
(
data_type
==
MatrixDataType
::
INT8_INT8_INT8_INT8
&&
layout
==
MatrixLayout
::
KM_KN_MN_MN
)
{
return
profile
(
I8
{},
I8
{},
I32
{},
I8
{},
I8
{},
Col
{},
Row
{},
Row
{},
Row
{});
}
else
if
(
data_type
==
MatrixDataType
::
INT8_INT8_INT8_INT8
&&
layout
==
MatrixLayout
::
KM_NK_MN_MN
)
{
return
profile
(
I8
{},
I8
{},
I32
{},
I8
{},
I8
{},
Col
{},
Col
{},
Row
{},
Row
{});
}
else
{
std
::
cout
<<
"this data_type & layout is not implemented"
<<
std
::
endl
;
...
...
Prev
1
2
Next
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