Support --march=native (onnx#3134)

* changes

Signed-off-by: Chen Tong <[email protected]>

* format

Signed-off-by: Chen Tong <[email protected]>

* linkage

Signed-off-by: Chen Tong <[email protected]>

* lib

Signed-off-by: Chen Tong <[email protected]>

---------

Signed-off-by: Chen Tong <[email protected]>

Author: chentong319

src/Compiler/CMakeLists.txt

@@ -56,6 +56,7 @@ add_onnx_mlir_library(OMCompilerOptions
   LINK_LIBS PUBLIC
   MLIRIR
   OMAccelerator
+  LLVMTargetParser
   )
 
 

src/Compiler/CompilerOptions.cpp

@@ -13,6 +13,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/Support/Debug.h"
+#include "llvm/TargetParser/Host.h"
 
 #include "ExternalUtil.hpp"
 #include "onnx-mlir/Compiler/OMCompilerRuntimeTypes.h"
@@ -173,7 +174,8 @@ static llvm::cl::opt<float, true> nnpaEpsilonOpt("nnpa-epsilon",
     llvm::cl::cat(OnnxMlirCommonOptions), llvm::cl::init(1e-5));
 
 static llvm::cl::opt<std::string, true> marchOpt("march",
-    llvm::cl::desc("Target architecture to generate code for."),
+    llvm::cl::desc("Target architecture to generate code for.\n"
+                   "--march=native will use the host's archituecture"),
     llvm::cl::value_desc("Target a specific architecture type"),
     llvm::cl::location(march), llvm::cl::cat(OnnxMlirCommonOptions),
     llvm::cl::ValueRequired);
@@ -1404,6 +1406,9 @@ void initCompilerConfig() {
 
   if (march == "z17")
     march = "arch15";
+
+  if (march == "native")
+    march = std::string(llvm::sys::getHostCPUName());
 }
 
 } // namespace onnx_mlir