diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index 67de7076fa..338d377a36 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -2515,7 +2515,7 @@ def aten_copy( ) -> TTensor: """copy(Tensor self, Tensor src, bool non_blocking=False) -> Tensor""" - return op.CastLike(src, self) + return op.Identity(op.CastLike(src, self)) @torch_op("aten::_to_copy", trace_only=True)