MCPcopy Create free account
hub / github.com/LBANN/lbann / start_data_store_mini_batch_exchange

Method start_data_store_mini_batch_exchange

src/data_readers/data_reader.cpp:297–311  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

295}
296
297void lbann::generic_data_reader::start_data_store_mini_batch_exchange(
298 El::Int current_position_in_data_set,
299 El::Int current_mini_batch_size,
300 bool at_new_epoch)
301{
302 // Make sure that every rank participates in the data store prior
303 // to seeing if the local rank's position is valid. Note that
304 // every rank will hold data that may be used in the last mini-batch
305 if (data_store_active()) {
306 m_data_store->start_exchange_mini_batch_data(current_position_in_data_set,
307 current_mini_batch_size,
308 at_new_epoch);
309 }
310 return;
311}
312
313void lbann::generic_data_reader::finish_data_store_mini_batch_exchange()
314{

Callers 2

Calls 1

Tested by

no test coverage detected