Fix name for stream affinity (#331)

Change from attr name from `iree.abi.affinity` to `stream.affinity`
iree-org · Dec 13, 2024 · ebc0fa4 · ebc0fa4
1 parent 4fcd272
commit ebc0fa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/iree/turbine/aot/support/ir_utils.py b/iree/turbine/aot/support/ir_utils.py
@@ -290,7 +290,7 @@ def create_tensor_global(
             if attrs.mutable:
                 ir_attrs["is_mutable"] = UnitAttr.get()
             if device:
-                ir_attrs["iree.abi.affinity"] = Attribute.parse(
+                ir_attrs["stream.affinity"] = Attribute.parse(
                     f"#hal.device.promise<@__device_{device.ordinal}>"
                 )
             if external: