Merge pull request #167 from amcamd/refactor_tests_2

refactor tests: gemm, gemm_strided_batched

Author: amcamd

clients/common/arg_check.cpp

@@ -98,6 +98,24 @@ void gemm_arg_check(rocblas_status status, rocblas_int M, rocblas_int N, rocblas
     #endif
 }
 
+void gemm_strided_batched_arg_check(rocblas_status status, rocblas_int M, rocblas_int N, rocblas_int K, 
+    rocblas_int lda, rocblas_int ldb, rocblas_int ldc, rocblas_int batch_count)
+{
+    #ifdef GOOGLE_TEST
+    if (M == 0 || N == 0 || K == 0 || batch_count == 0)
+    {
+        ASSERT_EQ(status, rocblas_status_success);
+    }
+    else
+    {
+        ASSERT_EQ(status, rocblas_status_invalid_size);
+    }
+    #else
+    std::cerr << "ERROR in arguments M, N, K, lda, ldb, ldc, batch_count: ";
+    std::cerr << M << ',' << N << ',' << K << ',' << lda << ',' << ldb << ',' << ldc << batch_count << std::endl;
+    #endif
+}
+
 void geam_arg_check(rocblas_status status, rocblas_int M, rocblas_int N,
     rocblas_int lda, rocblas_int ldb, rocblas_int ldc)
 {

clients/include/arg_check.h

@@ -43,6 +43,8 @@
 
     void gemm_arg_check(rocblas_status status, rocblas_int M, rocblas_int N, rocblas_int K, rocblas_int lda, rocblas_int ldb, rocblas_int ldc);
 
+    void gemm_strided_batched_arg_check(rocblas_status status, rocblas_int M, rocblas_int N, rocblas_int K, rocblas_int lda, rocblas_int ldb, rocblas_int ldc, rocblas_int batch_count);
+
     void geam_arg_check(rocblas_status status, rocblas_int M, rocblas_int N, rocblas_int lda, rocblas_int ldb, rocblas_int ldc);
 
     void trsm_arg_check(rocblas_status status, rocblas_int M, rocblas_int N, rocblas_int lda, rocblas_int ldb);

clients/include/testing_geam.hpp

@@ -76,36 +76,42 @@ void testing_geam_bad_arg()
 
     {
         T *dA_null = nullptr;
+
         status = rocblas_geam<T>(handle, transA, transB, M, N, &h_alpha, dA_null, lda, &h_beta, dB, ldb, dC, ldc);
 
         verify_rocblas_status_invalid_pointer(status, "ERROR: A is nullptr");
     }
     {
         T *dB_null = nullptr;
+
         status = rocblas_geam<T>(handle, transA, transB, M, N, &h_alpha, dA, lda, &h_beta, dB_null, ldb, dC, ldc);
 
         verify_rocblas_status_invalid_pointer(status, "ERROR: B is nullptr");
     }
     {
         T *dC_null = nullptr;
+
         status = rocblas_geam<T>(handle, transA, transB, M, N, &h_alpha, dA, lda, &h_beta, dB, ldb, dC_null, ldc);
 
         verify_rocblas_status_invalid_pointer(status, "ERROR: C is nullptr");
     }
     {
         T *h_alpha_null = nullptr;
+
         status = rocblas_geam<T>(handle, transA, transB, M, N, h_alpha_null, dA, lda, &h_beta, dB, ldb, dC, ldc);
 
         verify_rocblas_status_invalid_pointer(status, "ERROR: h_alpha is nullptr");
     }
     {
         T *h_beta_null= nullptr;
+
         status = rocblas_geam<T>(handle, transA, transB, M, N, &h_alpha, dA, lda, h_beta_null, dB, ldb, dC, ldc);
 
         verify_rocblas_status_invalid_pointer(status, "ERROR: h_beta is nullptr");
     }
     {
         rocblas_handle handle_null = nullptr;
+
         status = rocblas_geam<T>(handle_null, transA, transB, M, N, &h_alpha, dA, lda, &h_beta, dB, ldb, dC, ldc);
 
         verify_rocblas_status_invalid_handle(status);