github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/mlir/lib/Transforms/StripDebugInfo.cpp (about)

     1  //===- StripDebugInfo.cpp - Pass to strip debug information ---------------===//
     2  //
     3  // Copyright 2019 The MLIR Authors.
     4  //
     5  // Licensed under the Apache License, Version 2.0 (the "License");
     6  // you may not use this file except in compliance with the License.
     7  // You may obtain a copy of the License at
     8  //
     9  //   http://www.apache.org/licenses/LICENSE-2.0
    10  //
    11  // Unless required by applicable law or agreed to in writing, software
    12  // distributed under the License is distributed on an "AS IS" BASIS,
    13  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14  // See the License for the specific language governing permissions and
    15  // limitations under the License.
    16  // =============================================================================
    17  
    18  #include "mlir/IR/Function.h"
    19  #include "mlir/IR/Operation.h"
    20  #include "mlir/Pass/Pass.h"
    21  #include "mlir/Transforms/Passes.h"
    22  
    23  using namespace mlir;
    24  
    25  namespace {
    26  struct StripDebugInfo : public FunctionPass<StripDebugInfo> {
    27    void runOnFunction() override;
    28  };
    29  } // end anonymous namespace
    30  
    31  void StripDebugInfo::runOnFunction() {
    32    FuncOp func = getFunction();
    33    auto unknownLoc = UnknownLoc::get(&getContext());
    34  
    35    // Strip the debug info from the function and its operations.
    36    func.setLoc(unknownLoc);
    37    func.walk([&](Operation *op) { op->setLoc(unknownLoc); });
    38  }
    39  
    40  /// Creates a pass to strip debug information from a function.
    41  std::unique_ptr<FunctionPassBase> mlir::createStripDebugInfoPass() {
    42    return std::make_unique<StripDebugInfo>();
    43  }
    44  
    45  static PassRegistration<StripDebugInfo>
    46      pass("strip-debuginfo", "Strip debug info from functions and operations");