Function _copy_handler

backends/mlx/ops.py:947–964

Handle aten.copy - copy data from src to self. Schema: aten::copy(Tensor self, Tensor src, bool non_blocking=False) -> Tensor In functionalized Edge IR, this returns a copy of src (args[1]).

(P: MLXProgramBuilder, n: Node)

@REGISTRY.register(target=[torch.ops.aten.copy.default])
def _copy_handler(P: MLXProgramBuilder, n: Node) -> Slot:
    """Handle aten.copy - copy data from src to self.

    Schema: aten::copy(Tensor self, Tensor src, bool non_blocking=False) -> Tensor
    In functionalized Edge IR, this returns a copy of src (args[1]).
    """
    args = P.args(n)
    require_args(args, 2, 2, "aten.copy")
    require_kwargs(P.kwargs(n), {"non_blocking"}, "aten.copy")
    src = args[1]
    out = P.make_or_get_slot(n)
    P.emit(
        ContiguousNode(
            x=P.slot_to_tid(src),
            out=P.slot_to_tid(out),
        )
    )
    return out


@REGISTRY.register(target=[exir_ops.edge.dim_order_ops._clone_dim_order.default])