[llvm] [RISCV] Use hasFeature instead of checkFeature in llvm-exegesis. NFC (PR #131401)

[Craig Topper via llvm-commits]

https://github.com/topperc created https://github.com/llvm/llvm-project/pull/131401

Until recently checkFeature was quite slow. #130936

I was curious where we use checkFeature and noticed these. I thought we could use hasFeature instead of going through strings.

>From ac9133d8871566606aeca53902bc798921c47d50 Mon Sep 17 00:00:00 2001
From: Craig Topper <craig.topper at sifive.com>
Date: Fri, 14 Mar 2025 15:10:01 -0700
Subject: [PATCH] [RISCV] Use hasFeature instead of checkFeature in
 llvm-exegesis. NFC

Until recently checkFeature was quite slow. #130936

I was curious where we use checkFeature and noticed these. I thought
we could use hasFeature instead of going through strings.
---
 llvm/tools/llvm-exegesis/lib/RISCV/Target.cpp | 36 ++++++++++++++-----
 1 file changed, 27 insertions(+), 9 deletions(-)

diff --git a/llvm/tools/llvm-exegesis/lib/RISCV/Target.cpp b/llvm/tools/llvm-exegesis/lib/RISCV/Target.cpp
index 6d97a7ecfffb8..b29dd8739aa2e 100644
--- a/llvm/tools/llvm-exegesis/lib/RISCV/Target.cpp
+++ b/llvm/tools/llvm-exegesis/lib/RISCV/Target.cpp
@@ -156,15 +156,33 @@ template <class BaseT> class RISCVSnippetGenerator : public BaseT {
     // FIXME: We could have obtained these two constants from RISCVSubtarget
     // but in order to get that from TargetMachine, we need a Function.
     const MCSubtargetInfo &STI = State.getSubtargetInfo();
-    ELEN = STI.checkFeatures("+zve64x") ? 64 : 32;
-
-    std::string ZvlQuery;
-    for (unsigned Size = 32; Size <= 65536; Size *= 2) {
-      ZvlQuery = "+zvl";
-      raw_string_ostream SS(ZvlQuery);
-      SS << Size << "b";
-      if (STI.checkFeatures(SS.str()) && ZvlVLen < Size)
-        ZvlVLen = Size;
+    ELEN = STI.hasFeature(RISCV::FeatureStdExtZve64x) ? 64 : 32;
+
+    static_assert(RISCV::FeatureStdExtZvl64b == RISCV::FeatureStdExtZvl32b + 1);
+    static_assert(RISCV::FeatureStdExtZvl128b ==
+                  RISCV::FeatureStdExtZvl32b + 2);
+    static_assert(RISCV::FeatureStdExtZvl256b ==
+                  RISCV::FeatureStdExtZvl32b + 3);
+    static_assert(RISCV::FeatureStdExtZvl512b ==
+                  RISCV::FeatureStdExtZvl32b + 4);
+    static_assert(RISCV::FeatureStdExtZvl1024b ==
+                  RISCV::FeatureStdExtZvl32b + 5);
+    static_assert(RISCV::FeatureStdExtZvl2048b ==
+                  RISCV::FeatureStdExtZvl32b + 6);
+    static_assert(RISCV::FeatureStdExtZvl4096b ==
+                  RISCV::FeatureStdExtZvl32b + 7);
+    static_assert(RISCV::FeatureStdExtZvl8192b ==
+                  RISCV::FeatureStdExtZvl32b + 8);
+    static_assert(RISCV::FeatureStdExtZvl16384b ==
+                  RISCV::FeatureStdExtZvl32b + 9);
+    static_assert(RISCV::FeatureStdExtZvl32768b ==
+                  RISCV::FeatureStdExtZvl32b + 10);
+    static_assert(RISCV::FeatureStdExtZvl65536b ==
+                  RISCV::FeatureStdExtZvl32b + 11);
+    for (unsigned Feature = RISCV::FeatureStdExtZvl32b, Size = 32;
+         Size <= 65536; ++Feature, Size *= 2) {
+      if (STI.hasFeature(Feature))
+        ZvlVLen = std::max(ZvlVLen, Size);
     }
   }