Fix FlatSymbolRefAttr (#732)

FlatSymbolRefAttr creation was failing when exporting with `iree` at head due to LLVM changes.
nod-ai · Dec 31, 2024 · ffb0dd2 · ffb0dd2
1 parent f5e9cb4
commit ffb0dd2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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(