diff --git a/src/cli_main.cc b/src/cli_main.cc index 5c602f37..d18a77c2 100644 --- a/src/cli_main.cc +++ b/src/cli_main.cc @@ -444,6 +444,16 @@ class CLI { this->PrintHelp(); exit(1); } + + // FIXME:workaroud for local distributed machine learning on Occlum + for (int i = 2; i < argc; ++i) { + char name[128], val[128]; + if (sscanf(argv[i], "%[^=]=%s", name, val) == 2) { + //LOG(CONSOLE) << "setenv: " << std::string(name) << " = " << std::string(val); + setenv(std::string(name).c_str(), std::string(val).c_str(), 1); + } + } + for (int i = 0; i < argc; ++i) { std::string str {argv[i]}; if (str == "-h" || str == "--help") {