move finialize to outside profiler area

5 years ago · 3dd6045f2e
parent 55af9daa9b
commit 3dd6045f2e
1 changed files with 4 additions and 3 deletions
--- a/mpi_daxpy_nvtx.cc
+++ b/mpi_daxpy_nvtx.cc
@ -201,19 +201,20 @@ int main(int argc, char **argv) {
    printf("%d/%d ALLSUM = %f\n", world_rank, world_size, sum);

    // cleanup
-    nvtxRangePushA("cleanup");
+    nvtxRangePushA("free");
    //cudaFree(d_x);
    //cudaFree(d_y);
    cudaFree(m_x);
    cudaFree(m_y);
    cudaFree(m_allx);
    cudaFree(m_ally);
-    cublasDestroy(handle);

-    MPI_Finalize();
    nvtxRangePop();

    cudaProfilerStop();

+    cublasDestroy(handle);
+    MPI_Finalize();
+
    return EXIT_SUCCESS;
 }