CodeGen_MLIR: Add initial MLIR CodeGen

halide · May 21, 2023 · 129964e · 129964e
1 parent d234143
commit 129964e
Show file tree

Hide file tree

Showing 4 changed files with 625 additions and 1 deletion.
diff --git a/dependencies/llvm/CMakeLists.txt b/dependencies/llvm/CMakeLists.txt
@@ -10,7 +10,7 @@ set(CMAKE_MAP_IMPORTED_CONFIG_MINSIZEREL MinSizeRel Release RelWithDebInfo "")
 set(CMAKE_MAP_IMPORTED_CONFIG_RELWITHDEBINFO RelWithDebInfo Release MinSizeRel "")
 set(CMAKE_MAP_IMPORTED_CONFIG_RELEASE Release MinSizeRel RelWithDebInfo "")
 
-find_package(LLVM ${Halide_REQUIRE_LLVM_VERSION} REQUIRED)
+find_package(MLIR ${Halide_REQUIRE_LLVM_VERSION} REQUIRED)
 find_package(Clang REQUIRED CONFIG HINTS "${LLVM_DIR}/../clang" "${LLVM_DIR}/../lib/cmake/clang")
 
 set(LLVM_PACKAGE_VERSION "${LLVM_PACKAGE_VERSION}"
@@ -19,6 +19,7 @@ set(LLVM_PACKAGE_VERSION "${LLVM_PACKAGE_VERSION}"
 message(STATUS "Found LLVM ${LLVM_PACKAGE_VERSION}")
 message(STATUS "Using LLVMConfig.cmake in: ${LLVM_DIR}")
 message(STATUS "Using ClangConfig.cmake in: ${Clang_DIR}")
+message(STATUS "Using MLIRConfig.cmake in: ${MLIR_DIR}")
 
 if (LLVM_PACKAGE_VERSION VERSION_LESS 14.0)
     message(FATAL_ERROR "LLVM version must be 14.0 or newer")

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
@@ -33,6 +33,7 @@ set(HEADER_FILES
     CodeGen_Internal.h
     CodeGen_LLVM.h
     CodeGen_Metal_Dev.h
+    CodeGen_MLIR.h
     CodeGen_OpenCL_Dev.h
     CodeGen_OpenGLCompute_Dev.h
     CodeGen_Posix.h
@@ -201,6 +202,7 @@ set(SOURCE_FILES
     CodeGen_Internal.cpp
     CodeGen_LLVM.cpp
     CodeGen_Metal_Dev.cpp
+    CodeGen_MLIR.cpp
     CodeGen_OpenCL_Dev.cpp
     CodeGen_OpenGLCompute_Dev.cpp
     CodeGen_Posix.cpp