'Return the current state of the on-host and on-device RNGs.
Returns:
(np.array, np.array): the on-host and on-device RNG state vectors,
respectively'

def rng_get_state(self):

    dev_state = self._get_rand_state_dev()
    dev_state_local = np.zeros(NervanaGPU._RNG_POOL_SIZE).astype(np.uint32)
    drv.memcpy_dtoh(dev_state_local, dev_state)
    return (self.rng.get_state(), dev_state_local)
