square using CUDA JIT