OpCountAnalysis.h

Go to the documentation of this file.

//===- OpCountAnalysis.h - operation count analyses -----------------------===//
//
// Part of the LLVM Project, 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
//
//===----------------------------------------------------------------------===//
//
// This header file defines prototypes for methods that perform analysis
// involving the frequency of different kinds of operations found in a
// builtin.module.
//
//===----------------------------------------------------------------------===//
 
#ifndef CIRCT_ANALYSIS_OPCOUNT_ANALYSIS_H
#define CIRCT_ANALYSIS_OPCOUNT_ANALYSIS_H
 
#include "circt/Support/LLVM.h"
#include "mlir/IR/Operation.h"
#include "llvm/ADT/DenseMap.h"
 
namespace mlir {
class AnalysisManager;
} // namespace mlir
namespace circt {
namespace analysis {
 
class OpCountAnalysis {
public:
  OpCountAnalysis(Operation *moduleOp, mlir::AnalysisManager &am);
 
  /// Get the frequency of operations of a specific name
  size_t getOpCount(OperationName opName);
 
  /// Get the names of all distinct operations found by the analysis
  SmallVector<OperationName> getFoundOpNames();
 
  /// Get a map from number of operands to corresponding frequency for the given
  /// operation
  DenseMap<size_t, size_t> getOperandCountMap(OperationName opName);
 
private:
  DenseMap<OperationName, size_t> opCounts;
  DenseMap<OperationName, DenseMap<size_t, size_t>> operandCounts;
};
 
} // namespace analysis
} // namespace circt
 
#endif // CIRCT_ANALYSIS_OPCOUNT_ANALYSIS_H