codebase / stimlib

  #include "cuda_runtime.h"

  #include "device_launch_parameters.h"

  #include <stim/cuda/cudatools/callable.h>

  
  #ifndef CUDA_THREADS_H
  #define CUDA_THREADS_H
  
  #define MAX_GRID        65535
  
  __device__ unsigned int ThreadIndex1D()
  {
      return blockIdx.y * gridDim.x * blockDim.x + blockIdx.x * blockDim.x + threadIdx.x;
  }
  
  dim3 GenGrid1D(unsigned int N, unsigned int blocksize = 128)
  {
      dim3 dimgrid;
  
      dimgrid.x = (N + blocksize - 1)/blocksize;
      dimgrid.y = 1;
      dimgrid.z = 1;
  
      if(dimgrid.x > MAX_GRID)
      {
          dimgrid.y = (dimgrid.x + MAX_GRID - 1) / MAX_GRID;
          dimgrid.x = MAX_GRID;
      }
  
      return dimgrid;
  
  }
  
  
  #endif