I have this issue. And from what I read throughout other forums, many have this problem.
In my case when I turn the display off the WiFi is disconnected. When I turn it back on, I can see no WiFi, and my SIM data being used, WiFi reconnects in a couple of seconds. I have tried all the solutions...