Skip to content

GitLab

Commit cabe8d80 authored by wangshaojie6's avatar wangshaojie6
Browse files

add 2 more test cases

parent 0e243105
Hide whitespace changes
Inline Side-by-side
Showing with 12 additions and 10 deletions
example/42_groupnorm/groupnorm_sigmoid_fp16.cpp
View file @ cabe8d80
...@@ -56,17 +56,17 @@ using DeviceInstance = ...@@ -56,17 +56,17 @@ using DeviceInstance =
Rank, Rank,
NumReduceDim, NumReduceDim,
1024, // BlockSize 1024, // BlockSize
1, // ClusterM 1, // ClusterM
1024, // ClusterK 1024, // ClusterK
1, // SliceM 1, // SliceM
32, // SliceK 32, // SliceK
1, // SrcVecDim (0=M, 1=K) 1, // SrcVecDim (0=M, 1=K)
2, // SrcScalarPerVector 2, // SrcScalarPerVector
1, // GammaVecDim (0=M, 1=K) 1, // GammaVecDim (0=M, 1=K)
2, // GammaScalarPerVector 2, // GammaScalarPerVector
1, // BetaVecDim (0=M, 1=K) 1, // BetaVecDim (0=M, 1=K)
2, // BetaScalarPerVector 2, // BetaScalarPerVector
2>; // OutScalarPerVector 2>; // OutScalarPerVector
int main(int argc, char* argv[]) int main(int argc, char* argv[])
{ {
......
test/layernorm/test_groupnorm_fp16.cpp
View file @ cabe8d80
...@@ -26,6 +26,8 @@ class TestGroupnorm : public ::testing::Test ...@@ -26,6 +26,8 @@ class TestGroupnorm : public ::testing::Test
{256, 9, 9, 9, 9}, {256, 9, 9, 9, 9},
{1, 64, 64, 32, 10}, {1, 64, 64, 32, 10},
{1, 32, 32, 32, 20}, {1, 32, 32, 32, 20},
{2, 32, 32, 32, 30},
{2, 32, 32, 32, 40},
{1, 16, 16, 32, 40}}; {1, 16, 16, 32, 40}};
for(auto length : lengths) for(auto length : lengths)
......
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