From ffb0dd2c6be7106e725d49a25b591a3e467913f3 Mon Sep 17 00:00:00 2001 From: "Xida Ren (Cedar)" Date: Tue, 31 Dec 2024 13:31:53 -0500 Subject: [PATCH] Fix FlatSymbolRefAttr (#732) FlatSymbolRefAttr creation was failing when exporting with `iree` at head due to LLVM changes. --- sharktank/sharktank/kernels/base.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/sharktank/sharktank/kernels/base.py b/sharktank/sharktank/kernels/base.py index ce792b525..30791a828 100644 --- a/sharktank/sharktank/kernels/base.py +++ b/sharktank/sharktank/kernels/base.py @@ -102,9 +102,7 @@ def _get_jinja2_env() -> Environment: def call_function(target_function: Operation, *operands: Value) -> Sequence[Value]: - target_symbol = FlatSymbolRefAttr.get( - StringAttr(target_function.attributes["sym_name"]).value_bytes - ) + target_symbol = FlatSymbolRefAttr.get(target_function.attributes["sym_name"].value) ftype = FunctionType(TypeAttr(target_function.attributes["function_type"]).value) operands = [i for i in operands if i is not None] return Operation.create(