From b0dba9c1d70f94ee3cebc7e371b9d10e8529f3fa Mon Sep 17 00:00:00 2001 From: Scott Roy <161522778+metascroy@users.noreply.github.com> Date: Thu, 24 Jul 2025 15:58:54 -0700 Subject: [PATCH 1/2] up --- backends/apple/coreml/compiler/torch_ops.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/backends/apple/coreml/compiler/torch_ops.py b/backends/apple/coreml/compiler/torch_ops.py index 18a840972c6..655a474f600 100644 --- a/backends/apple/coreml/compiler/torch_ops.py +++ b/backends/apple/coreml/compiler/torch_ops.py @@ -15,6 +15,7 @@ _get_inputs, NUM_TO_NUMPY_DTYPE, NUM_TO_TORCH_DTYPE, + split, transpose, unbind, ) @@ -37,6 +38,11 @@ def unbind_copy(context, node): unbind(context, node) +@register_torch_op(override=False) +def split_copy(context, node): + split(context, node) + + # https://github.com/apple/coremltools/pull/2558 @register_torch_op( torch_alias=["torchao::dequantize_affine", "torchao.dequantize_affine"], From c24f3e8bcf64773901ec8298c8ea5221f2dfbbd3 Mon Sep 17 00:00:00 2001 From: Scott Roy <161522778+metascroy@users.noreply.github.com> Date: Mon, 28 Jul 2025 16:23:50 -0700 Subject: [PATCH 2/2] Update torch_ops.py --- backends/apple/coreml/compiler/torch_ops.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backends/apple/coreml/compiler/torch_ops.py b/backends/apple/coreml/compiler/torch_ops.py index 655a474f600..479d744a97e 100644 --- a/backends/apple/coreml/compiler/torch_ops.py +++ b/backends/apple/coreml/compiler/torch_ops.py @@ -38,6 +38,7 @@ def unbind_copy(context, node): unbind(context, node) +# https://github.com/apple/coremltools/pull/2563 @register_torch_op(override=False) def split_copy(context, node): split(context, node)