TRMV: Use the minimum x buffer size for copying to a temp buffer (#458)

pull/461/head
Cedric Nugteren 2023-05-07 20:03:16 +02:00 committed by GitHub
parent 3d0c227fa5
commit 4f24d92730
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions

View File

@ -1,4 +1,5 @@
Development version (next version)
- Fixes a minor issue with the expected input buffer size in the TRMV routine
- Fixes two small issues in the plotting script
- Modifications to improve performance on Qualcomm Adreno GPUs:
* Unique database entries for specific Adreno devices

View File

@ -36,8 +36,9 @@ void Xtrmv<T>::DoTrmv(const Layout layout, const Triangle triangle,
const Buffer<T> &x_buffer, const size_t x_offset, const size_t x_inc) {
// Creates a copy of X: a temporary scratch buffer
auto scratch_buffer = Buffer<T>(context_, n*x_inc + x_offset);
x_buffer.CopyTo(queue_, n*x_inc + x_offset, scratch_buffer);
const auto x_size = (1 + (n - 1) * x_inc) + x_offset;
auto scratch_buffer = Buffer<T>(context_, x_size);
x_buffer.CopyTo(queue_, x_size, scratch_buffer);
// The data is either in the upper or lower triangle
size_t is_upper = ((triangle == Triangle::kUpper && layout != Layout::kRowMajor) ||