Update file names and extensions for MLIR Python execution engine changes.

This commit is contained in:
Stella Laurenzo 2021-07-29 03:19:14 +00:00
parent 7e2174c253
commit 768f562641
1 changed files with 31 additions and 5 deletions

View File

@ -531,10 +531,17 @@ cc_library(
cc_library(
name = "MLIRBindingsPythonCore",
srcs = glob(
include = ["lib/Bindings/Python/*.cpp"],
exclude = ["lib/Bindings/Python/MainModule.cpp"],
),
srcs = [
"lib/Bindings/Python/DialectLinalg.cpp",
"lib/Bindings/Python/DialectSparseTensor.cpp",
"lib/Bindings/Python/IRAffine.cpp",
"lib/Bindings/Python/IRAttributes.cpp",
"lib/Bindings/Python/IRCore.cpp",
"lib/Bindings/Python/IRModule.cpp",
"lib/Bindings/Python/IRTypes.cpp",
"lib/Bindings/Python/Pass.cpp",
"lib/Bindings/Python/PybindUtils.cpp",
],
# These flags are needed for pybind11 to work.
copts = [
"-fexceptions",
@ -551,7 +558,6 @@ cc_library(
deps = [
":CAPIAsync",
":CAPIDebug",
":CAPIExecutionEngine",
":CAPIGPU",
":CAPIIR",
":CAPILinalg",
@ -613,6 +619,26 @@ cc_binary(
],
)
# Dynamic library with the MLIR Conversions Python extension.
cc_binary(
name = "_mlirExecutionEngine.so",
srcs = ["lib/Bindings/Python/ExecutionEngineModule.cpp"],
copts = PYBIND11_COPTS,
features = PYBIND11_FEATURES,
linkshared = 1,
linkstatic = 0,
tags = [
"manual", # External dependency
"nobuildkite", # TODO(gcmn): Add support for this target
],
deps = [
":CAPIExecutionEngine",
":MLIRBindingsPythonHeadersAndDeps",
"@pybind11",
"@python_runtime//:headers",
],
)
# Dynamic library with the MLIR Transforms Python extension.
cc_binary(
name = "_mlirTransforms.so",