From dc2862aed7bed303a5b575d4f190231165a367e9 Mon Sep 17 00:00:00 2001
From: Pratik Nayak <pratikvn@protonmail.com>
Date: Tue, 16 Jan 2024 16:06:22 +0100
Subject: [PATCH] fixup! Review updates

---
 dpcpp/base/batch_multi_vector_kernels.hpp.inc | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/dpcpp/base/batch_multi_vector_kernels.hpp.inc b/dpcpp/base/batch_multi_vector_kernels.hpp.inc
index 6900792e4af..c41eafd7efd 100644
--- a/dpcpp/base/batch_multi_vector_kernels.hpp.inc
+++ b/dpcpp/base/batch_multi_vector_kernels.hpp.inc
@@ -14,8 +14,9 @@ __dpct_inline__ void scale_kernel(
         const int row = li / x.num_rhs;
         const int col = li % x.num_rhs;
 
-        x.values[row * x.stride + col] = alpha.values[map(row, col, stride)] *
-                                         x.values[row * x.stride + col];
+        x.values[row * x.stride + col] =
+            alpha.values[map(row, col, alpha.stride)] *
+            x.values[row * x.stride + col];
     }
 }