Confronting predictions of the galaxy stellar mass function with observations at high redshift