Add missing #include
Correct __host__ __device__ & __thrust_hd_warning_disable__ annotations.