Update rabit.h