ports/openmpi
ports/openmpi
1
1
Форкнуть 0
Код Релизы Активность

Fix CUDA registration where we run out of memory being allocated.

Просмотр исходного кода 
This commit was SVN r32297.
Этот коммит содержится в:
Rolf vandeVaart 2014-07-23 21:10:17 +00:00
родитель 87aa15a180
Коммит 3f703afb97
1 изменённых файлов: 2 добавлений и 2 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

4
ompi/mca/mpool/sm/mpool_sm_module.c
Просмотреть файл

@ -12,7 +12,7 @@
* Copyright (c) 2009-2012 Cisco Systems, Inc. All rights reserved. * Copyright (c) 2009-2012 Cisco Systems, Inc. All rights reserved.
* Copyright (c) 2011-2012 Los Alamos National Security, LLC. * Copyright (c) 2011-2012 Los Alamos National Security, LLC.
* All rights reserved. * All rights reserved.
* Copyright (c) 2011 NVIDIA Corporation. All rights reserved. * Copyright (c) 2011-2014 NVIDIA Corporation. All rights reserved.
* $COPYRIGHT$ * $COPYRIGHT$
* *
* Additional copyrights may follow * Additional copyrights may follow
@ -97,7 +97,7 @@ void* mca_mpool_sm_alloc(
} }
#if OPAL_CUDA_SUPPORT #if OPAL_CUDA_SUPPORT
if (flags & MCA_MPOOL_FLAGS_CUDA_REGISTER_MEM) { if ((flags & MCA_MPOOL_FLAGS_CUDA_REGISTER_MEM) && (NULL != mseg.mbs_start_addr)) {
mca_common_cuda_register(mseg.mbs_start_addr, size, mca_common_cuda_register(mseg.mbs_start_addr, size,
mpool->mpool_component->mpool_version.mca_component_name); mpool->mpool_component->mpool_version.mca_component_name);
} }