Add special ROCM detection in OpenCL/inc_vendor.h
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user