Add more testcases

45a6b898 · rocking · 19690f5b · 45a6b898
Commit 45a6b898 authored Jan 17, 2023 by rocking
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

test/gemm_layernorm/test_gemm_add_relu_add_layernorm_fp16.cpp .../gemm_layernorm/test_gemm_add_relu_add_layernorm_fp16.cpp +2 -1

No files found.
--- a/test/gemm_layernorm/test_gemm_add_relu_add_layernorm_fp16.cpp
+++ b/test/gemm_layernorm/test_gemm_add_relu_add_layernorm_fp16.cpp
@@ -32,7 +32,8 @@ class TestGemmAddReluAddLayernorm : public ::testing::Test
    void Run()
    {
-        std::vector<std::vector<ck::index_t>> lengths = {{1024, 1024, 1024}};
+        std::vector<std::vector<ck::index_t>> lengths = {
+            {1024, 1024, 1024}, {2048, 640, 640}, {1, 1, 1}};
        for(auto length : lengths)
        {