- validate dims
- set A.s correctly
- fix assert_same_host for case when myid() not in pids