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

     1  //===- Configuration.cpp - Configuration object base classes --------------===//
     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/Quantizer/Support/Configuration.h"
    19  
    20  #include <limits>
    21  
    22  #include "mlir/IR/Builders.h"
    23  #include "mlir/IR/Identifier.h"
    24  #include "mlir/IR/MLIRContext.h"
    25  
    26  using namespace mlir;
    27  using namespace mlir::quantizer;
    28  
    29  TargetConfiguration::TargetConfiguration(SolverContext &context) {}
    30  
    31  void TargetConfiguration::addOpHandlerByName(StringRef name, OpHandlerFn fn) {
    32    opHandlers[name] = fn;
    33  }
    34  
    35  void TargetConfiguration::addRequireStatsOpByName(StringRef opName) {
    36    requireStatsOpNames.insert(opName);
    37  }
    38  
    39  bool TargetConfiguration::isRequireStatsOp(Operation *op) const {
    40    return requireStatsOpNames.find(op->getName().getStringRef()) !=
    41           requireStatsOpNames.end();
    42  }
    43  
    44  void TargetConfiguration::handleOp(Operation *op, CAGSlice &cag) const {
    45    auto found_it = opHandlers.find(op->getName().getStringRef());
    46    if (found_it != opHandlers.end())
    47      found_it->second(op, cag);
    48  }