Больше таймеров

src/main.cpp

@@ -57,7 +57,7 @@ int main(int argc, char** argv) {
     MPI_Comm_rank(MPI_COMM_WORLD, &rank);
     MPI_Comm_size(MPI_COMM_WORLD, &size);
 
-    // Читаем количество CPU потоков из переменной окружения (по умолчанию 2)
+    // Читаем количество CPU потоков из переменной окружения
     int num_cpu_threads = 2;
     const char* env_threads = std::getenv("NUM_CPU_THREADS");
     if (env_threads) {
@@ -80,15 +80,29 @@ int main(int argc, char** argv) {
 
     std::vector<Record> local_records;
 
+    // ====== ТАЙМЕРЫ ======
+    double time_load_data = 0.0;
+    double time_distribute = 0.0;
+
     if (rank == 0) {
         std::cout << "Rank 0 loading CSV..." << std::endl;
 
+        // Таймер загрузки данных
+        double t_load_start = MPI_Wtime();
+        
         // Запускаем из build
         auto records = load_csv("../data/data.csv");
 
         auto days = group_by_day(records);
         auto parts = split_days(days, size);
+        
+        time_load_data = MPI_Wtime() - t_load_start;
+        std::cout << "Rank 0: Data loading time: " << std::fixed << std::setprecision(3) 
+                  << time_load_data << "s" << std::endl;
 
+        // Таймер рассылки данных
+        double t_distribute_start = MPI_Wtime();
+        
         // Рассылаем данные
         for (int r = 0; r < size; r++) {
             auto vec = select_records_for_rank(days, parts[r]);
@@ -103,8 +117,13 @@ int main(int argc, char** argv) {
             MPI_Send(&count, 1, MPI_INT, r, 0, MPI_COMM_WORLD);
             MPI_Send(vec.data(), count * sizeof(Record), MPI_BYTE, r, 1, MPI_COMM_WORLD);
         }
+        
+        time_distribute = MPI_Wtime() - t_distribute_start;
     }
     else {
+        // Таймер получения данных
+        double t_receive_start = MPI_Wtime();
+        
         // Принимает данные
         int count = 0;
         MPI_Recv(&count, 1, MPI_INT, 0, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
@@ -112,9 +131,15 @@ int main(int argc, char** argv) {
         local_records.resize(count);
         MPI_Recv(local_records.data(), count * sizeof(Record),
                  MPI_BYTE, 0, 1, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
+        
+        time_distribute = MPI_Wtime() - t_receive_start;
     }
 
     MPI_Barrier(MPI_COMM_WORLD);
+    
+    // Вывод времени рассылки/получения данных
+    std::cout << "Rank " << rank << ": Data distribution time: " << std::fixed 
+              << std::setprecision(3) << time_distribute << "s" << std::endl;
 
     std::cout << "Rank " << rank << " received "
               << local_records.size() << " records" << std::endl;