map-cuda-kernel-small.cu

/*
 * squaring map kernel that runs in 1 block
 */

/*
 * runs on and callable from the device
 */

__device__ float square(float x) {
	return x*x;
}

/*
 * runs on device, callable from anywhere
 */

__global__ void map(float* out, float* in, int size) {
	int index = threadIdx.x;
	if (index >= size) return;
	out[index] = square(in[index]);
}