portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-10 07:55:35 +00:00
Code Releases Activity

Pull the trigger on offload

Browse Source
This commit is contained in:
paboyle 2018-03-18 14:45:29 +00:00
parent b1c02ec310
commit 3a3e3cac40
1 changed files with 23 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lib/threads/Pragmas.h
View File

@ -76,15 +76,26 @@ Author: paboyle <paboyle@ph.ed.ac.uk>
// Accelerator primitives; fall back to threading // Accelerator primitives; fall back to threading
////////////////////////////////////////////////////////////////////////////////// //////////////////////////////////////////////////////////////////////////////////
#ifdef GRID_NVCC #ifdef GRID_NVCC
template<typename lambda> __global__
void LambdaApply(int Num, lambda &Lambda)
{
int ss = blockIdx.x;
if ( ss < Num ) {
Lambda(ss);
}
}
#define accelerator_exec( ... ) \ #define accelerator_exec( ... ) \
auto lambda = [=] accelerator (void) { \ auto lambda = [=] accelerator (void) mutable { \
__VA_ARGS__; \ __VA_ARGS__; \
}; \ }; \
lambda(); lambda();
#define accelerator __host__ __device__ #define accelerator __host__ __device__
#define accelerator_inline __host__ __device__ inline #define accelerator_inline __host__ __device__ inline
// FIXME ; need to make this a CUDA kernel call
#if 0
#define accelerator_loop( iterator, range, ... ) \ #define accelerator_loop( iterator, range, ... ) \
typedef decltype(range.begin()) Iterator; \ typedef decltype(range.begin()) Iterator; \
auto lambda = [=] accelerator (Iterator iterator) mutable { \ auto lambda = [=] accelerator (Iterator iterator) mutable { \
@ -93,6 +104,16 @@ Author: paboyle <paboyle@ph.ed.ac.uk>
for(auto it=range.begin();it<range.end();it++){ \ for(auto it=range.begin();it<range.end();it++){ \
lambda(it); \ lambda(it); \
} }
#else
#define accelerator_loop( iterator, range, ... ) \
typedef decltype(range.begin()) Iterator; \
auto lambda = [=] accelerator (Iterator iterator) mutable { \
__VA_ARGS__; \
}; \
Iterator num = range.end(); \
LambdaApply<<<num,1>>>(num,lambda);
#endif
#define cpu_loop( iterator, range, ... ) thread_loop( (auto iterator = range.begin();iterator<range.end();iterator++), { __VA_ARGS__ }); #define cpu_loop( iterator, range, ... ) thread_loop( (auto iterator = range.begin();iterator<range.end();iterator++), { __VA_ARGS__ });
#define NVCC_DECLARE_VECTOR_ACCESSOR(Type) \ #define NVCC_DECLARE_VECTOR_ACCESSOR(Type) \