Extend the XNN_RESTRICT macro to use __restrict__ or __restrict…

… for `gcc`/`clang` and `MSVC`, respectively. PiperOrigin-RevId: 633848169
google · May 15, 2024 · 7e91832 · 7e91832
1 parent 8d5f4e0
commit 7e91832
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/xnnpack/common.h b/src/xnnpack/common.h
@@ -241,8 +241,14 @@
   #define XNN_MIN_ELEMENTS(count) static count
 #endif
 
-#if defined(__cplusplus) || XNN_COMPILER_MSVC
-  #define XNN_RESTRICT
+#if defined(__cplusplus)
+  #if XNN_COMPILER_MSVC
+    #define XNN_RESTRICT __restrict
+  #elif XNN_COMPILER_GCC || XNN_COMPILER_CLANG
+    #define XNN_RESTRICT __restrict__
+  #else
+    #define XNN_RESTRICT
+  #endif
 #else
   #define XNN_RESTRICT restrict
 #endif