Typed binary operators (#23)

* Implement typed expression stack * Typed negate * Start work on FileCheck based code generation tests
wpmed92 · Aug 22, 2024 · 9f7e3cf · 9f7e3cf
1 parent 5a0cb1b
commit 9f7e3cf
Show file tree

Hide file tree

Showing 5 changed files with 181 additions and 113 deletions.
diff --git a/include/CodeGen/MLIRCodeGen.h b/include/CodeGen/MLIRCodeGen.h
@@ -14,6 +14,7 @@
 #include "llvm/ADT/ScopedHashTable.h"
 #include <vector>
 #include <map>
+#include <utility>
 
 using namespace mlir;
 
@@ -35,7 +36,7 @@ class MLIRCodeGen : public ASTVisitor {
 public:
   MLIRCodeGen();
   void initModuleOp();
-  void dump();
+  void print();
   bool verify();
   void visit(TranslationUnit *) override;
   void visit(BinaryExpression *) override;
@@ -80,7 +81,7 @@ class MLIRCodeGen : public ASTVisitor {
   bool inGlobalScope = true;
   llvm::StringMap<spirv::FuncOp> functionMap;
   llvm::StringMap<StructDeclaration*> structDeclarations;
-  std::vector<Value> expressionStack;
+  std::vector<std::pair<Type*, Value>> expressionStack;
   StructDeclaration* currentBaseComposite = nullptr;
 
   llvm::ScopedHashTable<llvm::StringRef, SymbolTableEntry>
@@ -95,7 +96,7 @@ class MLIRCodeGen : public ASTVisitor {
   void createVariable(shaderpulse::Type *, VariableDeclaration *);
   void insertEntryPoint();
 
-  mlir::Value popExpressionStack();
+  std::pair<Type*, Value> popExpressionStack();
   mlir::Value currentBasePointer;
   Type* typeContext;
 };