From 8393dca99f6ff8fe688e442661eb0714735bdee3 Mon Sep 17 00:00:00 2001
From: Marcel Koch <marcel.koch@kit.edu>
Date: Thu, 17 Aug 2023 10:01:09 +0200
Subject: [PATCH] correctly define permutation map

---
 dpcpp/distributed/partition_helpers_kernels.dp.cpp | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/dpcpp/distributed/partition_helpers_kernels.dp.cpp b/dpcpp/distributed/partition_helpers_kernels.dp.cpp
index 46f72c8ef58..8b0171cd349 100644
--- a/dpcpp/distributed/partition_helpers_kernels.dp.cpp
+++ b/dpcpp/distributed/partition_helpers_kernels.dp.cpp
@@ -46,7 +46,14 @@ namespace dpcpp {
 namespace partition_helpers {
 
 struct stride {
-    // Some version requires [] while some requires (), so I added both
+#if ONEDPL_VERSION_MAJOR >= 2022 && ONEDPL_VERSION_MINOR >= 1
+    template <typename Index>
+    Index operator()(const Index& i) const
+    {
+        return i * 2;
+    }
+#else
+    // Some older version require [] while some require (), so I added both
     template <typename Index>
     Index operator[](const Index& i) const
     {
@@ -56,8 +63,9 @@ struct stride {
     template <typename Index>
     Index operator()(const Index& i) const
     {
-        return operator[](i);
+        return i * 2;
     }
+#endif
 };
 
 template <typename GlobalIndexType>