Restrict iree_bytecode_module to a single src file.
The underlying iree-translate / mlir-translate doesn't support a list of files, and we haven't defined what behavior we would want from passing multiple files:
iree-translate.exe: Too many positional arguments specified!
Can specify at most 1 positional arguments
PiperOrigin-RevId: 284796661
diff --git a/iree/samples/hal/BUILD b/iree/samples/hal/BUILD
index 004406d..da7c20a 100644
--- a/iree/samples/hal/BUILD
+++ b/iree/samples/hal/BUILD
@@ -25,7 +25,7 @@
iree_bytecode_module(
name = "simple_compute_test_module",
- srcs = ["simple_compute_test.mlir"],
+ src = "simple_compute_test.mlir",
cc_namespace = "iree::hal::samples",
)
diff --git a/iree/samples/rt/BUILD b/iree/samples/rt/BUILD
index c19d5d6..029c490 100644
--- a/iree/samples/rt/BUILD
+++ b/iree/samples/rt/BUILD
@@ -24,7 +24,7 @@
iree_bytecode_module(
name = "simple_module_test_bytecode_module",
- srcs = ["simple_module_test.mlir"],
+ src = "simple_module_test.mlir",
cc_namespace = "iree::rt::samples",
)
diff --git a/iree/samples/rt/vulkan/BUILD b/iree/samples/rt/vulkan/BUILD
index 516cfd0..d172fe3 100644
--- a/iree/samples/rt/vulkan/BUILD
+++ b/iree/samples/rt/vulkan/BUILD
@@ -23,7 +23,7 @@
iree_bytecode_module(
name = "simple_mul_bytecode_module",
- srcs = ["simple_mul.mlir"],
+ src = "simple_mul.mlir",
cc_namespace = "iree::rt::samples",
)
diff --git a/iree/tools/compilation.bzl b/iree/tools/compilation.bzl
index 0fe3085..5978bb0 100644
--- a/iree/tools/compilation.bzl
+++ b/iree/tools/compilation.bzl
@@ -19,13 +19,13 @@
# TODO(benvanik): port to a full starlark rule, document, etc.
def iree_bytecode_module(
name,
- srcs,
+ src,
translation = "-mlir-to-iree-module",
cc_namespace = None,
visibility = None):
native.genrule(
name = name,
- srcs = srcs,
+ srcs = [src],
outs = [
"%s.emod" % (name),
],
@@ -34,7 +34,8 @@
"$(location //iree/tools:iree-translate)",
translation,
"-o $(location %s.emod)" % (name),
- ] + ["$(locations %s)" % (src) for src in srcs]),
+ "$(location %s)" % (src),
+ ]),
]),
tools = [
"//iree/tools:iree-translate",
diff --git a/iree/vm2/BUILD b/iree/vm2/BUILD
index 771827f..100f7c8 100644
--- a/iree/vm2/BUILD
+++ b/iree/vm2/BUILD
@@ -23,7 +23,7 @@
iree_bytecode_module(
name = "bytecode_dispatch_test_module",
- srcs = ["bytecode_dispatch_test.mlir"],
+ src = "bytecode_dispatch_test.mlir",
cc_namespace = "iree::vm",
translation = "-iree-vm-ir-to-bytecode-module",
)
@@ -67,7 +67,7 @@
iree_bytecode_module(
name = "bytecode_module_benchmark_module",
- srcs = ["bytecode_module_benchmark.mlir"],
+ src = "bytecode_module_benchmark.mlir",
cc_namespace = "iree::vm",
translation = "-iree-vm-ir-to-bytecode-module",
)
