[compiler] fix compilation on AArch64 platform (#358)

- update Readme with AArch64 platform settings - fix potential compilation error on AArch64 platform
bytedance · Jun 20, 2024 · 01098f8 · 01098f8
1 parent c576d6f
commit 01098f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/compiler/README.md b/compiler/README.md
@@ -31,6 +31,7 @@ cmake -H./llvm \
       -DCMAKE_INSTALL_PREFIX=$(pwd)/build/install
 # via -DCMAKE_C_COMPILER=gcc/clang and -DCMAKE_CXX_COMPILER=g++/clang++
 # to specify gcc>=8.5 or clang>=7 
+# for Mac users, set -DLLVM_TARGETS_TO_BUILD="AArch64;NVPTX"
 
 cmake --build ./build --target all --target install
 ```

diff --git a/compiler/lib/Dialect/mhlo/CMakeLists.txt b/compiler/lib/Dialect/mhlo/CMakeLists.txt
@@ -122,6 +122,7 @@ add_mlir_dialect_library(ByteIRMhloPasses
   ByteIRMhloAnalysis
   ByteIRMhloDynamicShapeOpRegister
   MLIRAceDialect
+  MLIRCatDialect
 
   LINK_LIBS PUBLIC
   MLIRIR