maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[NFC][MLGO] The regalloc reward is float, not int64_t

This commit is contained in:
Mircea Trofin 2022-01-12 09:32:06 -08:00
parent f51b25a4b9
commit b2d2e93138
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp
View File

@ -162,7 +162,7 @@ static const std::vector<TensorSpec> InputFeatures{
#undef _DECL_FEATURES
static const TensorSpec Output =
TensorSpec::createSpec<int64_t>(DecisionName, {1});
static const TensorSpec Reward = TensorSpec::createSpec<int64_t>("reward", {1});
static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
#endif //#ifdef LLVM_HAVE_TF_API
} // namespace