colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do not push empty string as a command argument, it breaks on Linux. (#174)

This commit is contained in:
Tian Jin 2020-06-18 21:04:55 +08:00 committed by GitHub
parent 82d2caa542
commit b28fcbc745
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/MainUtils.cpp
View File

@ -56,6 +56,13 @@ struct Command {
return *this;
}
// Append a single optional string argument.
Command &appendStrOpt(const llvm::Optional<std::string> &arg) {
if (arg.hasValue())
_args.emplace_back(arg.getValue());
return *this;
}
// Append a list of string arguments.
Command &appendList(const std::vector<std::string> &args) {
_args.insert(_args.end(), args.begin(), args.end());
@ -218,7 +225,7 @@ void compileModuleToSharedLibrary(
std::string modelObjPath = outputBaseName + ".o";
llvm::FileRemover modelObjRemover(modelObjPath);
std::string runtimeDirInclFlag;
llvm::Optional<std::string> runtimeDirInclFlag;
if (getEnvVar("RUNTIME_DIR").hasValue())
runtimeDirInclFlag = "-L" + getEnvVar("RUNTIME_DIR").getValue();
@ -228,7 +235,7 @@ void compileModuleToSharedLibrary(
.appendStr(modelObjPath)
.appendStr(constPackObjPath.getValueOr(""))
.appendList({"-o", outputBaseName + ".so"})
.appendStr(runtimeDirInclFlag)
.appendStrOpt(runtimeDirInclFlag)
.appendList({"-lEmbeddedDataLoader", "-lcruntime"})
.exec();
}