DO NOT MERGE [2.2] small writes batch force optimization

This can be a regular AtomicReference.