| 295 | } |
| 296 | |
| 297 | void 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 | |
| 313 | void lbann::generic_data_reader::finish_data_store_mini_batch_exchange() |
| 314 | { |
no test coverage detected