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

[Mircea Trofin via llvm-commits]

Author: Mircea Trofin
Date: 2022-01-12T09:32:41-08:00
New Revision: b2d2e93138eb7934053368fb8fd1e7e5b93a281c

URL: https://github.com/llvm/llvm-project/commit/b2d2e93138eb7934053368fb8fd1e7e5b93a281c
DIFF: https://github.com/llvm/llvm-project/commit/b2d2e93138eb7934053368fb8fd1e7e5b93a281c.diff

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

Added: 
    

Modified: 
    llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp b/llvm/lib/CodeGen/MLRegallocEvictAdvisor.cpp
index 3310d0c2acc54..7a6c4fade06e4 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