shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
llvm-project/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
Alex Zinenko b9959590d4 [mlir][bazel] filegroups for Python CF, PDL, Tensor dialects 
These dialects have Python bindings and are tested, but were not
previously exposed through Bazel filegroups.

Differential Revision: https://reviews.llvm.org/D122138
2022-03-21 15:46:36 +01:00

782 lines
17 KiB
Python
Raw Blame History

# This file is licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
# Description:
# Python bindings for MLIR.
#
# We define separate filegroups for files in different directories so
# that downstream users can mirror the tree in their own py_library() rules.
load("//mlir:tblgen.bzl", "gentbl_filegroup", "td_library")
package(
default_visibility = ["//visibility:public"],
licenses = ["notice"],
)
##---------------------------------------------------------------------------##
# Core IR modules.
##---------------------------------------------------------------------------##
filegroup(
name = "ConversionsPyFiles",
srcs = glob([
"mlir/conversions/*.py",
]),
)
filegroup(
name = "DialectCorePyFiles",
srcs = [
"mlir/dialects/_ods_common.py",
],
)
filegroup(
name = "ExecutionEnginePyFiles",
srcs = [
"mlir/execution_engine.py",
],
)
filegroup(
name = "ExecutionEnginePyIFiles",
srcs = [
"mlir/_mlir_libs/_mlirExecutionEngine.pyi",
],
)
filegroup(
name = "IRPyFiles",
srcs = [
"mlir/ir.py",
],
)
filegroup(
name = "IRPyIFiles",
srcs = [
"mlir/_mlir_libs/_mlir/__init__.pyi",
"mlir/_mlir_libs/_mlir/ir.pyi",
],
)
filegroup(
name = "PassManagerPyFiles",
srcs = [
"mlir/passmanager.py",
],
)
filegroup(
name = "PassManagerPyIFiles",
srcs = [
"mlir/_mlir_libs/_mlir/passmanager.pyi",
],
)
filegroup(
name = "RuntimePyFiles",
srcs = glob([
"mlir/runtime/*.py",
]),
)
filegroup(
name = "TransformsPyFiles",
srcs = glob([
"mlir/transforms/*.py",
]),
)
filegroup(
name = "AllPassesRegistrationPyFiles",
srcs = glob([
"mlir/all_passes_registration/*.py",
]),
)
##---------------------------------------------------------------------------##
# Builtin dialect.
##---------------------------------------------------------------------------##
td_library(
name = "BuiltinOpsPyTdFiles",
srcs = [
"mlir/dialects/BuiltinOps.td",
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:BuiltinDialectTdFiles",
"//mlir:FunctionInterfacesTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "BuiltinOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=builtin",
],
"mlir/dialects/_builtin_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/BuiltinOps.td",
deps = [
":BuiltinOpsPyTdFiles",
],
)
filegroup(
name = "BuiltinOpsPyFiles",
srcs = [
"mlir/dialects/_builtin_ops_ext.py",
"mlir/dialects/builtin.py",
":BuiltinOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# Linalg dialect.
##---------------------------------------------------------------------------##
td_library(
name = "LinalgOpsPyTdFiles",
srcs = [
"mlir/dialects/LinalgOps.td",
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:LinalgOpsTdFiles",
"//mlir:LinalgStructuredOpsTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "LinalgOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=linalg",
],
"mlir/dialects/_linalg_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/LinalgOps.td",
deps = [
":LinalgOpsPyTdFiles",
],
)
filegroup(
name = "LinalgOpsPyFiles",
srcs = [
"mlir/dialects/_linalg_ops_ext.py",
":LinalgOpsPyGen",
],
)
filegroup(
name = "LinalgOpsPackagePyFiles",
srcs = glob(["mlir/dialects/linalg/*.py"]),
)
filegroup(
name = "LinalgOpsPackageOpDSLPyFiles",
srcs = glob(["mlir/dialects/linalg/opdsl/*.py"]),
)
filegroup(
name = "LinalgOpsPackageOpDSLLangPyFiles",
srcs = glob(["mlir/dialects/linalg/opdsl/lang/*.py"]),
)
filegroup(
name = "LinalgOpsPackageOpDSLOpsPyFiles",
srcs = glob(["mlir/dialects/linalg/opdsl/ops/*.py"]),
)
filegroup(
name = "LinalgOpsPackagePassesPyFiles",
srcs = glob(["mlir/dialects/linalg/passes/*.py"]),
)
##---------------------------------------------------------------------------##
# Arithmetic dialect.
##---------------------------------------------------------------------------##
td_library(
name = "ArithmeticOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
includes = ["../include"],
deps = [
"//mlir:ArithmeticOpsTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "ArithmeticOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=arith",
],
"mlir/dialects/_arith_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/ArithmeticOps.td",
deps = [
":ArithmeticOpsPyTdFiles",
],
)
filegroup(
name = "ArithmeticOpsPyFiles",
srcs = [
"mlir/dialects/_arith_ops_ext.py",
"mlir/dialects/arith.py",
":ArithmeticOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# ControlFlow dialect.
##---------------------------------------------------------------------------##
td_library(
name = "ControlFlowOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:ControlFlowOpsTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "ControlFlowOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=cf",
],
"mlir/dialects/_cf_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/ControlFlowOps.td",
deps = [
":ControlFlowOpsPyTdFiles",
],
)
filegroup(
name = "ControlFlowOpsPyFiles",
srcs = [
"mlir/dialects/cf.py",
":ControlFlowOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# Math dialect.
##---------------------------------------------------------------------------##
td_library(
name = "MathOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
includes = ["../include"],
deps = [
"//mlir:MathOpsTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "MathOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=math",
],
"mlir/dialects/_math_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/MathOps.td",
deps = [
":MathOpsPyTdFiles",
],
)
filegroup(
name = "MathOpsPyFiles",
srcs = [
"mlir/dialects/math.py",
":MathOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# MemRef dialect.
##---------------------------------------------------------------------------##
td_library(
name = "MemRefOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
includes = ["../include"],
deps = [
"//mlir:MemRefOpsTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "MemRefOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=memref",
],
"mlir/dialects/_memref_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/MemRefOps.td",
deps = [
":MemRefOpsPyTdFiles",
],
)
filegroup(
name = "MemRefOpsPyFiles",
srcs = [
"mlir/dialects/_memref_ops_ext.py",
"mlir/dialects/memref.py",
":MemRefOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# PDL dialect.
##---------------------------------------------------------------------------##
td_library(
name = "PDLPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
includes = ["../include"],
deps = [
"//mlir:OpBaseTdFiles",
"//mlir:PDLDialectTdFiles",
],
)
gentbl_filegroup(
name = "PDLPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=pdl",
],
"mlir/dialects/_pdl_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/PDLOps.td",
deps = [
":PDLPyTdFiles",
],
)
filegroup(
name = "PDLPyIFiles",
srcs = [
"mlir/_mlir_libs/_mlir/dialects/pdl.pyi",
],
)
filegroup(
name = "PDLPyFiles",
srcs = [
"mlir/dialects/_pdl_ops_ext.py",
"mlir/dialects/pdl.py",
":PDLPyGen",
],
)
##---------------------------------------------------------------------------##
# PythonTest dialect.
##---------------------------------------------------------------------------##
td_library(
name = "PythonTestPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:InferTypeOpInterfaceTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "PythonTestPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=python_test",
],
"mlir/dialects/_python_test_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "//mlir/test/python:python_test_ops.td",
deps = [
":PythonTestPyTdFiles",
],
)
filegroup(
name = "PythonTestPyFiles",
srcs = [
"mlir/dialects/python_test.py",
":PythonTestPyGen",
],
)
##---------------------------------------------------------------------------##
# Quant dialect.
##---------------------------------------------------------------------------##
filegroup(
name = "QuantPyIFiles",
srcs = [
"mlir/_mlir_libs/_mlir/dialects/quant.pyi",
],
)
filegroup(
name = "QuantPyFiles",
srcs = [
"mlir/dialects/quant.py",
],
)
##---------------------------------------------------------------------------##
# SCF dialect.
##---------------------------------------------------------------------------##
td_library(
name = "SCFPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
includes = ["../include"],
deps = [
"//mlir:OpBaseTdFiles",
"//mlir:SCFTdFiles",
],
)
gentbl_filegroup(
name = "SCFPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=scf",
],
"mlir/dialects/_scf_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/SCFOps.td",
deps = [
":SCFPyTdFiles",
],
)
filegroup(
name = "SCFPyFiles",
srcs = [
"mlir/dialects/_scf_ops_ext.py",
"mlir/dialects/scf.py",
":SCFPyGen",
],
)
##---------------------------------------------------------------------------##
# Shape dialect.
##---------------------------------------------------------------------------##
td_library(
name = "ShapeOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
includes = ["../include"],
deps = [
"//mlir:OpBaseTdFiles",
"//mlir:ShapeOpsTdFiles",
],
)
gentbl_filegroup(
name = "ShapeOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=shape",
],
"mlir/dialects/_shape_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/ShapeOps.td",
deps = [
":ShapeOpsPyTdFiles",
],
)
filegroup(
name = "ShapeOpsPyFiles",
srcs = [
"mlir/dialects/shape.py",
":ShapeOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# Func dialect.
##---------------------------------------------------------------------------##
td_library(
name = "FuncPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:FuncTdFiles",
"//mlir:OpBaseTdFiles",
],
)
gentbl_filegroup(
name = "FuncPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=func",
],
"mlir/dialects/_func_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/FuncOps.td",
deps = [
":FuncPyTdFiles",
],
)
filegroup(
name = "FuncPyFiles",
srcs = [
"mlir/dialects/_func_ops_ext.py",
"mlir/dialects/func.py",
":FuncPyGen",
],
)
##---------------------------------------------------------------------------##
# SparseTensor dialect.
##---------------------------------------------------------------------------##
td_library(
name = "SparseTensorOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:OpBaseTdFiles",
"//mlir:SparseTensorTdFiles",
],
)
gentbl_filegroup(
name = "SparseTensorOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=sparse_tensor",
],
"mlir/dialects/_sparse_tensor_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/SparseTensorOps.td",
deps = [
":SparseTensorOpsPyTdFiles",
],
)
filegroup(
name = "SparseTensorOpsPyFiles",
srcs = [
"mlir/dialects/sparse_tensor.py",
":SparseTensorOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# Tensor dialect.
##---------------------------------------------------------------------------##
td_library(
name = "TensorOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:OpBaseTdFiles",
"//mlir:TensorOpsTdFiles",
],
)
gentbl_filegroup(
name = "TensorOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=tensor",
],
"mlir/dialects/_tensor_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/TensorOps.td",
deps = [
":TensorOpsPyTdFiles",
],
)
filegroup(
name = "TensorOpsPyFiles",
srcs = [
"mlir/dialects/tensor.py",
":TensorOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# Tosa dialect.
##---------------------------------------------------------------------------##
td_library(
name = "TosaOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
deps = [
"//mlir:OpBaseTdFiles",
"//mlir:TosaDialectTdFiles",
],
)
gentbl_filegroup(
name = "TosaOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=tosa",
],
"mlir/dialects/_tosa_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/TosaOps.td",
deps = [
":TosaOpsPyTdFiles",
],
)
filegroup(
name = "TosaOpsPyFiles",
srcs = [
"mlir/dialects/tosa.py",
":TosaOpsPyGen",
],
)
##---------------------------------------------------------------------------##
# Vector dialect.
##---------------------------------------------------------------------------##
td_library(
name = "VectorOpsPyTdFiles",
srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td",
],
includes = ["../include"],
deps = [
"//mlir:OpBaseTdFiles",
"//mlir:VectorOpsTdFiles",
],
)
gentbl_filegroup(
name = "VectorOpsPyGen",
tbl_outs = [
(
[
"-gen-python-op-bindings",
"-bind-dialect=vector",
],
"mlir/dialects/_vector_ops_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",
td_file = "mlir/dialects/VectorOps.td",
deps = [
":VectorOpsPyTdFiles",
],
)
filegroup(
name = "VectorOpsPyFiles",
srcs = [
"mlir/dialects/vector.py",
":VectorOpsPyGen",
],
)
Reference in New Issue View Git Blame Copy Permalink