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]);
|
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));
|
||||||
|
Loading…
Reference in New Issue
Block a user