implement

Signed-off-by: chentong319 <[email protected]>
chentong319 · Oct 2, 2024 · a7c39a4 · a7c39a4
1 parent f8e2466
commit a7c39a4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/Compiler/CompilerOptions.cpp b/src/Compiler/CompilerOptions.cpp
@@ -1246,6 +1246,13 @@ void initCompilerConfig() {
     addCompilerConfig(CCM_SHARED_LIB_DEPS, extraLibs);
     addCompilerConfig(CCM_SHARED_LIB_PATH_DEPS, extraLibPaths);
   }
+
+  // Enable aggressive optimization for NNPA with -O3
+  if (OptimizationLevel == OptLevel::O3 &&
+      getTargetAccel().find("NNPA") != std::string::npos &&
+      getLLVMOption().find("enable-unsafe-fp-math") == std::string::npos) {
+    setLLVMOption(getLLVMOption() + " --enable-unsafe-fp-math");
+  }
 }
 
 } // namespace onnx_mlir