0720a12bbe9398a344acffd7eb8f6d24ad491268
[alexxy/gromacs.git] / src / gromacs / hardware / printhardware.cpp
1 /*
2  * This file is part of the GROMACS molecular simulation package.
3  *
4  * Copyright (c) 2012,2013,2014,2015,2016, The GROMACS development team.
5  * Copyright (c) 2017,2018,2019,2020, by the GROMACS development team, led by
6  * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
7  * and including many others, as listed in the AUTHORS file in the
8  * top-level source directory and at http://www.gromacs.org.
9  *
10  * GROMACS is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Lesser General Public License
12  * as published by the Free Software Foundation; either version 2.1
13  * of the License, or (at your option) any later version.
14  *
15  * GROMACS is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * Lesser General Public License for more details.
19  *
20  * You should have received a copy of the GNU Lesser General Public
21  * License along with GROMACS; if not, see
22  * http://www.gnu.org/licenses, or write to the Free Software Foundation,
23  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
24  *
25  * If you want to redistribute modifications to GROMACS, please
26  * consider that scientific software is very special. Version
27  * control is crucial - bugs must be traceable. We will be happy to
28  * consider code for inclusion in the official distribution, but
29  * derived work must not be called official GROMACS. Details are found
30  * in the README & COPYING files - if they are missing, get the
31  * official version at http://www.gromacs.org.
32  *
33  * To help us fund GROMACS development, we humbly ask that you cite
34  * the research papers on the package. Check out http://www.gromacs.org.
35  */
36 #include "gmxpre.h"
37
38 #include "printhardware.h"
39
40 #include "config.h"
41
42 #include <cstdlib>
43
44 #include <string>
45 #include <vector>
46
47 #include "gromacs/hardware/cpuinfo.h"
48 #include "gromacs/hardware/device_management.h"
49 #include "gromacs/hardware/hardwaretopology.h"
50 #include "gromacs/hardware/hw_info.h"
51 #include "gromacs/hardware/identifyavx512fmaunits.h"
52 #include "gromacs/simd/support.h"
53 #include "gromacs/utility/basedefinitions.h"
54 #include "gromacs/utility/basenetwork.h"
55 #include "gromacs/utility/cstringutil.h"
56 #include "gromacs/utility/fatalerror.h"
57 #include "gromacs/utility/gmxmpi.h"
58 #include "gromacs/utility/logger.h"
59 #include "gromacs/utility/programcontext.h"
60 #include "gromacs/utility/stringutil.h"
61 #include "gromacs/utility/sysinfo.h"
62
63 #include "architecture.h"
64
65 //! Constant used to help minimize preprocessed code
66 static constexpr bool bGPUBinary = (GMX_GPU != 0);
67
68 /*! \internal \brief
69  * Returns the GPU information text, one GPU per line.
70  */
71 static std::string sprint_gpus(const std::vector<std::unique_ptr<DeviceInformation>>& deviceInfoList)
72 {
73     std::vector<std::string> gpuStrings(0);
74     for (const auto& deviceInfo : deviceInfoList)
75     {
76         gpuStrings.emplace_back("    " + getDeviceInformationString(*deviceInfo));
77     }
78     return gmx::joinStrings(gpuStrings, "\n");
79 }
80
81 /* Give a suitable fatal error or warning if the build configuration
82    and runtime CPU do not match. */
83 static void check_use_of_rdtscp_on_this_cpu(const gmx::MDLogger& mdlog, const gmx::CpuInfo& cpuInfo)
84 {
85     bool binaryUsesRdtscp = GMX_USE_RDTSCP;
86
87     const char* programName = gmx::getProgramContext().displayName();
88
89     if (cpuInfo.supportLevel() < gmx::CpuInfo::SupportLevel::Features)
90     {
91         if (binaryUsesRdtscp)
92         {
93             GMX_LOG(mdlog.warning)
94                     .asParagraph()
95                     .appendTextFormatted(
96                             "The %s executable was compiled to use the rdtscp CPU instruction. "
97                             "We cannot detect the features of your current CPU, but will proceed "
98                             "anyway. "
99                             "If you get a crash, rebuild GROMACS with the GMX_USE_RDTSCP=OFF CMake "
100                             "option.",
101                             programName);
102         }
103     }
104     else
105     {
106         bool cpuHasRdtscp = cpuInfo.feature(gmx::CpuInfo::Feature::X86_Rdtscp);
107
108         if (!cpuHasRdtscp && binaryUsesRdtscp)
109         {
110             gmx_fatal(FARGS,
111                       "The %s executable was compiled to use the rdtscp CPU instruction. "
112                       "However, this is not supported by the current hardware and continuing would "
113                       "lead to a crash. "
114                       "Please rebuild GROMACS with the GMX_USE_RDTSCP=OFF CMake option.",
115                       programName);
116         }
117
118         if (cpuHasRdtscp && !binaryUsesRdtscp)
119         {
120             GMX_LOG(mdlog.warning)
121                     .asParagraph()
122                     .appendTextFormatted(
123                             "The current CPU can measure timings more accurately than the code in\n"
124                             "%s was configured to use. This might affect your simulation\n"
125                             "speed as accurate timings are needed for load-balancing.\n"
126                             "Please consider rebuilding %s with the GMX_USE_RDTSCP=ON CMake "
127                             "option.",
128                             programName, programName);
129         }
130     }
131 }
132
133 static std::string detected_hardware_string(const gmx_hw_info_t* hwinfo, bool bFullCpuInfo)
134 {
135     std::string s;
136
137     const gmx::CpuInfo&          cpuInfo = *hwinfo->cpuInfo;
138     const gmx::HardwareTopology& hwTop   = *hwinfo->hardwareTopology;
139
140     s = gmx::formatString("\n");
141     s += gmx::formatString("Running on %d node%s with total", hwinfo->nphysicalnode,
142                            hwinfo->nphysicalnode == 1 ? "" : "s");
143     if (hwinfo->ncore_tot > 0)
144     {
145         s += gmx::formatString(" %d cores,", hwinfo->ncore_tot);
146     }
147     s += gmx::formatString(" %d logical cores", hwinfo->nhwthread_tot);
148     if (canPerformDeviceDetection(nullptr))
149     {
150         s += gmx::formatString(", %d compatible GPU%s", hwinfo->ngpu_compatible_tot,
151                                hwinfo->ngpu_compatible_tot == 1 ? "" : "s");
152     }
153     else if (bGPUBinary)
154     {
155         if (isDeviceDetectionEnabled())
156         {
157             s += gmx::formatString(" (GPU detection failed)");
158         }
159         else
160         {
161             s += gmx::formatString(" (GPU detection deactivated)");
162         }
163     }
164     s += gmx::formatString("\n");
165
166     if (hwinfo->nphysicalnode > 1)
167     {
168         /* Print per node hardware feature counts */
169         if (hwinfo->ncore_max > 0)
170         {
171             s += gmx::formatString("  Cores per node:           %2d", hwinfo->ncore_min);
172             if (hwinfo->ncore_max > hwinfo->ncore_min)
173             {
174                 s += gmx::formatString(" - %2d", hwinfo->ncore_max);
175             }
176             s += gmx::formatString("\n");
177         }
178         s += gmx::formatString("  Logical cores per node:   %2d", hwinfo->nhwthread_min);
179         if (hwinfo->nhwthread_max > hwinfo->nhwthread_min)
180         {
181             s += gmx::formatString(" - %2d", hwinfo->nhwthread_max);
182         }
183         s += gmx::formatString("\n");
184         if (bGPUBinary)
185         {
186             s += gmx::formatString("  Compatible GPUs per node: %2d", hwinfo->ngpu_compatible_min);
187             if (hwinfo->ngpu_compatible_max > hwinfo->ngpu_compatible_min)
188             {
189                 s += gmx::formatString(" - %2d", hwinfo->ngpu_compatible_max);
190             }
191             s += gmx::formatString("\n");
192             if (hwinfo->ngpu_compatible_tot > 0)
193             {
194                 if (hwinfo->bIdenticalGPUs)
195                 {
196                     s += gmx::formatString("  All nodes have identical type(s) of GPUs\n");
197                 }
198                 else
199                 {
200                     /* This message will also appear with identical GPU types
201                      * when at least one node has no GPU.
202                      */
203                     s += gmx::formatString(
204                             "  Different nodes have different type(s) and/or order of GPUs\n");
205                 }
206             }
207         }
208     }
209
210 #if GMX_LIB_MPI
211     int  rank;
212     char host[STRLEN];
213
214     gmx_gethostname(host, STRLEN);
215
216     MPI_Comm_rank(MPI_COMM_WORLD, &rank);
217
218     // TODO Use a wrapper around MPI_Get_processor_name instead.
219     s += gmx::formatString("Hardware detected on host %s (the node of MPI rank %d):\n", host, rank);
220 #else
221     s += gmx::formatString("Hardware detected:\n");
222 #endif
223     s += gmx::formatString("  CPU info:\n");
224
225     s += gmx::formatString("    Vendor: %s\n", cpuInfo.vendorString().c_str());
226
227     s += gmx::formatString("    Brand:  %s\n", cpuInfo.brandString().c_str());
228
229     if (bFullCpuInfo)
230     {
231         s += gmx::formatString("    Family: %d   Model: %d   Stepping: %d\n", cpuInfo.family(),
232                                cpuInfo.model(), cpuInfo.stepping());
233
234         s += gmx::formatString("    Features:");
235         for (auto& f : cpuInfo.featureSet())
236         {
237             s += gmx::formatString(" %s", gmx::CpuInfo::featureString(f).c_str());
238         }
239         s += gmx::formatString("\n");
240     }
241
242     if (cpuInfo.feature(gmx::CpuInfo::Feature::X86_Avx512F))
243     {
244         int avx512fmaunits = gmx::identifyAvx512FmaUnits();
245         s += gmx::formatString("    Number of AVX-512 FMA units:");
246         if (avx512fmaunits > 0)
247         {
248             s += gmx::formatString(" %d", avx512fmaunits);
249             if (avx512fmaunits == 1)
250             {
251                 s += gmx::formatString(" (AVX2 is faster w/o 2 AVX-512 FMA units)");
252             }
253         }
254         else
255         {
256             s += gmx::formatString(" Cannot run AVX-512 detection - assuming 2");
257         }
258         s += gmx::formatString("\n");
259     }
260
261     s += gmx::formatString("  Hardware topology: ");
262     switch (hwTop.supportLevel())
263     {
264         case gmx::HardwareTopology::SupportLevel::None: s += gmx::formatString("None\n"); break;
265         case gmx::HardwareTopology::SupportLevel::LogicalProcessorCount:
266             s += gmx::formatString("Only logical processor count\n");
267             break;
268         case gmx::HardwareTopology::SupportLevel::Basic: s += gmx::formatString("Basic\n"); break;
269         case gmx::HardwareTopology::SupportLevel::Full: s += gmx::formatString("Full\n"); break;
270         case gmx::HardwareTopology::SupportLevel::FullWithDevices:
271             s += gmx::formatString("Full, with devices\n");
272             break;
273     }
274
275     if (!hwTop.isThisSystem())
276     {
277         s += gmx::formatString("  NOTE: Hardware topology cached or synthetic, not detected.\n");
278         if (char* p = std::getenv("HWLOC_XMLFILE"))
279         {
280             s += gmx::formatString("        HWLOC_XMLFILE=%s\n", p);
281         }
282     }
283
284     if (bFullCpuInfo)
285     {
286         if (hwTop.supportLevel() >= gmx::HardwareTopology::SupportLevel::Basic)
287         {
288             s += gmx::formatString("    Sockets, cores, and logical processors:\n");
289
290             for (auto& socket : hwTop.machine().sockets)
291             {
292                 s += gmx::formatString("      Socket %2d:", socket.id);
293                 for (auto& c : socket.cores)
294                 {
295                     s += gmx::formatString(" [");
296                     for (auto& t : c.hwThreads)
297                     {
298                         s += gmx::formatString(" %3d", t.logicalProcessorId);
299                     }
300                     s += gmx::formatString("]");
301                 }
302                 s += gmx::formatString("\n");
303             }
304         }
305         if (hwTop.supportLevel() >= gmx::HardwareTopology::SupportLevel::Full)
306         {
307             s += gmx::formatString("    Numa nodes:\n");
308             for (auto& n : hwTop.machine().numa.nodes)
309             {
310                 s += gmx::formatString("      Node %2d (%zu bytes mem):", n.id, n.memory);
311                 for (auto& l : n.logicalProcessorId)
312                 {
313                     s += gmx::formatString(" %3d", l);
314                 }
315                 s += gmx::formatString("\n");
316             }
317             s += gmx::formatString("      Latency:\n          ");
318             for (std::size_t j = 0; j < hwTop.machine().numa.nodes.size(); j++)
319             {
320                 s += gmx::formatString(" %5zu", j);
321             }
322             s += gmx::formatString("\n");
323             for (std::size_t i = 0; i < hwTop.machine().numa.nodes.size(); i++)
324             {
325                 s += gmx::formatString("     %5zu", i);
326                 for (std::size_t j = 0; j < hwTop.machine().numa.nodes.size(); j++)
327                 {
328                     s += gmx::formatString(" %5.2f", hwTop.machine().numa.relativeLatency[i][j]);
329                 }
330                 s += gmx::formatString("\n");
331             }
332
333
334             s += gmx::formatString("    Caches:\n");
335             for (auto& c : hwTop.machine().caches)
336             {
337                 s += gmx::formatString(
338                         "      L%d: %zu bytes, linesize %d bytes, assoc. %d, shared %d ways\n",
339                         c.level, c.size, c.linesize, c.associativity, c.shared);
340             }
341         }
342         if (hwTop.supportLevel() >= gmx::HardwareTopology::SupportLevel::FullWithDevices)
343         {
344             s += gmx::formatString("    PCI devices:\n");
345             for (auto& d : hwTop.machine().devices)
346             {
347                 s += gmx::formatString(
348                         "      %04x:%02x:%02x.%1x  Id: %04x:%04x  Class: 0x%04x  Numa: %d\n", d.domain,
349                         d.bus, d.dev, d.func, d.vendorId, d.deviceId, d.classId, d.numaNodeId);
350             }
351         }
352     }
353
354     if (bGPUBinary && !hwinfo->deviceInfoList.empty())
355     {
356         s += gmx::formatString("  GPU info:\n");
357         s += gmx::formatString("    Number of GPUs detected: %d\n",
358                                static_cast<int>(hwinfo->deviceInfoList.size()));
359         s += sprint_gpus(hwinfo->deviceInfoList) + "\n";
360     }
361     return s;
362 }
363
364 void gmx_print_detected_hardware(FILE*                fplog,
365                                  const bool           warnToStdErr,
366                                  const gmx::MDLogger& mdlog,
367                                  const gmx_hw_info_t* hwinfo)
368 {
369     const gmx::CpuInfo& cpuInfo = *hwinfo->cpuInfo;
370
371     if (fplog != nullptr)
372     {
373         std::string detected;
374
375         detected = detected_hardware_string(hwinfo, TRUE);
376
377         fprintf(fplog, "%s\n", detected.c_str());
378     }
379
380     // Do not spam stderr with all our internal information unless
381     // there was something that actually went wrong; general information
382     // belongs in the logfile.
383
384     /* Check the compiled SIMD instruction set against that of the node
385      * with the lowest SIMD level support (skip if SIMD detection did not work)
386      */
387     if (cpuInfo.supportLevel() >= gmx::CpuInfo::SupportLevel::Features)
388     {
389         gmx::simdCheck(static_cast<gmx::SimdType>(hwinfo->simd_suggest_min), fplog, warnToStdErr);
390     }
391
392     /* For RDTSCP we only check on our local node and skip the MPI reduction, only on x86 */
393     if (gmx::c_architecture == gmx::Architecture::X86)
394     {
395         check_use_of_rdtscp_on_this_cpu(mdlog, cpuInfo);
396     }
397 }