Skip to content

Commit 266c94b

Browse files
authored
Register split_copy in CoreML (#12836)
As titled.
1 parent e4914b4 commit 266c94b

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

backends/apple/coreml/compiler/torch_ops.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
_get_inputs,
1616
NUM_TO_NUMPY_DTYPE,
1717
NUM_TO_TORCH_DTYPE,
18+
split,
1819
transpose,
1920
unbind,
2021
)
@@ -37,6 +38,12 @@ def unbind_copy(context, node):
3738
unbind(context, node)
3839

3940

41+
# https://github.com/apple/coremltools/pull/2563
42+
@register_torch_op(override=False)
43+
def split_copy(context, node):
44+
split(context, node)
45+
46+
4047
# https://github.com/apple/coremltools/pull/2558
4148
@register_torch_op(
4249
torch_alias=["torchao::dequantize_affine", "torchao.dequantize_affine"],

0 commit comments

Comments
 (0)