big-data/hadoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

YARN-10391. --module-gpu functionality is broken in container-executor. Contributed by Eric Badger

Browse Source 
(cherry picked from commit 4a7deae478)
This commit is contained in:
Jim Brennan 2020-08-17 16:41:15 +00:00
parent 49f8ae965e
commit 392a62278a
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-nodemanager/src/main/native/container-executor/impl/main.c
View File

@ -658,14 +658,14 @@ int main(int argc, char **argv) {
assert_valid_setup(argv[0]); assert_valid_setup(argv[0]);
int operation = -1; int operation = -1;
int ret = validate_arguments(argc, argv, &operation);
if (ret != 0) {
flush_and_close_log_files();
return ret;
}
int exit_code = 0; int exit_code = 0;
exit_code = validate_arguments(argc, argv, &operation);
if (exit_code != 0 || operation == -1) {
// if operation is still -1, the work was done in validate_arguments
// e.g. for --module-gpu
goto cleanup;
}
switch (operation) { switch (operation) {
case CHECK_SETUP: case CHECK_SETUP:
@ -831,6 +831,7 @@ int main(int argc, char **argv) {
break; break;
} }
cleanup:
if (exit_code) { if (exit_code) {
fprintf(ERRORFILE, "Nonzero exit code=%d, error message='%s'\n", exit_code, fprintf(ERRORFILE, "Nonzero exit code=%d, error message='%s'\n", exit_code,
get_error_message(exit_code)); get_error_message(exit_code));