diff --git a/applications/reconstruct/src/main.cpp b/applications/reconstruct/src/main.cpp index 96cd4a8de1059928771d5dc4d94c3ea0c3da7fdd..8dd372a520f10c3a8e066c2ca6e726e164de7eb5 100644 --- a/applications/reconstruct/src/main.cpp +++ b/applications/reconstruct/src/main.cpp @@ -29,6 +29,8 @@ #include <ftl/registration.hpp> +#include <cuda_profiler_api.h> + #ifdef WIN32 #pragma comment(lib, "Rpcrt4.lib") #endif @@ -157,13 +159,21 @@ static void run(ftl::Configurable *root) { return true; }); + LOG(INFO) << "Shutting down..."; ftl::timer::stop(); net->shutdown(); + ftl::pool.stop(); + + cudaProfilerStop(); + + LOG(INFO) << "Deleting..."; + delete align; delete splat; delete virt; delete stream; delete net; + LOG(INFO) << "Done."; } int main(int argc, char **argv) {