Compile to spirv

.gitignore

@@ -2,4 +2,5 @@
 build
 .vscode
 test/CodeGen/*.mlir
-example/*.mlir
+example/*.mlir
+*.spv

CMakeLists.txt

@@ -84,6 +84,7 @@ if(ENABLE_CODEGEN)
       MLIRLLVMToLLVMIRTranslation
       MLIRMemRefDialect
       MLIRSPIRVDialect
+      MLIRSPIRVSerialization
       MLIRParser
       MLIRPass
       MLIRSideEffectInterfaces

compiler/shaderpulse.cpp

@@ -84,13 +84,21 @@ int main(int argc, char** argv) {
         mlirCodeGen.print();
 
         if (!mlirCodeGen.verify()) {
-            std::cout << "Error verifying the SPIR-V module" << std::endl;
-         return -1;
+            std::cout << "Error verifying the SPIR-V module." << std::endl;
+            return -1;
+        }
+
+        bool success = mlirCodeGen.saveToFile(outputPath);
+
+        if (!success) {
+            std::cout << "Failed to save spirv mlir to file.";
+            return -1;
         }
 
-        bool succes = mlirCodeGen.saveToFile(outputPath);
+        success = mlirCodeGen.emitSpirv(outputPath.replace_extension(".spv"));
 
-        if (!succes) {
+        if (!success) {
+            std::cout << "Failed to emit spirv binary.";
             return -1;
         }
     }

example/square.glsl

@@ -0,0 +1,14 @@
+layout(local_size_x = 1, local_size_y = 1, local_size_z = 1) in;
+
+layout(binding = 0) buffer InBuffer {
+    int data[];
+};
+
+layout(binding = 1) buffer OutBuffer {
+    int result[];
+};
+
+void main() {
+    uint idx = gl_GlobalInvocationID.x;
+    result[idx] = data[idx] * data[idx];
+}

include/CodeGen/MLIRCodeGen.h

@@ -42,6 +42,7 @@ class MLIRCodeGen : public ASTVisitor {
   void initModuleOp();
   void print();
   bool saveToFile(const std::filesystem::path& outputPath);
+  bool emitSpirv(const std::filesystem::path& outputPath);
   bool verify();
   void visit(TranslationUnit *) override;
   void visit(BinaryExpression *) override;

lib/CodeGen/MLIRCodeGen.cpp

@@ -5,6 +5,7 @@
 #include <iostream>
 #include <cassert>
 #include <fstream>
+#include "mlir/Target/SPIRV/Serialization.h"
 #include <llvm/Support/raw_ostream.h>
 #include <llvm/Support/MemoryBuffer.h>
 #include <utility>
@@ -213,6 +214,19 @@ bool MLIRCodeGen::saveToFile(const std::filesystem::path& outputPath) {
   return true;
 }
 
+bool MLIRCodeGen::emitSpirv(const std::filesystem::path& outputPath) {
+  llvm::SmallVector<uint32_t, 128> spirvBinary;
+  mlir::LogicalResult result = mlir::spirv::serialize(spirvModule, spirvBinary);
+  if (failed(result)) {
+      std::cerr << "Failed to serialize SPIR-V module." << std::endl;
+      return false;
+  }
+
+  std::ofstream outFile(outputPath, std::ios::binary);
+  outFile.write(reinterpret_cast<const char*>(spirvBinary.data()), spirvBinary.size() * sizeof(uint32_t));
+  return true;
+}
+
 bool MLIRCodeGen::verify() { return !failed(mlir::verify(spirvModule)); }
 
 void MLIRCodeGen::insertEntryPoint() {