Add special ROCM detection in OpenCL/inc_vendor.h

4 years ago · 3561e7b8d7
parent 247d1a91a5
commit 3561e7b8d7
1 changed files with 9 additions and 0 deletions
--- a/OpenCL/inc_vendor.h
+++ b/OpenCL/inc_vendor.h
@ -88,6 +88,10 @@
 #define IS_GENERIC
 #endif

+#if defined IS_AMD && HAS_VPERM == 1
+#define IS_ROCM
+#endif
+
 #define LOCAL_MEM_TYPE_LOCAL  1
 #define LOCAL_MEM_TYPE_GLOBAL 2

@ -152,6 +156,11 @@
 #define USE_ROTATE
 #endif

+#ifdef IS_ROCM
+#define USE_BITSELECT
+#define USE_ROTATE
+#endif
+
 #ifdef IS_OPENCL
 //#define USE_BITSELECT
 //#define USE_ROTATE