diff --git a/clients/common/rocblas_gentest.py b/clients/common/rocblas_gentest.py index e2437e28b..f2b7ab4a1 100755 --- a/clients/common/rocblas_gentest.py +++ b/clients/common/rocblas_gentest.py @@ -195,6 +195,19 @@ def setdefaults(test): # Do not put constant defaults here -- use rocblas_common.yaml for that. # These are only for dynamic defaults # TODO: This should be ideally moved to YAML file, with eval'd expressions. + + if all([x in test for x in ('M', 'incx', 'strideScale')]) and test['function']=='ger_strided_batched': + test.setdefault('stride_x', int(test['M'] * abs(test['incx']) * + test['strideScale'])) + else: + test.setdefault('stride_x', 0) + + if all([x in test for x in ('N', 'incy', 'strideScale')]) and test['function']=='ger_strided_batched': + test.setdefault('stride_y', int(test['N'] * abs(test['incy']) * + test['strideScale'])) + else: + test.setdefault('stride_y', 0) + if test['transA'] == '*' or test['transB'] == '*': test.setdefault('lda', 0) test.setdefault('ldb', 0) diff --git a/clients/gtest/ger_gtest.cpp b/clients/gtest/ger_gtest.cpp index 65907aa0d..6154bc04d 100644 --- a/clients/gtest/ger_gtest.cpp +++ b/clients/gtest/ger_gtest.cpp @@ -7,12 +7,77 @@ #include "rocblas_datatype2string.hpp" #include "rocblas_test.hpp" #include "testing_ger.hpp" +#include "testing_ger_batched.hpp" +#include "testing_ger_strided_batched.hpp" #include "type_dispatch.hpp" #include #include namespace { + // possible gemv test cases + enum ger_test_type + { + GER, + GER_BATCHED, + GER_STRIDED_BATCHED, + }; + + //ger test template + template