diff --git a/iree/turbine/aot/support/ir_utils.py b/iree/turbine/aot/support/ir_utils.py index 4a402691..288ef323 100644 --- a/iree/turbine/aot/support/ir_utils.py +++ b/iree/turbine/aot/support/ir_utils.py @@ -295,7 +295,7 @@ def create_tensor_global( f"#hal.device.promise<@__device_{device.ordinal}>" ) else: - queues = ", ".join(device.queues) + queues = ', '.join(device.queues) ir_attrs["iree.abi.affinity"] = Attribute.parse( f"#hal.device.promise<@__device_{device.ordinal}, [{queues}]>" )