add test layout

c58a3877 · Jing Zhang · e08e22b6 · c58a3877
Commit c58a3877 authored Jul 14, 2022 by Jing Zhang
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

test/grouped_gemm/grouped_gemm_fp16.cpp test/grouped_gemm/grouped_gemm_fp16.cpp +3 -0

No files found.
--- a/test/grouped_gemm/grouped_gemm_fp16.cpp
+++ b/test/grouped_gemm/grouped_gemm_fp16.cpp
@@ -55,6 +55,9 @@ int main()
    bool res = true;
    res = res && TestGroupedGemm<Row, Row, Row>();
+    res = res && TestGroupedGemm<Row, Col, Row>();
+    res = res && TestGroupedGemm<Col, Row, Row>();
+    res = res && TestGroupedGemm<Col, Col, Row>();
    std::cout << "TestGroupedGemm ..... " << (res ? "SUCCESS" : "FAILURE") << std::endl;