[Mlir-commits] [llvm] [mlir] [MLGO] Do not hardcode 33 eviction feature columns (PR #224598)

[Arseniy Obolenskiy]

================
@@ -622,11 +631,13 @@ float MLEvictAdvisor::getInitialQueueSize(const MachineFunction &MF) {
 
 MLEvictAdvisor::MLEvictAdvisor(const MachineFunction &MF, const RAGreedy &RA,
                                MLModelRunner *Runner,
+                               ArrayRef<TensorSpec> InputFeatures,
                                const MachineBlockFrequencyInfo &MBFI,
                                const MachineLoopInfo &Loops)
-    : RegAllocEvictionAdvisor(MF, RA), DefaultAdvisor(MF, RA),
-      Runner(std::move(Runner)), MBFI(MBFI), Loops(Loops),
-      InitialQSize(MLEvictAdvisor::getInitialQueueSize(MF)) {
+    : RegAllocEvictionAdvisor(MF, RA), InputFeatures(InputFeatures),
+      NumColumns(InputFeatures[FeatureIDs::mask].shape()[1]),
----------------
aobolensk wrote:

simplified this part 

https://github.com/llvm/llvm-project/pull/224598