correctly define permutation map

ginkgo-project · Aug 17, 2023 · 8393dca · 8393dca
1 parent 09bfa7f
commit 8393dca
Showing 1 changed file with 10 additions and 2 deletions.
diff --git 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>