parse into datastructure

GPUOpen-Drivers · Oct 21, 2024 · 55e8cdd · 55e8cdd
1 parent 4e7f24c
commit 55e8cdd
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/include/llvm-dialects/TableGen/Operations.h b/include/llvm-dialects/TableGen/Operations.h
@@ -102,6 +102,7 @@ class Operation : public OperationBase {
   std::string name;
   std::string mnemonic;
   std::vector<Trait *> traits;
+  std::map<llvm::StringRef, std::vector<Trait *>> value_traits;
 
   std::vector<NamedValue> results;
 

diff --git a/lib/TableGen/Operations.cpp b/lib/TableGen/Operations.cpp
@@ -346,6 +346,29 @@ bool Operation::parse(raw_ostream &errs, GenDialectsContext *context,
   for (RecordTy *traitRec : record->getValueAsListOfDefs("traits"))
     op->traits.push_back(context->getTrait(traitRec));
 
+  ListInit *List = record->getValueAsListInit("value_traits");
+  for (const Init *I : List->getValues()) {
+    if (const DagInit *DI = dyn_cast<DagInit>(I)) {
+      if (DI->getNumArgs() != 1) {
+        errs << "value_traits " << *DI << " is missing argument name";
+        return false;
+      }
+      StringRef name = DI->getArgNameStr(0);
+      std::vector<Trait*>& traits = op->value_traits[name];
+
+      if (const DefInit *Op = dyn_cast<DefInit>(DI->getOperator()))
+        traits.push_back(context->getTrait(Op->getDef()));
+      else {
+        errs << "value_traits " << *DI << " is not of form (Trait $arg)";
+        return false;
+      }
+    }
+    else {
+      errs << "value_traits was not a list of DAG's";
+      return false;
+    }
+  }
+
   EvaluationPlanner evaluation(op->m_system);
 
   for (const auto &arg : op->getFullArguments()) {