Merge pull request #18 from AMDComputeLibraries/grid_launch_index

Grid launch index fix
Este commit está contenido en:
Siu Chi Chan
2016-03-23 11:26:28 -05:00
+22 -1
Ver fichero
@@ -428,6 +428,27 @@ __device__ inline float __dsqrt_rz(double x) {return hc::fast_math::sqrt(x); };
/** /**
* Kernel launching * Kernel launching
*/ */
#if __hcc_workweek__ >= 16123
#define hipThreadIdx_x (amp_get_local_id(0))
#define hipThreadIdx_y (amp_get_local_id(1))
#define hipThreadIdx_z (amp_get_local_id(2))
#define hipBlockIdx_x (hc_get_group_id(0))
#define hipBlockIdx_y (hc_get_group_id(1))
#define hipBlockIdx_z (hc_get_group_id(2))
#define hipBlockDim_x (amp_get_local_size(0))
#define hipBlockDim_y (amp_get_local_size(1))
#define hipBlockDim_z (amp_get_local_size(2))
#define hipGridDim_x (hc_get_num_groups(0))
#define hipGridDim_y (hc_get_num_groups(1))
#define hipGridDim_z (hc_get_num_groups(2))
#else
#define hipThreadIdx_x (amp_get_local_id(2)) #define hipThreadIdx_x (amp_get_local_id(2))
#define hipThreadIdx_y (amp_get_local_id(1)) #define hipThreadIdx_y (amp_get_local_id(1))
#define hipThreadIdx_z (amp_get_local_id(0)) #define hipThreadIdx_z (amp_get_local_id(0))
@@ -444,7 +465,7 @@ __device__ inline float __dsqrt_rz(double x) {return hc::fast_math::sqrt(x); };
#define hipGridDim_y (hc_get_num_groups(1)) #define hipGridDim_y (hc_get_num_groups(1))
#define hipGridDim_z (hc_get_num_groups(0)) #define hipGridDim_z (hc_get_num_groups(0))
#endif
#define __syncthreads() hc_barrier(CLK_LOCAL_MEM_FENCE) #define __syncthreads() hc_barrier(CLK_LOCAL_MEM_FENCE)