This is an archive of the discontinued LLVM Phabricator instance.

Differential D122138

[mlir][bazel] filegroups for Python CF, PDL, Tensor dialects
ClosedPublic

Authored by ftynse on Mar 21 2022, 7:28 AM.

Details

Reviewers
None
Commits
rGb9959590d425: [mlir][bazel] filegroups for Python CF, PDL, Tensor dialects
Summary

These dialects have Python bindings and are tested, but were not
previously exposed through Bazel filegroups.

Diff Detail

Event Timeline

ftynse created this revision.Mar 21 2022, 7:28 AM
Herald added a project: Restricted Project. · View Herald TranscriptMar 21 2022, 7:28 AM
Herald added subscribers: sdasgup3, wenzhicui, wrengr and 20 others. · View Herald Transcript
ftynse requested review of this revision.Mar 21 2022, 7:28 AM
Herald added a project: Restricted Project. · View Herald TranscriptMar 21 2022, 7:28 AM
Herald added subscribers: llvm-commits, stephenneuendorffer, nicolasvasilache. · View Herald Transcript
Harbormaster completed remote builds in B155380: Diff 416930.Mar 21 2022, 7:33 AM
ftynse updated this revision to Diff 416935.Mar 21 2022, 7:39 AM

Rebase

Harbormaster completed remote builds in B155383: Diff 416935.Mar 21 2022, 7:45 AM
This revision was not accepted when it landed; it landed in state Needs Review.Mar 21 2022, 7:46 AM
Closed by commit rGb9959590d425: [mlir][bazel] filegroups for Python CF, PDL, Tensor dialects (authored by ftynse). · Explain Why
This revision was automatically updated to reflect the committed changes.
ftynse added a commit: rGb9959590d425: [mlir][bazel] filegroups for Python CF, PDL, Tensor dialects.

Revision Contents

PathSize
utils/
bazel/
llvm-project-overlay/
mlir/
python/
130 lines

Diff 416930

utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel

# This file is licensed under the Apache License v2.0 with LLVM Exceptions. # This file is licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information. # See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
# Description: # Description:
# Python bindings for MLIR. # Python bindings for MLIR.
# #
# We define separate filegroups for files in different directories so # We define separate filegroups for files in different directories so
# that downstream users can mirror the tree in their own py_library() rules. # that downstream users can mirror the tree in their own py_library() rules.
load("//mlir:tblgen.bzl", "gentbl_filegroup", "td_library") load("//mlir:tblgen.bzl", "gentbl_filegroup", "td_library")
package( package(
default_visibility = [ default_visibility = ["//visibility:public"],
"//visibility:public",
],
licenses = ["notice"], licenses = ["notice"],
) )
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
# Core IR modules. # Core IR modules.
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
filegroup( filegroup(
▲ Show 20 Lines • Show All 225 Lines • ▼ Show 20 Linesfilegroup(
srcs = [ srcs = [
"mlir/dialects/_arith_ops_ext.py", "mlir/dialects/_arith_ops_ext.py",
"mlir/dialects/arith.py", "mlir/dialects/arith.py",
":ArithmeticOpsPyGen", ":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. # Math dialect.
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
td_library( td_library(
name = "MathOpsPyTdFiles", name = "MathOpsPyTdFiles",
srcs = [ srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td", "//mlir:include/mlir/Bindings/Python/Attributes.td",
], ],
▲ Show 20 Lines • Show All 161 Lines • ▼ Show 20 Linesfilegroup(
name = "PythonTestPyFiles", name = "PythonTestPyFiles",
srcs = [ srcs = [
"mlir/dialects/python_test.py", "mlir/dialects/python_test.py",
":PythonTestPyGen", ":PythonTestPyGen",
], ],
) )
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
# 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 = "PDLPyFiles",
srcs = [
"mlir/dialects/_pdl_ops_ext.py",
"mlir/dialects/pdl.py",
":PDLPyGen",
],
)
##---------------------------------------------------------------------------##
# Quant dialect. # Quant dialect.
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
filegroup( filegroup(
name = "QuantPyIFiles", name = "QuantPyIFiles",
srcs = [ srcs = [
"mlir/_mlir_libs/_mlir/dialects/quant.pyi", "mlir/_mlir_libs/_mlir/dialects/quant.pyi",
], ],
) )
filegroup( filegroup(
name = "QuantPyFiles", name = "QuantPyFiles",
srcs = [ srcs = [
"mlir/dialects/quant.py", "mlir/dialects/quant.py",
], ],
) )
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
# SCF dialect. # SCF dialect.
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
td_library( td_library(
name = "SCFPyTdFiles", name = "SCFPyTdFiles",
srcs = [ srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td", "//mlir:include/mlir/Bindings/Python/Attributes.td",
▲ Show 20 Lines • Show All 153 Lines • ▼ Show 20 Linesfilegroup(
name = "SparseTensorOpsPyFiles", name = "SparseTensorOpsPyFiles",
srcs = [ srcs = [
"mlir/dialects/sparse_tensor.py", "mlir/dialects/sparse_tensor.py",
":SparseTensorOpsPyGen", ":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. # Tosa dialect.
##---------------------------------------------------------------------------## ##---------------------------------------------------------------------------##
td_library( td_library(
name = "TosaOpsPyTdFiles", name = "TosaOpsPyTdFiles",
srcs = [ srcs = [
"//mlir:include/mlir/Bindings/Python/Attributes.td", "//mlir:include/mlir/Bindings/Python/Attributes.td",
], ],
▲ Show 20 Lines • Show All 73 Lines • Show Last 20 Lines