Build with MLIR for Emscripten builds in CI

Author: robertoraggi

.github/workflows/ci.yml

@@ -139,7 +139,14 @@ jobs:
           restore-keys: |
             ${{ runner.os }}-emscripten
 
-      - name: Install dependenciesch
+      - name: Download MLIR
+        uses: actions/download-artifact@v4
+        with:
+          name: mlir
+          path: build.em/llvm-project/install/
+        continue-on-error: true
+
+      - name: Install dependencies
         run: |
           npm ci
 

package.json

@@ -41,6 +41,7 @@
     "update-tests": "zx scripts/update-tests.mjs",
     "cxx-gen-ast": "node packages/cxx-gen-ast",
     "cxx-gen-lsp": "node packages/cxx-gen-lsp packages/cxx-gen-lsp/metaModel.json packages/cxx-gen-lsp -o src/lsp/cxx/lsp",
-    "download-lsp-model": "zx scripts/download-lsp-model.mjs"
+    "download-lsp-model": "zx scripts/download-lsp-model.mjs",
+    "download-mlir": "gh run download -n mlir --dir build.em/llvm-project/install/"
   }
 }

packages/cxx-frontend/scripts/build.js

@@ -28,10 +28,17 @@ $.verbose = true;
 // get the root directory of the project
 const projectRootSourcePath = path.join(__dirname, "../../..");
 
+const hasMLIR = fs.existsSync(
+  path.join(
+    projectRootSourcePath,
+    "build.em/llvm-project/install/bin/llvm-tblgen.js"
+  )
+);
+
 // get the source path of the cxx-frontend package
 const cxxFrontendSourcePath = path.join(
   projectRootSourcePath,
-  "packages/cxx-frontend",
+  "packages/cxx-frontend"
 );
 
 // set the current working directory to the cxx-frontend source path
@@ -104,7 +111,7 @@ async function dockerBuild() {
   } else {
     cmakeOptions.push(
       "-DCMAKE_BUILD_TYPE=MinSizeRel",
-      "-DCXX_INTERPROCEDURAL_OPTIMIZATION=ON",
+      "-DCXX_INTERPROCEDURAL_OPTIMIZATION=ON"
     );
   }
 
@@ -129,12 +136,20 @@ async function emsdkBuild({ cmake, emcmake, flatc, kwgen }) {
     `-B ${projectRootSourcePath}/build.em`,
   ];
 
+  if (hasMLIR) {
+    cmakeOptions.push(
+      `-DCXX_ENABLE_MLIR=YES`,
+      `-DLLVM_DIR=${projectRootSourcePath}/build.em/llvm-project/install/lib/cmake/llvm`,
+      `-DMLIR_DIR=${projectRootSourcePath}/build.em/llvm-project/install/lib/cmake/mlir`
+    );
+  }
+
   if (argv.debug) {
     cmakeOptions.push(`-DCMAKE_BUILD_TYPE=Debug`);
   } else {
     cmakeOptions.push(
       `-DCMAKE_BUILD_TYPE=MinSizeRel`,
-      `-DCXX_INTERPROCEDURAL_OPTIMIZATION=${CMAKE_INTERPROCEDURAL_OPTIMIZATION}`,
+      `-DCXX_INTERPROCEDURAL_OPTIMIZATION=${CMAKE_INTERPROCEDURAL_OPTIMIZATION}`
     );
   }
 
@@ -146,7 +161,9 @@ async function emsdkBuild({ cmake, emcmake, flatc, kwgen }) {
 async function emsdkBuildPresets() {
   const cmake = await which("cmake", { nothrow: true });
 
-  await $`${cmake} -S ${projectRootSourcePath} --preset emscripten`;
+  const preset = hasMLIR ? "emscripten-mlir" : "emscripten";
+
+  await $`${cmake} -S ${projectRootSourcePath} --preset ${preset}`;
 
   await $`${cmake} --build ${projectRootSourcePath}/build.em --target install`;
 }