From 4ffe7bde773392ec148a012bceed4e4888db5545 Mon Sep 17 00:00:00 2001 From: Scott Todd Date: Fri, 15 Nov 2024 11:35:17 -0800 Subject: [PATCH] Add ending {} to translation info attribute. --- iree/turbine/kernel/compiler/dispatch_codegen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.