Cold-Air-Networks/mirror-llvm-project
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 7

[MLIR][Python] Do not install ExecutionEngine wrapper when disabled (#211250)

Browse source 
The `_mlirExecutionEngine` extension is declared only when
`MLIR_ENABLE_EXECUTION_ENGINE` is enabled. However,
`execution_engine.py` and its type stub are currently added to the
Python package unconditionally.

This leaves an installed `mlir.execution_engine` module that fails
immediately when imported because its native extension is absent. It
also exposes a type stub for an unavailable API.

Making sure they're added only if `MLIR_ENABLE_EXECUTION_ENGINE` is
enabled.
This commit is contained in:
Anutosh Bhat 2026-07-23 12:45:12 +05:30 committed by GitHub
parent 520a930da2
commit c151d33ace
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
mlir/python/CMakeLists.txt
View file

@ -41,15 +41,26 @@ declare_mlir_python_sources(MLIRPythonSources.Core.Python.Extras
extras/meta.py
)
# Keep this source group unconditional because runtime/*.py contains generic
# memref helpers that do not depend on the ExecutionEngine extension. Only the
# wrapper and type stub require _mlirExecutionEngine.
set(_execution_engine_python_sources)
if(MLIR_ENABLE_EXECUTION_ENGINE)
set(_execution_engine_python_sources
execution_engine.py
_mlir_libs/_mlirExecutionEngine.pyi
)
endif()
declare_mlir_python_sources(MLIRPythonSources.ExecutionEngine
ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"
ADD_TO_PARENT MLIRPythonSources
SOURCES
execution_engine.py
_mlir_libs/_mlirExecutionEngine.pyi
${_execution_engine_python_sources}
SOURCES_GLOB
runtime/*.py
)
unset(_execution_engine_python_sources)
declare_mlir_python_sources(MLIRPythonCAPI.HeaderSources
ROOT_DIR "${MLIR_SOURCE_DIR}/include"