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_ROCM
Commits
f67cfe03
Commit
f67cfe03
authored
Feb 06, 2025
by
Andriy Roshchenko
Browse files
Update the example to result in a non-zero value
parent
2d4fb7d5
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
1 deletion
+7
-1
test/mx_mfma_op/mx_mfma_op.hpp
test/mx_mfma_op/mx_mfma_op.hpp
+7
-1
No files found.
test/mx_mfma_op/mx_mfma_op.hpp
View file @
f67cfe03
...
@@ -777,6 +777,12 @@ matmul(const AType* a, const ScaleType* xa, const BType* b, const ScaleType* xb,
...
@@ -777,6 +777,12 @@ matmul(const AType* a, const ScaleType* xa, const BType* b, const ScaleType* xb,
mfma_type_selector
<
AFragT
,
BFragT
,
AccumFragT
,
BLOCK_M
,
BLOCK_N
>
{}(
mfma_type_selector
<
AFragT
,
BFragT
,
AccumFragT
,
BLOCK_M
,
BLOCK_N
>
{}(
fragA
,
fragXa
,
fragB
,
fragXb
,
fragAcc
);
fragA
,
fragXa
,
fragB
,
fragXb
,
fragAcc
);
if
(
threadIdx
.
x
==
0
||
threadIdx
.
x
==
32
)
{
printf
(
"thread: %u -- xA: %x
\n
"
,
threadIdx
.
x
,
bit_cast
<
uint32_t
>
(
fragXa
));
printf
(
"thread: %u -- xB: %x
\n
"
,
threadIdx
.
x
,
bit_cast
<
uint32_t
>
(
fragXb
));
}
__syncthreads
();
__syncthreads
();
for
(
int
i
=
0
;
i
<
vectorSize
(
fragC
);
++
i
)
for
(
int
i
=
0
;
i
<
vectorSize
(
fragC
);
++
i
)
...
@@ -989,7 +995,7 @@ struct TestMXMFMA
...
@@ -989,7 +995,7 @@ struct TestMXMFMA
}
}
for
(
size_t
i
=
32
;
i
<
64
;
i
++
)
for
(
size_t
i
=
32
;
i
<
64
;
i
++
)
{
{
a_m_k
(
0
,
i
)
=
type_convert
<
ADataType
>
(
-
2.0
f
);
a_m_k
(
0
,
i
)
=
type_convert
<
ADataType
>
(
2.0
f
);
}
}
// printf("f8 1: %x \n", type_convert<ADataType>(1.0f).data);
// printf("f8 1: %x \n", type_convert<ADataType>(1.0f).data);
...
...
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