bugfix: missing __host__ __device__