diff --git a/backends/apple/coreml/compiler/torch_ops.py b/backends/apple/coreml/compiler/torch_ops.py index 18a840972c6..479d744a97e 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,12 @@ 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) + + # https://github.com/apple/coremltools/pull/2558 @register_torch_op( torch_alias=["torchao::dequantize_affine", "torchao.dequantize_affine"],