Skip to content

Commit 2c16395

Browse files
authored
[tuner] add default attribute to tuning spec (#1016)
This PR adds the default attribute `iree_codegen.tuning_spec_with_default_entrypoint` to the generated tuning spec. fixes #816 Signed-off-by: Bangtian Liu <[email protected]>
1 parent 5ca7686 commit 2c16395

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tuner/tuner/spec_builder.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
def get_placeholder_spec(context: ir.Context) -> ir.Module:
2020
spec_text = f"""
21-
module attributes {{ transform.with_named_sequence }} {{
21+
module attributes {{ transform.with_named_sequence, iree_codegen.tuning_spec_with_default_entrypoint }} {{
2222
transform.named_sequence
2323
@__kernel_config(%variant_op: !transform.any_op {{transform.readonly}}) -> !transform.any_op
2424
attributes {{ iree_codegen.tuning_spec_entrypoint }} {{
@@ -76,7 +76,7 @@ def build_td_spec(
7676
captured_values.add(operand)
7777
bbargs_str = ", ".join(bbargs)
7878
spec_text = f"""
79-
module attributes {{ transform.with_named_sequence }} {{
79+
module attributes {{ transform.with_named_sequence, iree_codegen.tuning_spec_with_default_entrypoint }} {{
8080
// Annotation Transform
8181
transform.named_sequence @apply_op_config(%op: !transform.any_op {{transform.readonly}},
8282
%config: !transform.any_param {{transform.readonly}}) {{

0 commit comments

Comments
 (0)