We have a lot of those and should just use `thrust::square`.
We have a lot of those and should just use
thrust::square.