bda/gpu-mpi-tests
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

move finialize to outside profiler area

Browse Source
This commit is contained in:
Bryce Allen
2020-08-07 13:02:07 -04:00
parent 55af9daa9b
commit 3dd6045f2e
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
mpi_daxpy_nvtx.cc
View File

@@ -201,19 +201,20 @@ int main(int argc, char **argv) {
printf("%d/%d ALLSUM = %f\n", world_rank, world_size, sum); printf("%d/%d ALLSUM = %f\n", world_rank, world_size, sum);
// cleanup // cleanup
nvtxRangePushA("cleanup"); nvtxRangePushA("free");
//cudaFree(d_x); //cudaFree(d_x);
//cudaFree(d_y); //cudaFree(d_y);
cudaFree(m_x); cudaFree(m_x);
cudaFree(m_y); cudaFree(m_y);
cudaFree(m_allx); cudaFree(m_allx);
cudaFree(m_ally); cudaFree(m_ally);
cublasDestroy(handle);
MPI_Finalize();
nvtxRangePop(); nvtxRangePop();
cudaProfilerStop(); cudaProfilerStop();
cublasDestroy(handle);
MPI_Finalize();
return EXIT_SUCCESS; return EXIT_SUCCESS;
} }