Commit 94642acf authored by Jun Liu's avatar Jun Liu
Browse files

Merge branch 'develop' into amd-develop

parents b9eb4de3 7f57b2e0
...@@ -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 == 18) if(argc == 19)
{ {
n_warmup = std::stoi(argv[16]); n_warmup = std::stoi(argv[16]);
n_iter = std::stoi(argv[17]); n_iter = std::stoi(argv[17]);
......
...@@ -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);
} }
}; };
......
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