diff --git a/iree/turbine/kernel/compiler/dispatch_codegen.py b/iree/turbine/kernel/compiler/dispatch_codegen.py index 89d626cd..2add03f6 100644 --- a/iree/turbine/kernel/compiler/dispatch_codegen.py +++ b/iree/turbine/kernel/compiler/dispatch_codegen.py @@ -167,7 +167,7 @@ def abi_type(binding: BindingDesc): def_func_op.attributes["translation_info"] = Attribute.parse( f"#iree_codegen.translation_info" + f"subgroup_size={subgroup_size}, {{}}>" ) # Define the export.