aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorYijia Gu <yijiagu@google.com>2023-08-23 21:39:28 -0700
committerYijia Gu <yijiagu@google.com>2023-08-23 21:39:28 -0700
commit83cc73a3d401e09eb818f7f0db15d65890773c3f (patch)
tree992311a3e9edf5c7a8b0f2507af570cdd8ba3ed2
parent8b82ae0b8d523f2de3a1dabad9d006b57917af71 (diff)
downloadllvm-83cc73a3d401e09eb818f7f0db15d65890773c3f.zip
llvm-83cc73a3d401e09eb818f7f0db15d65890773c3f.tar.gz
llvm-83cc73a3d401e09eb818f7f0db15d65890773c3f.tar.bz2
add openmp python binding in bazel
-rw-r--r--utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel33
1 files changed, 32 insertions, 1 deletions
diff --git a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
index f36bd236..f15f6de 100644
--- a/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
+++ b/utils/bazel/llvm-project-overlay/mlir/python/BUILD.bazel
@@ -543,6 +543,37 @@ filegroup(
)
##---------------------------------------------------------------------------##
+# OpenMP dialect.
+##---------------------------------------------------------------------------##
+
+gentbl_filegroup(
+ name = "OpenMPOpsPyGen",
+ tbl_outs = [
+ (
+ [
+ "-gen-python-op-bindings",
+ "-bind-dialect=omp",
+ ],
+ "mlir/dialects/_omp_ops_gen.py",
+ ),
+ ],
+ tblgen = "//third_party/llvm/llvm-project/mlir:mlir-tblgen",
+ td_file = "mlir/dialects/OpenMPOps.td",
+ deps = [
+ "//third_party/llvm/llvm-project/mlir:OpBaseTdFiles",
+ "//third_party/llvm/llvm-project/mlir:OpenMPOpsTdFiles",
+ ],
+)
+
+filegroup(
+ name = "OpenMPOpsPyFiles",
+ srcs = [
+ "mlir/dialects/openmp.py",
+ ":OpenMPOpsPyGen",
+ ],
+)
+
+##---------------------------------------------------------------------------##
# PDL dialect.
##---------------------------------------------------------------------------##
@@ -1097,7 +1128,7 @@ gentbl_filegroup(
"-gen-python-enum-bindings",
"-bind-dialect=transform",
],
- "mlir/dialects/_structure_transform_enum_gen.py",
+ "mlir/dialects/_structured_transform_enum_gen.py",
),
],
tblgen = "//mlir:mlir-tblgen",