From e856fa720ff45d319225ef1ee4c3f7467e90e823 Mon Sep 17 00:00:00 2001
From: Pedram Alizadeh <pmohamma@amd.com>
Date: Tue, 25 Apr 2023 13:44:43 -0400
Subject: [PATCH] Revert "fixing the error message for mpirun when number of
 requested GPUs exceeds the limits (#33)" (#36)

This reverts commit e146460810a88b1a47b22308be63702485994fce.
---
 src/common.cu | 9 ---------
 1 file changed, 9 deletions(-)

diff --git a/src/common.cu b/src/common.cu
index eb0743a52f..332cc3f272 100644
--- a/src/common.cu
+++ b/src/common.cu
@@ -1127,13 +1127,11 @@ int main(int argc, char* argv[]) {
   }
 
   HIPCHECK(hipGetDeviceCount(&numDevices));
-#ifndef MPI_SUPPORT  
   if (nGpus > numDevices)
   {
       fprintf(stderr, "[ERROR] The number of requested GPUs (%d) is greater than the number of GPUs available (%d)\n", nGpus, numDevices);
       return testNcclError;
   }
-#endif
   if (minBytes > maxBytes) {
     fprintf(stderr, "invalid sizes for 'minbytes' and 'maxbytes': %llu > %llu\n",
            (unsigned long long)minBytes,
@@ -1156,14 +1154,7 @@ int main(int argc, char* argv[]) {
     return -1;
   }
 #ifdef MPI_SUPPORT
-  int nProcs = 1;
   MPI_Init(&argc, &argv);
-  MPI_Comm_size(MPI_COMM_WORLD, &nProcs);
-  if (nGpus * nProcs > numDevices)
-  {
-      fprintf(stderr, "[ERROR] The number of requested GPUs (%d) is greater than the number of GPUs available (%d)\n", nGpus*nProcs, numDevices);
-      return testNcclError;
-  }
 #endif
   TESTCHECK(run());
   return 0;