diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-06 19:26:38 +0100 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-01-06 19:26:38 +0100 |
commit | 00687f8d81894dfc31be824601ac8159637ed468 (patch) | |
tree | a116e459ff3c5ff1d30af4cebceba02d68b3836c | |
parent | f1e3b35541245e9c9561592c24705bb23000498b (diff) |
Prevented half-precision batched routines from failing in the tests
-rw-r--r-- | test/routines/levelx/xaxpybatched.hpp | 2 | ||||
-rw-r--r-- | test/routines/levelx/xgemmbatched.hpp | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/test/routines/levelx/xaxpybatched.hpp b/test/routines/levelx/xaxpybatched.hpp index 9a09b47f..c7f6e40d 100644 --- a/test/routines/levelx/xaxpybatched.hpp +++ b/test/routines/levelx/xaxpybatched.hpp @@ -62,7 +62,7 @@ class TestXaxpyBatched { for (auto batch = size_t{0}; batch < args.batch_count; ++batch) { args.x_offsets[batch] = batch * PerBatchSizeX(args) + args.x_offset; args.y_offsets[batch] = batch * PerBatchSizeY(args) + args.y_offset; - args.alphas[batch] = args.alpha + Constant<T>(static_cast<double>(batch)); + args.alphas[batch] = args.alpha + Constant<T>(static_cast<double>(batch + 1)); } } diff --git a/test/routines/levelx/xgemmbatched.hpp b/test/routines/levelx/xgemmbatched.hpp index e237a018..b787ca27 100644 --- a/test/routines/levelx/xgemmbatched.hpp +++ b/test/routines/levelx/xgemmbatched.hpp @@ -86,8 +86,8 @@ class TestXgemmBatched { args.a_offsets[batch] = batch * PerBatchSizeA(args) + args.a_offset; args.b_offsets[batch] = batch * PerBatchSizeB(args) + args.b_offset; args.c_offsets[batch] = batch * PerBatchSizeC(args) + args.c_offset; - args.alphas[batch] = args.alpha + Constant<T>(static_cast<double>(batch)); - args.betas[batch] = args.beta + Constant<T>(static_cast<double>(batch)); + args.alphas[batch] = args.alpha + Constant<T>(static_cast<double>(batch + 1)); + args.betas[batch] = args.beta + Constant<T>(static_cast<double>(batch + 1)); } } |