mirror of https://github.com/llvm/torch-mlir
[Stablehlo] enable stablehlo's python extension binding (#3529)
parent
5bee9aac63
commit
4bb7ddf601
|
@ -118,10 +118,6 @@ else()
|
|||
set(MLIR_INCLUDE_DIRS "${MLIR_INCLUDE_DIR};${MLIR_GENERATED_INCLUDE_DIR}")
|
||||
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_BINARY_DIR "${CMAKE_CURRENT_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.
|
||||
if (TORCH_MLIR_ENABLE_STABLEHLO)
|
||||
set(STABLEHLO_BUILD_EMBEDDED ON)
|
||||
set(STABLEHLO_ENABLE_BINDINGS_PYTHON ON)
|
||||
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/externals/stablehlo
|
||||
${CMAKE_CURRENT_BINARY_DIR}/stablehlo
|
||||
EXCLUDE_FROM_ALL)
|
||||
|
|
|
@ -102,6 +102,10 @@ set(_source_components
|
|||
TorchMLIRPythonTorchExtensionsSources
|
||||
)
|
||||
|
||||
if(TORCH_MLIR_ENABLE_STABLEHLO)
|
||||
list(APPEND _source_components StablehloPythonExtensions)
|
||||
endif()
|
||||
|
||||
add_mlir_python_common_capi_library(TorchMLIRAggregateCAPI
|
||||
INSTALL_COMPONENT TorchMLIRPythonModules
|
||||
INSTALL_DESTINATION python_packages/torch_mlir/torch_mlir/_mlir_libs
|
||||
|
@ -116,4 +120,4 @@ add_mlir_python_modules(TorchMLIRPythonModules
|
|||
DECLARED_SOURCES ${_source_components}
|
||||
COMMON_CAPI_LINK_LIBS
|
||||
TorchMLIRAggregateCAPI
|
||||
)
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue