fix nvshmem free

Author: emrgncr

nvshmem/mpi_all_to_all_builtin/mpi_all_to_all_builtin.cu

@@ -92,8 +92,8 @@ int main(int argc, char *argv[]) {
  random_fill_host(tmp, data_size * size);
 
  nvshmemx_putmem_on_stream(sendbuff, tmp, data_size * size, mype_node, stream);
- nvshmemx_barrier_all_on_stream(stream);
  CUDA_CHECK(cudaStreamSynchronize(stream));
+ nvshmemx_barrier_all_on_stream(stream);
 
  free(tmp);
 
@@ -134,7 +134,7 @@ int main(int argc, char *argv[]) {
 
  // free device buffers
 
- CUDA_CHECK(cudaFree(sendbuff));
+ nvshmem_free(sendbuff);
  nvshmem_free(recvbuff);
 
  nvshmem_finalize();