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

This commit is contained in:
Jim Brennan 2020-08-17 16:41:15 +00:00
parent 092bfe7c8e
commit 4a7deae478

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));