1 //===- InlineModelFeatureMaps.h - common model runner defs ------*- C++ -*-===//
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
7 //===----------------------------------------------------------------------===//
10 #ifndef LLVM_ANALYSIS_INLINEMODELFEATUREMAPS_H
11 #define LLVM_ANALYSIS_INLINEMODELFEATUREMAPS_H
19 // List of features. Each feature is defined through a triple:
20 // - the name of an enum member, which will be the feature index
21 // - a textual name, used for Tensorflow model binding (so it needs to match the
22 // names used by the Tensorflow model)
23 // - a documentation description. Currently, that is not used anywhere
24 // programmatically, and serves as workaround to inability of inserting comments
26 #define INLINE_FEATURE_ITERATOR(M) \
27 M(CalleeBasicBlockCount, "callee_basic_block_count", \
28 "number of basic blocks of the callee") \
29 M(CallSiteHeight, "callsite_height", \
30 "position of the call site in the original call graph - measured from " \
32 M(NodeCount, "node_count", \
33 "total current number of defined functions in the module") \
34 M(NrCtantParams, "nr_ctant_params", \
35 "number of parameters in the call site that are constants") \
36 M(CostEstimate, "cost_estimate", "total cost estimate (threshold - free)") \
37 M(EdgeCount, "edge_count", \
38 "number of module-internal users of the caller, +1 if the caller is " \
39 "exposed externally") \
40 M(CallerUsers, "caller_users", \
41 "number of blocks reached from a conditional instruction, in the caller") \
42 M(CallerConditionallyExecutedBlocks, "caller_conditionally_executed_blocks", \
43 "number of blocks reached from a conditional instruction, in the caller") \
44 M(CallerBasicBlockCount, "caller_basic_block_count", \
45 "number of basic blocks in the caller") \
46 M(CalleeConditionallyExecutedBlocks, "callee_conditionally_executed_blocks", \
47 "number of blocks reached from a conditional instruction, in the callee") \
48 M(CalleeUsers, "callee_users", \
49 "number of blocks reached from a conditional instruction, in the callee")
51 enum class FeatureIndex : size_t {
52 #define POPULATE_INDICES(INDEX_NAME, NAME, COMMENT) INDEX_NAME,
53 INLINE_FEATURE_ITERATOR(POPULATE_INDICES)
54 #undef POPULATE_INDICES
58 constexpr size_t NumberOfFeatures =
59 static_cast<size_t>(FeatureIndex::NumberOfFeatures);
61 extern const std::array<std::string, NumberOfFeatures> FeatureNameMap;
63 extern const char *const DecisionName;
64 extern const char *const DefaultDecisionName;
65 extern const char *const RewardName;
67 using InlineFeatures = std::vector<int64_t>;
70 #endif // LLVM_ANALYSIS_INLINEMODELFEATUREMAPS_H