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
fd956f89
Commit
fd956f89
authored
May 04, 2023
by
Adam Osewski
Browse files
Log more information and initialize with constant value.
parent
406f71a6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
22 additions
and
1 deletion
+22
-1
test/grouped_gemm/test_grouped_gemm_util.hpp
test/grouped_gemm/test_grouped_gemm_util.hpp
+22
-1
No files found.
test/grouped_gemm/test_grouped_gemm_util.hpp
View file @
fd956f89
...
@@ -17,6 +17,18 @@
...
@@ -17,6 +17,18 @@
namespace
ck
{
namespace
ck
{
namespace
test
{
namespace
test
{
template
<
typename
Range
>
std
::
string
serialize_range
(
const
Range
&
range
)
{
std
::
stringstream
ss
;
for
(
auto
&
r
:
range
)
{
ss
<<
r
<<
", "
;
}
std
::
string
str
=
ss
.
str
();
return
std
::
string
(
str
.
begin
(),
str
.
end
()
-
2
);
}
template
<
typename
Tuple
>
template
<
typename
Tuple
>
class
TestGroupedGemm
:
public
testing
::
TestWithParam
<
int
>
class
TestGroupedGemm
:
public
testing
::
TestWithParam
<
int
>
{
{
...
@@ -30,7 +42,7 @@ class TestGroupedGemm : public testing::TestWithParam<int>
...
@@ -30,7 +42,7 @@ class TestGroupedGemm : public testing::TestWithParam<int>
public:
public:
bool
verify_
=
true
;
bool
verify_
=
true
;
int
init_method_
=
2
;
// decimal value initialization
int
init_method_
=
0
;
// decimal value initialization
bool
log_
=
false
;
bool
log_
=
false
;
bool
bench_
=
false
;
// measure kernel performance
bool
bench_
=
false
;
// measure kernel performance
...
@@ -44,6 +56,15 @@ class TestGroupedGemm : public testing::TestWithParam<int>
...
@@ -44,6 +56,15 @@ class TestGroupedGemm : public testing::TestWithParam<int>
const
std
::
vector
<
int
>&
StrideCs
,
const
std
::
vector
<
int
>&
StrideCs
,
int
kbatch
=
1
)
int
kbatch
=
1
)
{
{
std
::
cout
<<
"Ms: ["
<<
serialize_range
(
Ms
)
<<
"] "
<<
"Ns: ["
<<
serialize_range
(
Ns
)
<<
"] "
<<
"Ks: ["
<<
serialize_range
(
Ks
)
<<
"] "
<<
"StrideAs: ["
<<
serialize_range
(
StrideAs
)
<<
"] "
<<
"StrideBs: ["
<<
serialize_range
(
StrideBs
)
<<
"] "
<<
"StrideCs: ["
<<
serialize_range
(
StrideCs
)
<<
"] "
<<
"kbatch: "
<<
kbatch
<<
std
::
endl
;
bool
pass
=
ck
::
profiler
::
profile_grouped_gemm_impl
<
ADataType
,
bool
pass
=
ck
::
profiler
::
profile_grouped_gemm_impl
<
ADataType
,
BDataType
,
BDataType
,
EDataType
,
EDataType
,
...
...
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