Skip to content

Commit 289eb99

Browse files
committed
[mlir][bazel] Add SPIRV python binding
1 parent dea30ac commit 289eb99

File tree

1 file changed

+31
-0
lines changed
  • utils/bazel/llvm-project-overlay/mlir/python

1 file changed

+31
-0
lines changed

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

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1057,6 +1057,37 @@ filegroup(
10571057
],
10581058
)
10591059

1060+
##---------------------------------------------------------------------------##
1061+
# SPIRV dialect.
1062+
##---------------------------------------------------------------------------##
1063+
1064+
gentbl_filegroup(
1065+
name = "SPIRVOpsPyGen",
1066+
tbl_outs = [
1067+
(
1068+
[
1069+
"-gen-python-op-bindings",
1070+
"-bind-dialect=spirv",
1071+
],
1072+
"mlir/dialects/_spirv_ops_gen.py",
1073+
),
1074+
],
1075+
tblgen = "//mlir:mlir-tblgen",
1076+
td_file = "mlir/dialects/SPIRVOps.td",
1077+
deps = [
1078+
"//mlir:OpBaseTdFiles",
1079+
"//mlir:SPIRVOpsTdFiles",
1080+
],
1081+
)
1082+
1083+
filegroup(
1084+
name = "SPIRVOpsPyFiles",
1085+
srcs = [
1086+
"mlir/dialects/spirv.py",
1087+
":SPIRVOpsPyGen",
1088+
],
1089+
)
1090+
10601091
##---------------------------------------------------------------------------##
10611092
# Tensor dialect.
10621093
##---------------------------------------------------------------------------##

0 commit comments

Comments
 (0)