Fix checks for OpenMP halo_exchange_begin