Commit 23e2309d authored by illsilin's avatar illsilin
Browse files

enable smfmac test

parent ebc4561f
...@@ -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")
......
...@@ -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)
......
...@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment