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
2ead0404
Commit
2ead0404
authored
Oct 19, 2022
by
Anthony Chang
Browse files
report failure when result not validated
parent
f91153c3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
5 deletions
+7
-5
test/gemm/gemm_standalone_xdl_fp16.cpp
test/gemm/gemm_standalone_xdl_fp16.cpp
+7
-5
No files found.
test/gemm/gemm_standalone_xdl_fp16.cpp
View file @
2ead0404
...
@@ -117,6 +117,7 @@ int main(int argc, char* argv[])
...
@@ -117,6 +117,7 @@ int main(int argc, char* argv[])
return
0
;
return
0
;
}
}
bool
pass
=
true
;
for
(
auto
&
p
:
problems
)
for
(
auto
&
p
:
problems
)
{
{
GemmParams
&
problem_size
=
std
::
get
<
0
>
(
p
);
GemmParams
&
problem_size
=
std
::
get
<
0
>
(
p
);
...
@@ -132,28 +133,29 @@ int main(int argc, char* argv[])
...
@@ -132,28 +133,29 @@ int main(int argc, char* argv[])
if
(
!
layout_config
.
ARowMajor
&&
!
layout_config
.
BRowMajor
)
if
(
!
layout_config
.
ARowMajor
&&
!
layout_config
.
BRowMajor
)
{
{
auto
op_ptr
=
dynamic_cast
<
DeviceGemmNN
*>
(
ops
[
0
].
get
());
auto
op_ptr
=
dynamic_cast
<
DeviceGemmNN
*>
(
ops
[
0
].
get
());
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
pass
&=
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
}
}
else
if
(
!
layout_config
.
ARowMajor
&&
layout_config
.
BRowMajor
)
else
if
(
!
layout_config
.
ARowMajor
&&
layout_config
.
BRowMajor
)
{
{
auto
op_ptr
=
dynamic_cast
<
DeviceGemmNT
*>
(
ops
[
0
].
get
());
auto
op_ptr
=
dynamic_cast
<
DeviceGemmNT
*>
(
ops
[
0
].
get
());
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
pass
&=
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
}
}
else
if
(
layout_config
.
ARowMajor
&&
!
layout_config
.
BRowMajor
)
else
if
(
layout_config
.
ARowMajor
&&
!
layout_config
.
BRowMajor
)
{
{
auto
op_ptr
=
dynamic_cast
<
DeviceGemmTN
*>
(
ops
[
0
].
get
());
auto
op_ptr
=
dynamic_cast
<
DeviceGemmTN
*>
(
ops
[
0
].
get
());
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
pass
&=
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
}
}
else
if
(
layout_config
.
ARowMajor
&&
layout_config
.
BRowMajor
)
else
if
(
layout_config
.
ARowMajor
&&
layout_config
.
BRowMajor
)
{
{
auto
op_ptr
=
dynamic_cast
<
DeviceGemmTT
*>
(
ops
[
0
].
get
());
auto
op_ptr
=
dynamic_cast
<
DeviceGemmTT
*>
(
ops
[
0
].
get
());
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
pass
&=
ck
::
gemm_util
::
TestGemm
<
AccDataType
>
{}(
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
op_ptr
,
problem_size
,
do_verification
,
time_kernel
);
}
}
}
}
return
0
;
std
::
cout
<<
(
pass
?
"ALL TESTS PASSED"
:
"SOME TESTS FAILED"
)
<<
std
::
endl
;
return
pass
?
0
:
1
;
}
}
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