#pragma once #include #include __device__ void add(const float* a, const float* b, float* output);