From 67553243ddc5abc4442e8e2d31e4d40ca8221ba1 Mon Sep 17 00:00:00 2001
From: Nicolas Pope <nwpope@utu.fi>
Date: Fri, 27 Sep 2019 08:28:42 +0300
Subject: [PATCH] Stop profiler before exit
---
applications/reconstruct/src/main.cpp | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/applications/reconstruct/src/main.cpp b/applications/reconstruct/src/main.cpp
index 96cd4a8de..8dd372a52 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) {
--
GitLab