summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCedric Nugteren <web@cedricnugteren.nl>2018-01-06 19:26:38 +0100
committerCedric Nugteren <web@cedricnugteren.nl>2018-01-06 19:26:38 +0100
commit00687f8d81894dfc31be824601ac8159637ed468 (patch)
treea116e459ff3c5ff1d30af4cebceba02d68b3836c
parentf1e3b35541245e9c9561592c24705bb23000498b (diff)
Prevented half-precision batched routines from failing in the tests
-rw-r--r--test/routines/levelx/xaxpybatched.hpp2
-rw-r--r--test/routines/levelx/xgemmbatched.hpp4
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));
}
}