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
94642acf
Commit
94642acf
authored
Aug 05, 2024
by
Jun Liu
Browse files
Merge branch 'develop' into amd-develop
parents
b9eb4de3
7f57b2e0
Changes
42
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
37 additions
and
33 deletions
+37
-33
profiler/src/profile_gemm_multiply_multiply.cpp
profiler/src/profile_gemm_multiply_multiply.cpp
+1
-1
test/smfmac_op/smfmac_op_xdl.cpp
test/smfmac_op/smfmac_op_xdl.cpp
+36
-32
No files found.
profiler/src/profile_gemm_multiply_multiply.cpp
View file @
94642acf
...
@@ -76,7 +76,7 @@ int profile_gemm_multiply_multiply(int argc, char* argv[])
...
@@ -76,7 +76,7 @@ int profile_gemm_multiply_multiply(int argc, char* argv[])
int
n_warmup
=
1
;
int
n_warmup
=
1
;
int
n_iter
=
10
;
int
n_iter
=
10
;
uint64_t
rotating
=
0
;
uint64_t
rotating
=
0
;
if
(
argc
==
1
8
)
if
(
argc
==
1
9
)
{
{
n_warmup
=
std
::
stoi
(
argv
[
16
]);
n_warmup
=
std
::
stoi
(
argv
[
16
]);
n_iter
=
std
::
stoi
(
argv
[
17
]);
n_iter
=
std
::
stoi
(
argv
[
17
]);
...
...
test/smfmac_op/smfmac_op_xdl.cpp
View file @
94642acf
...
@@ -13,6 +13,7 @@
...
@@ -13,6 +13,7 @@
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
#include "ck/tensor_operation/gpu/element/element_wise_operation.hpp"
#include "ck/tensor_operation/gpu/element/element_wise_operation.hpp"
#include "test/smfmac_op/smfmac_op_util.hpp"
#include "test/smfmac_op/smfmac_op_util.hpp"
#include "ck/host_utility/device_prop.hpp"
using
BF16
=
ck
::
bhalf_t
;
using
BF16
=
ck
::
bhalf_t
;
using
F16
=
ck
::
half_t
;
using
F16
=
ck
::
half_t
;
...
@@ -39,6 +40,8 @@ class TestSmfmac : public ::testing::Test
...
@@ -39,6 +40,8 @@ class TestSmfmac : public ::testing::Test
void
Run
()
void
Run
()
{
{
bool
pass
=
true
;
bool
pass
=
true
;
if
(
ck
::
get_device_name
()
==
"gfx942"
)
{
constexpr
auto
matmul_default
=
ck
::
smfmac_op_util
::
matmul
<
Src1Type
,
constexpr
auto
matmul_default
=
ck
::
smfmac_op_util
::
matmul
<
Src1Type
,
Src1VecSize
,
Src1VecSize
,
Src2Type
,
Src2Type
,
...
@@ -52,7 +55,8 @@ class TestSmfmac : public ::testing::Test
...
@@ -52,7 +55,8 @@ class TestSmfmac : public ::testing::Test
constexpr
auto
smfmac_kernel_container
=
std
::
make_tuple
(
matmul_default
);
constexpr
auto
smfmac_kernel_container
=
std
::
make_tuple
(
matmul_default
);
ck
::
static_for
<
0
,
std
::
tuple_size_v
<
decltype
(
smfmac_kernel_container
)
>
,
1
>
{}([
&
](
auto
i
)
{
ck
::
static_for
<
0
,
std
::
tuple_size_v
<
decltype
(
smfmac_kernel_container
)
>
,
1
>
{}(
[
&
](
auto
i
)
{
pass
&=
ck
::
smfmac_op_util
::
TestSmfmac
<
pass
&=
ck
::
smfmac_op_util
::
TestSmfmac
<
std
::
tuple_element_t
<
i
.
value
,
decltype
(
smfmac_kernel_container
)
>
,
std
::
tuple_element_t
<
i
.
value
,
decltype
(
smfmac_kernel_container
)
>
,
Src1Type
,
Src1Type
,
...
@@ -71,7 +75,7 @@ class TestSmfmac : public ::testing::Test
...
@@ -71,7 +75,7 @@ class TestSmfmac : public ::testing::Test
N
,
N
,
K
>
{}(
std
::
get
<
ck
::
Number
<
i
>
{}
>
(
smfmac_kernel_container
));
K
>
{}(
std
::
get
<
ck
::
Number
<
i
>
{}
>
(
smfmac_kernel_container
));
});
});
}
EXPECT_TRUE
(
pass
);
EXPECT_TRUE
(
pass
);
}
}
};
};
...
...
Prev
1
2
3
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