diff --git a/clients/gtest/matmul_common.yaml b/clients/gtest/matmul_common.yaml index faa4e5853f..a37ebeeff9 100644 --- a/clients/gtest/matmul_common.yaml +++ b/clients/gtest/matmul_common.yaml @@ -1,4 +1,14 @@ Spmm Matrix Sizes: +- &one_matrix_size_range +- { M: 1, N: 1, K: 512, lda: 1, ldb: 1, ldc: 1, ldd: 1 } +- { M: 1, N: 1, K: 512, lda: 1, ldb: 512, ldc: 1, ldd: 1 } +- { M: 1, N: 1, K: 512, lda: 512, ldb: 1, ldc: 1, ldd: 1 } +- { M: 1, N: 1, K: 512, lda: 512, ldb: 512, ldc: 1, ldd: 1 } +- { M: 1, N: 1024, K: 512, lda: 1, ldb: 1, ldc: 1, ldd: 1 } +- { M: 1, N: 1024, K: 512, lda: 1, ldb: 512, ldc: 1, ldd: 1 } +- { M: 1, N: 1024, K: 512, lda: 512, ldb: 1, ldc: 1, ldd: 1 } +- { M: 1, N: 1024, K: 512, lda: 512, ldb: 512, ldc: 1, ldd: 1 } + - &small_matrix_size_range - { M: 8, N: 8, K: 8, lda: 8, ldb: 8, ldc: 8, ldd: 8 } - { M: 16, N: 16, K: 16, lda: 16, ldb: 16, ldc: 16, ldd: 16 } diff --git a/clients/gtest/matmul_gtest.yaml b/clients/gtest/matmul_gtest.yaml index a93f2be4c4..beea0555fe 100644 --- a/clients/gtest/matmul_gtest.yaml +++ b/clients/gtest/matmul_gtest.yaml @@ -48,6 +48,14 @@ Tests: - {name: alpha_beta_zero_NaN, category: pre_checkin, precision: *real_precisions, function: matmul, transA: N, transB: N, M: 256, N: 128, K: 64, alpha: [ .NaN, 2 ], beta: [ .NaN, 2 ] } +- name: matmul_one + category: quick + function: + matmul: *real_precisions + matrix_size: *one_matrix_size_range + transA_transB: *transA_transB_range + alpha_beta: *alpha_beta_range + - name: matmul_small category: quick function: