implement (#2963)

chentong319 · web-flow · commit ccf955244f19 · 2024-10-02T13:17:29.000-04:00
Signed-off-by: chentong319 &lt;chentong@us.ibm.com&gt;
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

Original file line number	Diff line number	Diff line change
`@@ -1246,6 +1246,13 @@ void initCompilerConfig() {`
`1246`	`1246`	`addCompilerConfig(CCM_SHARED_LIB_DEPS, extraLibs);`
`1247`	`1247`	`addCompilerConfig(CCM_SHARED_LIB_PATH_DEPS, extraLibPaths);`
`1248`	`1248`	`}`
	`1249`	`+`
	`1250`	`+ // Enable aggressive optimization for NNPA with -O3`
	`1251`	`+ if (OptimizationLevel == OptLevel::O3 &&`
	`1252`	`+ getTargetAccel().find("NNPA") != std::string::npos &&`
	`1253`	`+ getLLVMOption().find("enable-unsafe-fp-math") == std::string::npos) {`
	`1254`	`+ setLLVMOption(getLLVMOption() + " --enable-unsafe-fp-math");`
	`1255`	`+ }`
`1249`	`1256`	`}`
`1250`	`1257`
`1251`	`1258`	`} // namespace onnx_mlir`