MLIR based backend for Numba

.gitignore

@@ -12,6 +12,7 @@
 .nfs*
 tags
 MANIFEST
+CMakeLists.txt.user
 
 build/
 docs/_build/

mlir-compiler/CMakeLists.txt

@@ -0,0 +1,25 @@
+cmake_minimum_required(VERSION 3.5)
+
+project(mlir_compiler LANGUAGES CXX C)
+
+set(CMAKE_CXX_STANDARD 14)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+set(CMAKE_POSITION_INDEPENDENT_CODE ON)
+
+add_subdirectory(mlir-compiler)
+
+if(DEFINED DPCOMP_TREE)
+    message(STATUS "Out of tree PLIER is used")
+    add_subdirectory(${DPCOMP_TREE} plier)
+elseif(DEFINED DPCOMP_DIR)
+    message(STATUS "PLIER from DPCOMP_DIR is used")
+    target_include_directories(${PROJECT_NAME} PRIVATE
+        ${DPCOMP_DIR}/include
+        )
+    target_link_directories(${PROJECT_NAME} PRIVATE
+        ${DPCOMP_DIR}
+        ${DPCOMP_DIR}/Release
+        )
+else()
+    message(FATAL_ERROR "dpcomp not found")
+endif()

mlir-compiler/llvm-sha.txt

@@ -0,0 +1 @@
+7b153b43d3a14d76975039408c4b922beb576735

mlir-compiler/mlir-compiler/CMakeLists.txt

@@ -0,0 +1,76 @@
+
+find_package(pybind11 REQUIRED)
+
+find_package(LLVM REQUIRED CONFIG)
+find_package(MLIR REQUIRED CONFIG)
+
+list(APPEND CMAKE_MODULE_PATH "${MLIR_CMAKE_DIR}")
+list(APPEND CMAKE_MODULE_PATH "${LLVM_CMAKE_DIR}")
+include(TableGen)
+include(AddLLVM)
+include(AddMLIR)
+include(HandleLLVMOptions)
+
+set(SOURCES_LIST
+    src/pipelines/base_pipeline.cpp
+    src/pipelines/lower_to_llvm.cpp
+    src/pipelines/parallel_to_tbb.cpp
+    src/pipelines/plier_to_linalg.cpp
+    src/pipelines/plier_to_std.cpp
+    src/lowering.cpp
+    src/mangle.cpp
+    src/py_func_resolver.cpp
+    src/py_linalg_resolver.cpp
+    src/py_map_types.cpp
+    src/py_module.cpp
+    )
+set(HEADERS_LIST
+    src/pipelines/base_pipeline.hpp
+    src/pipelines/lower_to_llvm.hpp
+    src/pipelines/parallel_to_tbb.hpp
+    src/pipelines/plier_to_linalg.hpp
+    src/pipelines/plier_to_std.hpp
+    src/lowering.hpp
+    src/mangle.hpp
+    src/py_func_resolver.hpp
+    src/py_linalg_resolver.hpp
+    src/py_map_types.hpp
+    src/py_module.hpp
+    )
+
+pybind11_add_module(${PROJECT_NAME} ${SOURCES_LIST} ${HEADERS_LIST})
+
+if (MSVC)
+    target_compile_options(${PROJECT_NAME} PRIVATE /EHsc)
+endif ()
+
+if (CMAKE_SYSTEM_NAME STREQUAL Linux)
+    target_link_options(${PROJECT_NAME} PRIVATE "LINKER:--version-script=export.txt")
+endif()
+
+if (CMAKE_SYSTEM_NAME STREQUAL Darwin)
+    target_link_libraries(${PROJECT_NAME} PRIVATE "-Wl,-exported_symbols_list,export_darwin.txt")
+endif()
+
+target_compile_definitions(${PROJECT_NAME} PRIVATE ${LLVM_DEFINITIONS})
+
+target_link_libraries(${PROJECT_NAME} PRIVATE
+    plier
+    LLVM${LLVM_NATIVE_ARCH}CodeGen
+    LLVM${LLVM_NATIVE_ARCH}Desc
+    LLVMTarget
+    MLIRIR
+    MLIRLLVMIR
+    MLIRLLVMToLLVMIRTranslation
+    MLIRTransforms
+    MLIRStandardOpsTransforms
+    MLIRLinalgTransforms
+    MLIRSCFToStandard
+    MLIRTensorTransforms
+    )
+
+target_include_directories(${PROJECT_NAME} PRIVATE
+    ./src
+    ${LLVM_INCLUDE_DIRS}
+    ${MLIR_INCLUDE_DIRS}
+    )

mlir-compiler/mlir-compiler/export.txt

@@ -0,0 +1,4 @@
+{
+	global: PyInit_mlir_compiler;
+	local: *;
+};

mlir-compiler/mlir-compiler/export_darwin.txt

@@ -0,0 +1 @@
+_PyInit_mlir_compiler