only emit target description warning in verify phase and correct the target op (#216)

only emit warning in verify phase and correct the pass target op

Author: zhiczhong

include/gc/Analysis/TargetDescriptionAnalysis.h

@@ -55,11 +55,15 @@ class TargetDescriptionAnalysisBase {
   // the map from device type to device string
   static llvm::DenseMap<DeviceType, std::string> DeviceKeyMap;
 
+  // set the emit warning flag
+  void setEmitWarning(bool emit) { emitWarning = emit; }
+
 private:
   MLIRContext *ctx;
   DeviceType device;
   DataLayout layout;
   Location loc;
+  bool emitWarning = false;
 };
 
 class CPUTargetDescriptionAnalysis : public TargetDescriptionAnalysisBase {

lib/gc/Analysis/TargetDescriptionAnalysis.cpp

@@ -25,7 +25,9 @@ template <typename T>
 void TargetDescriptionAnalysisBase::emitNotFoundWarning(Location loc,
                                                         StringRef key,
                                                         T value) {
-  mlir::emitWarning(loc) << key << " not found, using default value " << value;
+  if (emitWarning)
+    mlir::emitWarning(loc) << key << " not found, using default value "
+                           << value;
 }
 
 static bool isIntegerNumber(const std::string &token) {

lib/gc/Transforms/Pipeline.cpp

@@ -134,7 +134,7 @@ void populateLLVMPasses(mlir::OpPassManager &pm) {
 
 void populateCPUPipeline(mlir::OpPassManager &pm) {
   // verify the target description attribute
-  pm.addNestedPass<func::FuncOp>(createVerifyTargetDescription());
+  pm.addPass(createVerifyTargetDescription());
   // front-end, oneDNN graph dialect
   populateFrontendPasses(pm);
   // middle-end, LinalgX/Linalg/tensor dialects

lib/gc/Transforms/VerifyTargetDescription.cpp

@@ -24,6 +24,7 @@ namespace {
 static LogicalResult verifyCPUTargetDescription(RewriterBase &rewriter,
                                                 Operation *op) {
   CPUTargetDescriptionAnalysis cpuTargetDesc(op);
+  cpuTargetDesc.setEmitWarning(true);
   Location loc = op->getLoc();
 
   // Check if the num_threads is existed and greater than 0