Check prefix

Author: MacDue

llvm/lib/Target/AArch64/Utils/AArch64SMEAttributes.cpp

@@ -14,12 +14,7 @@
 
 using namespace llvm;
 
-void SMEAttrs::set(unsigned M, bool Enable) {
-  if (Enable)
-    Bitmask |= M;
-  else
-    Bitmask &= ~M;
-
+void SMEAttrs::validate() const {
   // Streaming Mode Attrs
   assert(!(hasStreamingInterface() && hasStreamingCompatibleInterface()) &&
          "SM_Enabled and SM_Compatible are mutually exclusive");
@@ -81,6 +76,10 @@ SMEAttrs::SMEAttrs(const AttributeList &Attrs) {
 
 void SMEAttrs::addKnownFunctionAttrs(StringRef FuncName,
                                      const TargetLowering &TLI) {
+  // If the function name does not start with a _ or #_ is not a builtin.
+  if (!FuncName.starts_with('_') && !FuncName.starts_with("#_"))
+    return;
+
   struct SMERoutineAttr {
     RTLIB::Libcall LC{RTLIB::UNKNOWN_LIBCALL};
     unsigned Attrs{SMEAttrs::Normal};

llvm/lib/Target/AArch64/Utils/AArch64SMEAttributes.h

@@ -62,7 +62,15 @@ class SMEAttrs {
     addKnownFunctionAttrs(FuncName, TLI);
   };
 
-  void set(unsigned M, bool Enable = true);
+  void set(unsigned M, bool Enable = true) {
+    if (Enable)
+      Bitmask |= M;
+    else
+      Bitmask &= ~M;
+#ifndef NDEBUG
+    validate();
+#endif
+  }
 
   // Interfaces to query PSTATE.SM
   bool hasStreamingBody() const { return Bitmask & SM_Body; }
@@ -149,6 +157,7 @@ class SMEAttrs {
 
 private:
   void addKnownFunctionAttrs(StringRef FuncName, const TargetLowering &TLI);
+  void validate() const;
 };
 
 /// SMECallAttrs is a utility class to hold the SMEAttrs for a callsite. It has