1
1

Fix CUDA-aware support with sendi optimization. Need to make sure copy function

is initialized.

This commit was SVN r30437.
Этот коммит содержится в:
Rolf vandeVaart 2014-01-27 18:35:01 +00:00
родитель 0b5a1b5de1
Коммит 791a3a5ec6

Просмотреть файл

@ -1,5 +1,5 @@
/* /*
* Copyright (c) 2011-2013 NVIDIA Corporation. All rights reserved. * Copyright (c) 2011-2014 NVIDIA Corporation. All rights reserved.
* $COPYRIGHT$ * $COPYRIGHT$
* *
* Additional copyrights may follow * Additional copyrights may follow
@ -47,13 +47,16 @@ void mca_cuda_convertor_init(opal_convertor_t* convertor, const void *pUserBuf)
opal_cuda_support_init(); opal_cuda_support_init();
} }
/* This is needed to handle case where convertor is not fully initialized
* like when trying to do a sendi with convertor on the statck */
convertor->cbmemcpy = (memcpy_fct_t)&opal_cuda_memcpy;
/* If not enabled, then nothing else to do */ /* If not enabled, then nothing else to do */
if (!opal_cuda_enabled) { if (!opal_cuda_enabled) {
return; return;
} }
if (ftable.gpu_is_gpu_buffer(pUserBuf)) { if (ftable.gpu_is_gpu_buffer(pUserBuf)) {
convertor->cbmemcpy = (memcpy_fct_t)&opal_cuda_memcpy;
convertor->flags |= CONVERTOR_CUDA; convertor->flags |= CONVERTOR_CUDA;
} }
} }