Optimizations: use __restrict__ instead of Kokkos restrict macro as i…

…t seems to be more reliable
kokkos · Mar 14, 2024 · 26442c4 · 26442c4
1 parent f02060d
commit 26442c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pykokkos/core/optimizations/restrict_views.py b/pykokkos/core/optimizations/restrict_views.py
@@ -111,7 +111,7 @@ def get_restrict_ptr_type(view_type: str) -> Tuple[cppast.ClassType, int]:
     rank: int = view_dtype.count("*")
 
     # Replace "**" with "*"
-    restrict_type: str = view_dtype.replace(rank * "*", "*") + " KOKKOS_RESTRICT"
+    restrict_type: str = view_dtype.replace(rank * "*", "*") + " __restrict__"
     decltype = cppast.ClassType(restrict_type)
 
     return decltype, rank