Batch update from KV now does batching to device