@@ -76,12 +76,15 @@ How do I verify that Open MPI has been built with CUDA support?
76
76
---------------------------------------------------------------
77
77
78
78
Verify that Open MPI has been built with cuda using ``ompi_info ``
79
+ with one of the following commands.
79
80
80
81
.. code-block :: sh
81
82
82
83
# Use ompi_info to verify cuda support in Open MPI
83
- shell$ ./ ompi_info | grep " MPI extensions"
84
+ shell$ ompi_info | grep " MPI extensions"
84
85
MPI extensions: affinity, cuda, pcollreq
86
+ shell$ ompi_info --parsable --all | grep mpi_built_with_cuda_support:value
87
+ mca:mpi:base:param:mpi_built_with_cuda_support:value:true
85
88
86
89
/////////////////////////////////////////////////////////////////////////
87
90
@@ -190,19 +193,6 @@ Libfabric's API.
190
193
191
194
/////////////////////////////////////////////////////////////////////////
192
195
193
-
194
- How can I tell if Open MPI was built with CUDA support?
195
- -------------------------------------------------------
196
-
197
- Use the ``ompi_info `` command:
198
-
199
- .. code-block ::
200
-
201
- shell$ ompi_info --parsable --all | grep mpi_built_with_cuda_support:value
202
- mca:mpi:base:param:mpi_built_with_cuda_support:value:true
203
-
204
- /////////////////////////////////////////////////////////////////////////
205
-
206
196
Can I get additional CUDA debug-level information at run-time?
207
197
--------------------------------------------------------------
208
198
@@ -580,6 +570,8 @@ example of using the CUDA-aware macro and run-time check.
580
570
581
571
int main(int argc, char *argv[])
582
572
{
573
+ MPI_Init(&argc, &argv);
574
+
583
575
printf("Compile time check:\n");
584
576
#if defined(MPIX_CUDA_AWARE_SUPPORT) && MPIX_CUDA_AWARE_SUPPORT
585
577
printf("This MPI library has CUDA-aware support.\n", MPIX_CUDA_AWARE_SUPPORT);
@@ -589,7 +581,7 @@ example of using the CUDA-aware macro and run-time check.
589
581
printf("This MPI library cannot determine if there is CUDA-aware support.\n");
590
582
#endif /* MPIX_CUDA_AWARE_SUPPORT */
591
583
592
- printf("Run time check:n");
584
+ printf("Run time check:\ n");
593
585
#if defined(MPIX_CUDA_AWARE_SUPPORT)
594
586
if (1 == MPIX_Query_cuda_support()) {
595
587
printf("This MPI library has CUDA-aware support.\n");
@@ -600,6 +592,8 @@ example of using the CUDA-aware macro and run-time check.
600
592
printf("This MPI library cannot determine if there is CUDA-aware support.\n");
601
593
#endif /* MPIX_CUDA_AWARE_SUPPORT */
602
594
595
+ MPI_Finalize();
596
+
603
597
return 0;
604
598
}
605
599
0 commit comments