add 2 more test cases

cabe8d80 · wangshaojie6 · 0e243105 · cabe8d80 · cabe8d80
Commit cabe8d80 authored Sep 28, 2022 by wangshaojie6
Showing with 12 additions and 10 deletions

example/42_groupnorm/groupnorm_sigmoid_fp16.cpp example/42_groupnorm/groupnorm_sigmoid_fp16.cpp +10 -10

test/layernorm/test_groupnorm_fp16.cpp test/layernorm/test_groupnorm_fp16.cpp +2 -0

No files found.
--- a/example/42_groupnorm/groupnorm_sigmoid_fp16.cpp
+++ b/example/42_groupnorm/groupnorm_sigmoid_fp16.cpp
--- a/test/layernorm/test_groupnorm_fp16.cpp
+++ b/test/layernorm/test_groupnorm_fp16.cpp
@@ -26,6 +26,8 @@ class TestGroupnorm : public ::testing::Test
                                                         {256, 9, 9, 9, 9},
                                                         {1, 64, 64, 32, 10},
                                                         {1, 32, 32, 32, 20},
+                                                         {2, 32, 32, 32, 30},
+                                                         {2, 32, 32, 32, 40},
                                                         {1, 16, 16, 32, 40}};

        for(auto length : lengths)