.. _program_listing_file_src_flamegpu_util_cleanup.cu:

Program Listing for File cleanup.cu
===================================

|exhale_lsh| :ref:`Return to documentation for file <file_src_flamegpu_util_cleanup.cu>` (``src/flamegpu/util/cleanup.cu``)

.. |exhale_lsh| unicode:: U+021B0 .. UPWARDS ARROW WITH TIP LEFTWARDS

.. code-block:: cpp

   #include "flamegpu/util/cleanup.h"
   
   #ifdef FLAMEGPU_ENABLE_MPI
   #include <mpi.h>
   #endif
   
   #include <cuda_runtime.h>
   #include "flamegpu/simulation/detail/CUDAErrorChecking.cuh"
   #include "flamegpu/detail/JitifyCache.h"
   
   namespace flamegpu {
   namespace util {
   
   void cleanup() {
   #ifdef FLAMEGPU_ENABLE_MPI
       int init_flag = 0;
       int fin_flag = 0;
       // MPI can only be init and finalized once
       MPI_Initialized(&init_flag);
       MPI_Finalized(&fin_flag);
       if (init_flag && !fin_flag) {
           MPI_Finalize();
       }
   #endif
       int originalDevice = 0;
       gpuErrchk(cudaGetDevice(&originalDevice));
       // Reset all cuda devices for memcheck / profiling purposes.
       int devices = 0;
       gpuErrchk(cudaGetDeviceCount(&devices));
       // @todo - this would be better to be only devices touched by flamegpu since the last call to cleanup.
       for (int device = 0; device < devices; ++device) {
           gpuErrchk(cudaSetDevice(device));
           gpuErrchk(cudaDeviceReset());
       }
       // resume the old device, but do not create a new context via reset or memsets
       gpuErrchk(cudaSetDevice(originalDevice));
   }
   
   void clearRTCDiskCache() {
       detail::JitifyCache::clearDiskCache();
   }
   
   }  // namespace util
   }  // namespace flamegpu