1 #pragma once 2 3 #include <cuda.h> 4 #include <cuda_runtime.h> 5 6 __device__ void add(const float* a, const float* b, float* output); 7