| // Copyright 2021 The IREE Authors |
| // |
| // Licensed under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| #include "iree/compiler/Dialect/Util/IR/UtilOps.h" |
| #include "iree/compiler/GlobalOptimization/Passes.h" |
| |
| namespace mlir::iree_compiler::GlobalOptimization { |
| |
| #define GEN_PASS_DEF_CLEANUPNUMERICNARROWINGPASS |
| #include "iree/compiler/GlobalOptimization/Passes.h.inc" |
| |
| namespace { |
| |
| class CleanupNumericNarrowingPass |
| : public impl::CleanupNumericNarrowingPassBase< |
| CleanupNumericNarrowingPass> { |
| void runOnOperation() override { |
| getOperation()->walk([](IREE::Util::NumericOptionalNarrowOp op) { |
| op.getResult().replaceAllUsesWith(op.getOperand()); |
| op->erase(); |
| }); |
| } |
| }; |
| |
| } // namespace |
| } // namespace mlir::iree_compiler::GlobalOptimization |
