llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Stablehlo] enable stablehlo's python extension binding (#3529)

pull/3530/head
Yuanqiang Liu 2024-07-10 13:00:13 +08:00 committed by GitHub
parent 5bee9aac63
commit 4bb7ddf601
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
CMakeLists.txt
View File

@ -118,10 +118,6 @@ else()
set(MLIR_INCLUDE_DIRS "${MLIR_INCLUDE_DIR};${MLIR_GENERATED_INCLUDE_DIR}") set(MLIR_INCLUDE_DIRS "${MLIR_INCLUDE_DIR};${MLIR_GENERATED_INCLUDE_DIR}")
endif() endif()
if (TORCH_MLIR_ENABLE_STABLEHLO)
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/externals/stablehlo)
endif()
set(TORCH_MLIR_SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}") set(TORCH_MLIR_SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}")
set(TORCH_MLIR_BINARY_DIR "${CMAKE_CURRENT_BINARY_DIR}") set(TORCH_MLIR_BINARY_DIR "${CMAKE_CURRENT_BINARY_DIR}")
message(STATUS "Building torch-mlir project at ${TORCH_MLIR_SOURCE_DIR} (into ${TORCH_MLIR_BINARY_DIR})") message(STATUS "Building torch-mlir project at ${TORCH_MLIR_SOURCE_DIR} (into ${TORCH_MLIR_BINARY_DIR})")
@ -225,6 +221,7 @@ endif()
# do not even compile on all platforms. # do not even compile on all platforms.
if (TORCH_MLIR_ENABLE_STABLEHLO) if (TORCH_MLIR_ENABLE_STABLEHLO)
set(STABLEHLO_BUILD_EMBEDDED ON) set(STABLEHLO_BUILD_EMBEDDED ON)
set(STABLEHLO_ENABLE_BINDINGS_PYTHON ON)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/externals/stablehlo add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/externals/stablehlo
${CMAKE_CURRENT_BINARY_DIR}/stablehlo ${CMAKE_CURRENT_BINARY_DIR}/stablehlo
EXCLUDE_FROM_ALL) EXCLUDE_FROM_ALL)

6
python/CMakeLists.txt
View File

@ -102,6 +102,10 @@ set(_source_components
TorchMLIRPythonTorchExtensionsSources TorchMLIRPythonTorchExtensionsSources
) )
if(TORCH_MLIR_ENABLE_STABLEHLO)
list(APPEND _source_components StablehloPythonExtensions)
endif()
add_mlir_python_common_capi_library(TorchMLIRAggregateCAPI add_mlir_python_common_capi_library(TorchMLIRAggregateCAPI
INSTALL_COMPONENT TorchMLIRPythonModules INSTALL_COMPONENT TorchMLIRPythonModules
INSTALL_DESTINATION python_packages/torch_mlir/torch_mlir/_mlir_libs INSTALL_DESTINATION python_packages/torch_mlir/torch_mlir/_mlir_libs
@ -116,4 +120,4 @@ add_mlir_python_modules(TorchMLIRPythonModules
DECLARED_SOURCES ${_source_components} DECLARED_SOURCES ${_source_components}
COMMON_CAPI_LINK_LIBS COMMON_CAPI_LINK_LIBS
TorchMLIRAggregateCAPI TorchMLIRAggregateCAPI
) )