diff options
author | Mircea Trofin <mtrofin@google.com> | 2022-01-12 09:32:06 -0800 |
---|---|---|
committer | Mircea Trofin <mtrofin@google.com> | 2022-01-12 09:32:41 -0800 |
commit | b2d2e93138eb7934053368fb8fd1e7e5b93a281c (patch) | |
tree | 57210c059fc9e94f4b153beb4fef35b3fb6597b7 | |
parent | f51b25a4b97a59b0ff7b485d4f2ce1e293ffb739 (diff) |
[NFC][MLGO] The regalloc reward is float, not int64_t
-rw-r--r-- | llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp b/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp index 3310d0c2acc5..7a6c4fade06e 100644 --- a/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp +++ b/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp @@ -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 |