| 170 | } |
| 171 | |
| 172 | void lbann::finalize_lbann(lbann_comm* comm) |
| 173 | { |
| 174 | #ifdef LBANN_HAS_NVSHMEM |
| 175 | nvshmem::finalize(); |
| 176 | #endif // LBANN_HAS_NVSHMEM |
| 177 | // MPI_Errhandler_free( &err_handle ); |
| 178 | #ifdef LBANN_HAS_DISTCONV |
| 179 | dc::finalize(); |
| 180 | #endif |
| 181 | #ifdef LBANN_HAS_DNN_LIB |
| 182 | dnn_lib::destroy(); |
| 183 | #endif |
| 184 | #ifdef LBANN_HAS_EMBEDDED_PYTHON |
| 185 | python::finalize(); |
| 186 | #endif |
| 187 | #ifdef LBANN_HAS_NVSHMEM |
| 188 | nvshmem::finalize(); |
| 189 | #endif // LBANN_HAS_SHMEM |
| 190 | #ifdef LBANN_HAS_SHMEM |
| 191 | shmem_finalize(); |
| 192 | #endif // LBANN_HAS_SHMEM |
| 193 | #ifdef LBANN_HAS_CALIPER |
| 194 | finalize_caliper(); |
| 195 | #endif |
| 196 | |
| 197 | if (comm != nullptr) { |
| 198 | delete comm; |
| 199 | } |
| 200 | El::Finalize(); |
| 201 | } |
| 202 | |
| 203 | auto lbann::initialize(int& argc, char**& argv) -> world_comm_ptr |
| 204 | { |