YARN-10391. --module-gpu functionality is broken in container-executor. Contributed by Eric Badger
This commit is contained in:
parent
092bfe7c8e
commit
4a7deae478
|
@ -658,14 +658,14 @@ int main(int argc, char **argv) {
|
|||
assert_valid_setup(argv[0]);
|
||||
|
||||
int operation = -1;
|
||||
int ret = validate_arguments(argc, argv, &operation);
|
||||
|
||||
if (ret != 0) {
|
||||
flush_and_close_log_files();
|
||||
return ret;
|
||||
}
|
||||
|
||||
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) {
|
||||
case CHECK_SETUP:
|
||||
|
@ -831,6 +831,7 @@ int main(int argc, char **argv) {
|
|||
break;
|
||||
}
|
||||
|
||||
cleanup:
|
||||
if (exit_code) {
|
||||
fprintf(ERRORFILE, "Nonzero exit code=%d, error message='%s'\n", exit_code,
|
||||
get_error_message(exit_code));
|
||||
|
|
Loading…
Reference in New Issue