diff options
Diffstat (limited to 'llvm/include/llvm/Analysis/InlineModelFeatureMaps.h')
-rw-r--r-- | llvm/include/llvm/Analysis/InlineModelFeatureMaps.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/llvm/include/llvm/Analysis/InlineModelFeatureMaps.h b/llvm/include/llvm/Analysis/InlineModelFeatureMaps.h index 1afa8a825f15..fb8236c28b25 100644 --- a/llvm/include/llvm/Analysis/InlineModelFeatureMaps.h +++ b/llvm/include/llvm/Analysis/InlineModelFeatureMaps.h @@ -10,6 +10,8 @@ #ifndef LLVM_ANALYSIS_INLINEMODELFEATUREMAPS_H #define LLVM_ANALYSIS_INLINEMODELFEATUREMAPS_H +#include "llvm/Analysis/TensorSpec.h" + #include <array> #include <string> #include <vector> @@ -127,7 +129,7 @@ inlineCostFeatureToMlFeature(InlineCostFeatureIndex Feature) { constexpr size_t NumberOfFeatures = static_cast<size_t>(FeatureIndex::NumberOfFeatures); -extern const std::array<std::string, NumberOfFeatures> FeatureNameMap; +extern const std::array<TensorSpec, NumberOfFeatures> FeatureMap; extern const char *const DecisionName; extern const char *const DefaultDecisionName; |