Skip to content

GitLab

Commit 73fc0421 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Allow enabling only fp16 gemm instances in profiler

parent 194c9f1b
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
profiler/src/profile_gemm.cpp
View file @ 73fc0421
......@@ -121,6 +121,8 @@ int profile_gemm(int argc, char* argv[])
return pass ? 0 : 1;
};
if (false) ;
#ifdef __fp32__
if(data_type == GemmDataType::F32_F32_F32 && layout == GemmMatrixLayout::MK_KN_MN)
{
return profile(Row{}, Row{}, Row{}, F32{}, F32{}, F32{}, F32{});
......@@ -137,6 +139,8 @@ int profile_gemm(int argc, char* argv[])
{
return profile(Col{}, Col{}, Row{}, F32{}, F32{}, F32{}, F32{});
}
#endif
#ifdef __fp16__
else if(data_type == GemmDataType::F16_F16_F16 && layout == GemmMatrixLayout::MK_KN_MN)
{
return profile(Row{}, Row{}, Row{}, F16{}, F16{}, F32{}, F16{});
......@@ -153,6 +157,7 @@ int profile_gemm(int argc, char* argv[])
{
return profile(Col{}, Col{}, Row{}, F16{}, F16{}, F32{}, F16{});
}
#endif
#ifdef __bf16__
else if(data_type == GemmDataType::BF16_BF16_BF16 && layout == GemmMatrixLayout::MK_KN_MN)
{
......
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