Fix checks for OpenMP halo_exchange_{begin,end}