Add comments and test for atomic load/store and mem2reg.