diff --git a/iree/turbine/aot/support/ir_utils.py b/iree/turbine/aot/support/ir_utils.py
index 348cf292..7ab36e57 100644
--- 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: