__device__ int function(int a, int b)
{
  return b * b * a + 1;
}