Skip to content

Commit 81408d7

Browse files
committed
pml/ucx: Remove unused cuda code
Signed-off-by: William Zhang <wilzhang@amazon.com>
1 parent 269ea5b commit 81408d7

File tree

2 files changed

+0
-16
lines changed

2 files changed

+0
-16
lines changed

ompi/mca/pml/ucx/pml_ucx.c

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,6 @@
2525
#include "ompi/runtime/ompi_spc.h"
2626
#include "ompi/mca/pml/base/pml_base_bsend.h"
2727
#include "opal/mca/common/ucx/common_ucx.h"
28-
#if OPAL_CUDA_SUPPORT
29-
#include "opal/cuda/common_cuda.h"
30-
#endif /* OPAL_CUDA_SUPPORT */
3128
#include "pml_ucx_request.h"
3229

3330
#include <inttypes.h>
@@ -265,26 +262,14 @@ int mca_pml_ucx_open(void)
265262
}
266263

267264
ompi_pml_ucx.request_size = attr.request_size;
268-
ompi_pml_ucx.cuda_initialized = false;
269265

270-
#if HAVE_UCP_ATTR_MEMORY_TYPES && OPAL_CUDA_SUPPORT
271-
if (attr.memory_types & UCS_BIT(UCS_MEMORY_TYPE_CUDA)) {
272-
mca_common_cuda_stage_one_init();
273-
ompi_pml_ucx.cuda_initialized = true;
274-
}
275-
#endif
276266
return OMPI_SUCCESS;
277267
}
278268

279269
int mca_pml_ucx_close(void)
280270
{
281271
PML_UCX_VERBOSE(1, "mca_pml_ucx_close");
282272

283-
#if OPAL_CUDA_SUPPORT
284-
if (ompi_pml_ucx.cuda_initialized) {
285-
mca_common_cuda_fini();
286-
}
287-
#endif
288273
if (ompi_pml_ucx.ucp_context != NULL) {
289274
ucp_cleanup(ompi_pml_ucx.ucp_context);
290275
ompi_pml_ucx.ucp_context = NULL;

ompi/mca/pml/ucx/pml_ucx.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,6 @@ struct mca_pml_ucx_module {
5757
mca_pml_ucx_freelist_t convs;
5858

5959
int priority;
60-
bool cuda_initialized;
6160
bool request_leak_check;
6261
uint32_t op_attr_nonblocking;
6362
};

0 commit comments

Comments
 (0)