|
19 | 19 | * and Technology (RIST). All rights reserved.
|
20 | 20 | * Copyright (c) 2018-2022 IBM Corporation. All rights reserved.
|
21 | 21 | * Copyright (c) 2021-2024 Nanook Consulting All rights reserved.
|
22 |
| - * Copyright (c) 2022-2024 Triad National Security, LLC. All rights |
| 22 | + * Copyright (c) 2022-2025 Triad National Security, LLC. All rights |
23 | 23 | * reserved.
|
24 | 24 | * $COPYRIGHT$
|
25 | 25 | *
|
@@ -194,6 +194,7 @@ static struct option ompioptions[] = {
|
194 | 194 | PMIX_OPTION_DEFINE(PRTE_CLI_DISABLE_RECOVERY, PMIX_ARG_NONE),
|
195 | 195 | PMIX_OPTION_DEFINE(PRTE_CLI_CONTINUOUS, PMIX_ARG_NONE),
|
196 | 196 | PMIX_OPTION_DEFINE("with-ft", PMIX_ARG_REQD),
|
| 197 | + PMIX_OPTION_DEFINE("disable-gpu-support", PMIX_ARG_NONE), |
197 | 198 |
|
198 | 199 | /* mpiexec mandated form launch key parameters - MPI 4.0 */
|
199 | 200 | PMIX_OPTION_DEFINE("initial-errhandler", PMIX_ARG_REQD),
|
@@ -1612,6 +1613,15 @@ static int parse_env(char **srcenv, char ***dstenv,
|
1612 | 1613 | }
|
1613 | 1614 | }
|
1614 | 1615 |
|
| 1616 | + if (NULL != (opt = pmix_cmd_line_get_param(results, "disable-gpu-support"))) { |
| 1617 | + rc = check_cache(&cache, &cachevals, "disable_gpu_support", "true"); |
| 1618 | + if (PRTE_SUCCESS != rc) { |
| 1619 | + PMIX_ARGV_FREE_COMPAT(cache); |
| 1620 | + PMIX_ARGV_FREE_COMPAT(cachevals); |
| 1621 | + return rc; |
| 1622 | + } |
| 1623 | + } |
| 1624 | + |
1615 | 1625 | if (pmix_cmd_line_is_taken(results, "display-comm") &&
|
1616 | 1626 | pmix_cmd_line_is_taken(results, "display-comm-finalize")) {
|
1617 | 1627 | PMIX_SETENV_COMPAT("OMPI_MCA_ompi_display_comm", "mpi_init,mpi_finalize", true, dstenv);
|
|
0 commit comments