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
23e2309d
Commit
23e2309d
authored
Jan 07, 2025
by
illsilin
Browse files
enable smfmac test
parent
ebc4561f
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
CMakeLists.txt
CMakeLists.txt
+1
-1
test/CMakeLists.txt
test/CMakeLists.txt
+1
-1
test/smfmac_op/smfmac_op_xdl.cpp
test/smfmac_op/smfmac_op_xdl.cpp
+1
-1
No files found.
CMakeLists.txt
View file @
23e2309d
...
@@ -185,7 +185,7 @@ if (SUPPORTED_GPU_TARGETS MATCHES "gfx9")
...
@@ -185,7 +185,7 @@ if (SUPPORTED_GPU_TARGETS MATCHES "gfx9")
add_definitions
(
-DCK_USE_XDL
)
add_definitions
(
-DCK_USE_XDL
)
set
(
CK_USE_XDL
"ON"
)
set
(
CK_USE_XDL
"ON"
)
endif
()
endif
()
if
(
SUPPORTED_GPU_TARGETS MATCHES
"gfx94"
)
if
(
SUPPORTED_GPU_TARGETS MATCHES
"gfx94"
OR SUPPORTED_GPU_TARGETS MATCHES
"gfx95"
)
message
(
"Enabling FP8 gemms on native architectures"
)
message
(
"Enabling FP8 gemms on native architectures"
)
add_definitions
(
-DCK_USE_GFX94
)
add_definitions
(
-DCK_USE_GFX94
)
set
(
CK_USE_GFX94
"ON"
)
set
(
CK_USE_GFX94
"ON"
)
...
...
test/CMakeLists.txt
View file @
23e2309d
...
@@ -206,7 +206,7 @@ add_subdirectory(wrapper)
...
@@ -206,7 +206,7 @@ add_subdirectory(wrapper)
if
(
SUPPORTED_GPU_TARGETS MATCHES
"gfx11"
)
if
(
SUPPORTED_GPU_TARGETS MATCHES
"gfx11"
)
add_subdirectory
(
wmma_op
)
add_subdirectory
(
wmma_op
)
endif
()
endif
()
if
(
SUPPORTED_GPU_TARGETS MATCHES
"gfx942"
AND CK_HIP_VERSION_MAJOR GREATER_EQUAL 6 AND CK_HIP_VERSION_MINOR GREATER_EQUAL 2
)
# smfmac needs ROCm6.2
if
(
SUPPORTED_GPU_TARGETS MATCHES
"gfx942"
OR SUPPORTED_GPU_TARGETS MATCHES
"gfx950"
)
# smfmac needs ROCm6.2
add_subdirectory
(
smfmac_op
)
add_subdirectory
(
smfmac_op
)
endif
()
endif
()
add_subdirectory
(
position_embedding
)
add_subdirectory
(
position_embedding
)
...
...
test/smfmac_op/smfmac_op_xdl.cpp
View file @
23e2309d
...
@@ -40,7 +40,7 @@ class TestSmfmac : public ::testing::Test
...
@@ -40,7 +40,7 @@ class TestSmfmac : public ::testing::Test
void
Run
()
void
Run
()
{
{
bool
pass
=
true
;
bool
pass
=
true
;
if
(
ck
::
get_device_name
()
==
"gfx942"
)
if
(
ck
::
get_device_name
()
==
"gfx942"
||
ck
::
get_device_name
()
==
"gfx950"
)
{
{
constexpr
auto
matmul_default
=
ck
::
smfmac_op_util
::
matmul
<
Src1Type
,
constexpr
auto
matmul_default
=
ck
::
smfmac_op_util
::
matmul
<
Src1Type
,
Src1VecSize
,
Src1VecSize
,
...
...
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