Skip to content

Commit 7063d19

Browse files
committed
mca_cuda_convertor_init is called in MPI_Init if using pre-connect, so do not init kernel support until confirming buffer is gpu buffer.
1 parent 36d117b commit 7063d19

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

opal/datatype/opal_datatype_cuda.c

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -83,10 +83,9 @@ void mca_cuda_convertor_init(opal_convertor_t* convertor, const void *pUserBuf,
8383

8484
if (ftable.gpu_is_gpu_buffer(pUserBuf, convertor)) {
8585
convertor->flags |= CONVERTOR_CUDA;
86-
}
87-
88-
if (OPAL_SUCCESS != opal_cuda_kernel_support_init()) {
89-
opal_cuda_kernel_support_fini();
86+
if (OPAL_SUCCESS != opal_cuda_kernel_support_init()) {
87+
opal_cuda_kernel_support_fini();
88+
}
9089
}
9190

9291
convertor->stream = NULL;

0 commit comments

Comments
 (0)