diff options
author | Yijia Gu <yijiagu@google.com> | 2023-08-23 21:39:28 -0700 |
---|---|---|
committer | Yijia Gu <yijiagu@google.com> | 2023-08-23 21:39:28 -0700 |
commit | 83cc73a3d401e09eb818f7f0db15d65890773c3f (patch) | |
tree | 992311a3e9edf5c7a8b0f2507af570cdd8ba3ed2 | |
parent | 8b82ae0b8d523f2de3a1dabad9d006b57917af71 (diff) | |
download | llvm-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.bazel | 33 |
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", |