2 months should be fine.

I went back to China and stayed there for about 2 months. When I came back, no question was asked about that.
And also from other posts regarding this questions, 2-3 months should be fine.
Enjoy your stay! I had a wonderful trip back to China.