Call cudaDeviceReset from NVBENCH_MAIN.

2026-04-20 06:48:53 +00:00 · 2021-02-13 09:59:08 -05:00
parent 3b57127571
commit 2125ada770
1 changed files with 2 additions and 0 deletions
--- a/nvbench/main.cuh
+++ b/nvbench/main.cuh
@@ -2,6 +2,7 @@

 #include <nvbench/benchmark_base.cuh>
 #include <nvbench/benchmark_manager.cuh>
+#include <nvbench/cuda_call.cuh>
 #include <nvbench/detail/markdown_format.cuh>
 #include <nvbench/option_parser.cuh>

@@ -12,6 +13,7 @@
  try                                                                          \
  {                                                                            \
    NVBENCH_MAIN_BODY(argc, argv);                                             \
+    NVBENCH_CUDA_CALL(cudaDeviceReset());                                      \
    return 0;                                                                  \
  }                                                                            \
  catch (std::exception & e)                                                   \