Skip to content

GitLab

"...hubert/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "d07aebf504ce4a9f7637f7a5f9b1dce263ba0a34"
Unverified Commit ac0d8066 authored by Jianfeng Yan's avatar Jianfeng Yan Committed by GitHub
Browse files

Fix typo in batched gemm profiler (#176) 

* forgot passing BatchedCount in some profiler_batched_gemm

* delete default BatchCount
parent abf4bdb9
Hide whitespace changes
Inline Side-by-side
Showing with 15 additions and 8 deletions
profiler/include/profile_batched_gemm_impl.hpp
View file @ ac0d8066
......@@ -70,7 +70,7 @@ bool profile_batched_gemm_impl(int do_verification,
int StrideA,
int StrideB,
int StrideC,
int BatchCount = 1)
int BatchCount)
{
bool pass = true;
......
profiler/src/profile_batched_gemm.cpp
View file @ ac0d8066
......@@ -128,7 +128,8 @@ int profile_batched_gemm(int argc, char* argv[])
K,
(StrideA < 0) ? M : StrideA,
(StrideB < 0) ? N : StrideB,
(StrideC < 0) ? N : StrideC);
(StrideC < 0) ? N : StrideC,
BatchCount);
}
else if(data_type == GemmDataType::F16_F16_F16 && layout == GemmMatrixLayout::KM_NK_MN)
{
......@@ -147,7 +148,8 @@ int profile_batched_gemm(int argc, char* argv[])
K,
(StrideA < 0) ? M : StrideA,
(StrideB < 0) ? K : StrideB,
(StrideC < 0) ? N : StrideC);
(StrideC < 0) ? N : StrideC,
BatchCount);
}
else if(data_type == GemmDataType::BF16_BF16_BF16 && layout == GemmMatrixLayout::MK_KN_MN)
{
......@@ -206,7 +208,8 @@ int profile_batched_gemm(int argc, char* argv[])
K,
(StrideA < 0) ? M : StrideA,
(StrideB < 0) ? N : StrideB,
(StrideC < 0) ? N : StrideC);
(StrideC < 0) ? N : StrideC,
BatchCount);
}
else if(data_type == GemmDataType::BF16_BF16_BF16 && layout == GemmMatrixLayout::KM_NK_MN)
{
......@@ -225,7 +228,8 @@ int profile_batched_gemm(int argc, char* argv[])
K,
(StrideA < 0) ? M : StrideA,
(StrideB < 0) ? K : StrideB,
(StrideC < 0) ? N : StrideC);
(StrideC < 0) ? N : StrideC,
BatchCount);
}
else if(data_type == GemmDataType::F32_F32_F32 && layout == GemmMatrixLayout::MK_KN_MN)
{
......@@ -284,7 +288,8 @@ int profile_batched_gemm(int argc, char* argv[])
K,
(StrideA < 0) ? M : StrideA,
(StrideB < 0) ? N : StrideB,
(StrideC < 0) ? N : StrideC);
(StrideC < 0) ? N : StrideC,
BatchCount);
}
else if(data_type == GemmDataType::F32_F32_F32 && layout == GemmMatrixLayout::KM_NK_MN)
{
......@@ -303,7 +308,8 @@ int profile_batched_gemm(int argc, char* argv[])
K,
(StrideA < 0) ? M : StrideA,
(StrideB < 0) ? K : StrideB,
(StrideC < 0) ? N : StrideC);
(StrideC < 0) ? N : StrideC,
BatchCount);
}
else if(data_type == GemmDataType::INT8_INT8_INT8 && layout == GemmMatrixLayout::MK_KN_MN)
{
......@@ -362,7 +368,8 @@ int profile_batched_gemm(int argc, char* argv[])
K,
(StrideA < 0) ? M : StrideA,
(StrideB < 0) ? N : StrideB,
(StrideC < 0) ? N : StrideC);
(StrideC < 0) ? N : StrideC,
BatchCount);
}
else if(data_type == GemmDataType::INT8_INT8_INT8 && layout == GemmMatrixLayout::KM_NK_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