memcg: rework memcg_update_kmem_limit synchronization