all

vla5924-practice · Oct 18, 2024 · 07ae23e · 07ae23e
1 parent 376ddbd
commit 07ae23e
Show file tree

Hide file tree

Showing 4 changed files with 292 additions and 26 deletions.
diff --git a/compiler/include/compiler/backend/optree/semantizer/traits.hpp b/compiler/include/compiler/backend/optree/semantizer/traits.hpp
@@ -1,5 +1,7 @@
 #pragma once
 
+#include <algorithm>
+
 #include "compiler/optree/operation.hpp"
 #include "compiler/optree/types.hpp"
 
@@ -53,6 +55,17 @@ struct HasOperands {
     }
 };
 
+struct HasOperandsOfType {
+    static bool verify(const Operation::Ptr &op, SemantizerContext &ctx, size_t numOperands, const Type::Ptr &type) {
+        if (op->numOperands() == numOperands &&
+            std::all_of(op->operands.begin(), op->operands.end(),
+                        [&](const Value::Ptr &operand) { return operand->hasType(type); }))
+            return true;
+        ctx.pushOpError(op) << "must have " << numOperands << " operands of " << type;
+        return false;
+    }
+};
+
 struct HasResults {
     static bool verify(const Operation::Ptr &op, SemantizerContext &ctx, size_t numResults) {
         if (op->numResults() == numResults)
@@ -80,6 +93,17 @@ struct HasInwards {
     }
 };
 
+struct HasInwardsOfType {
+    static bool verify(const Operation::Ptr &op, SemantizerContext &ctx, size_t numInwards, const Type::Ptr &type) {
+        if (op->numInwards() == numInwards &&
+            std::all_of(op->inwards.begin(), op->inwards.end(),
+                        [&](const Value::Ptr &inward) { return inward->hasType(type); }))
+            return true;
+        ctx.pushOpError(op) << "must have " << numInwards << " inwards of " << type;
+        return false;
+    }
+};
+
 struct HasAttributes {
     static bool verify(const Operation::Ptr &op, SemantizerContext &ctx, size_t numAttrs) {
         if (op->numAttrs() == numAttrs)

diff --git a/compiler/include/compiler/optree/adaptors.hpp b/compiler/include/compiler/optree/adaptors.hpp
@@ -46,6 +46,9 @@ struct FunctionOp : Adaptor {
 
     OPTREE_ADAPTOR_ATTRIBUTE(name, setName, std::string, 0)
     OPTREE_ADAPTOR_ATTRIBUTE_TYPE(type, FunctionType, 1)
+
+    size_t numArguments() const;
+    size_t numResults() const;
 };
 
 struct FunctionCallOp : Adaptor {