From e26482833aa16c4c4d2d6bf07fc701dcdf4e0204 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Thu, 25 Jul 2024 04:17:11 -0400
Subject: [PATCH] Revert "debug"

This reverts commit 51a94b6b75de85cfd98e233fff6c9876fc73b377.
---
 .github/workflows/test_cc.yml | 2 +-
 source/op/pt/comm.cc          | 7 -------
 2 files changed, 1 insertion(+), 8 deletions(-)

diff --git a/.github/workflows/test_cc.yml b/.github/workflows/test_cc.yml
index 05344f6994..ebbfc4d960 100644
--- a/.github/workflows/test_cc.yml
+++ b/.github/workflows/test_cc.yml
@@ -53,7 +53,7 @@ jobs:
       env:
         DP_BUILD_TESTING: 1
       if: ${{ !matrix.check_memleak }}
-    - run: pytest -s --cov=deepmd source/lmp/tests
+    - run: pytest --cov=deepmd source/lmp/tests
       env:
         OMP_NUM_THREADS: 1
         TF_INTRA_OP_PARALLELISM_THREADS: 1
diff --git a/source/op/pt/comm.cc b/source/op/pt/comm.cc
index e886ab245a..a25dfbd542 100644
--- a/source/op/pt/comm.cc
+++ b/source/op/pt/comm.cc
@@ -82,7 +82,6 @@ class Border : public torch::autograd::Function<Border> {
     int nghost = nghost_tensor.item<int>();
     int ntotal = nlocal + nghost;
     torch::Tensor recv_g1_tensor = g1;
-    printf("nlocal: %d, nghost: %d, lineno: %d\n", nlocal, nghost, __LINE__);
 
 #ifdef USE_MPI
     int mpi_init = 0;
@@ -121,9 +120,6 @@ class Border : public torch::autograd::Function<Border> {
     for (int iswap = 0; iswap < nswap; ++iswap) {
       int nrecv = recvnum[iswap];
       int nsend = sendnum[iswap];
-      printf("me: %d, iswap: %d, nrecv: %d, nsend: %d, lineno: %d\n", me, iswap,
-             nrecv, nsend, __LINE__);
-
       torch::Tensor isendlist =
           torch::from_blob(sendlist[iswap], {nsend}, int32_options)
               .to(recv_g1_tensor.device());
@@ -218,7 +214,6 @@ class Border : public torch::autograd::Function<Border> {
     MPI_Comm_size(world, &world_size);
     MPI_Datatype mpi_type = get_mpi_type<FPTYPE>();
     MPI_Request request;
-    printf("world_size: %d, rank: %d, lineno: %d\n", world_size, me, __LINE__);
 #if defined(GOOGLE_CUDA) || defined(TENSORFLOW_USE_ROCM)
     if (world_size != 1) {
       int version, subversion;
@@ -270,8 +265,6 @@ class Border : public torch::autograd::Function<Border> {
     for (int iswap = nswap - 1; iswap >= 0; --iswap) {
       int nrecv = recvnum[iswap];
       int nsend = sendnum[iswap];
-      printf("me: %d, iswap: %d, nrecv: %d, nsend: %d, lineno: %d\n", me, iswap,
-             nrecv, nsend, __LINE__);
 
       torch::Tensor irecvlist;
       if (nrecv) {