xref: /llvm-project/mlir/lib/Transforms/StripDebugInfo.cpp (revision 67d0d7ac0acb0665d6a09f61278fbcf51f0114c2) 
1 //===- StripDebugInfo.cpp - Pass to strip debug information ---------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "mlir/Transforms/Passes.h"
10 
11 #include "mlir/IR/BuiltinOps.h"
12 #include "mlir/IR/Operation.h"
13 #include "mlir/Pass/Pass.h"
14 
15 namespace mlir {
16 #define GEN_PASS_DEF_STRIPDEBUGINFO
17 #include "mlir/Transforms/Passes.h.inc"
18 } // namespace mlir
19 
20 using namespace mlir;
21 
22 namespace {
23 struct StripDebugInfo : public impl::StripDebugInfoBase<StripDebugInfo> {
24   void runOnOperation() override;
25 };
26 } // namespace
27 
runOnOperation()28 void StripDebugInfo::runOnOperation() {
29   auto unknownLoc = UnknownLoc::get(&getContext());
30 
31   // Strip the debug info from all operations.
32   getOperation()->walk([&](Operation *op) {
33     op->setLoc(unknownLoc);
34     // Strip block arguments debug info.
35     for (Region &region : op->getRegions()) {
36       for (Block &block : region.getBlocks()) {
37         for (BlockArgument &arg : block.getArguments()) {
38           arg.setLoc(unknownLoc);
39         }
40       }
41     }
42   });
43 }
44 
45 /// Creates a pass to strip debug information from a function.
createStripDebugInfoPass()46 std::unique_ptr<Pass> mlir::createStripDebugInfoPass() {
47   return std::make_unique<StripDebugInfo>();
48 }
49