Add an OpInfo test for split_with_sizes_copy (#118512)
Adding an `OpInfo` test for `split_with_sizes_copy` so we can use it to test [CUDA fast path for split_with_sizes_copy.out](https://github.com/pytorch/pytorch/pull/117203). Since the `OpInfo` test doesn't exist yet and introducing it requires modifications to the `CompositeExplicitAutograd` impl, adding the `OpInfo` test in a separate PR to establish a healthy baseline.
Changes made:
- Registered a batching rule for `split_with_sizes_copy`.
- Registered a decomposition for `split_with_sizes_copy`.
- Registered a DTensor prop rule for `split_with_sizes_copy`.
- Added required dtype and device checks to the composite impl.
- Added output resize to the composite impl.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/118512
Approved by: https://github.com/albanD
diff --git a/test/expect/HasDecompTest.test_has_decomposition.expect b/test/expect/HasDecompTest.test_has_decomposition.expect
index 24fce4f..9f7e2ed 100644
--- a/test/expect/HasDecompTest.test_has_decomposition.expect
+++ b/test/expect/HasDecompTest.test_has_decomposition.expect
@@ -1288,8 +1288,6 @@
aten::special_shifted_chebyshev_polynomial_w.x_scalar_out
aten::split_copy.Tensor
aten::split_copy.Tensor_out
-aten::split_with_sizes_copy
-aten::split_with_sizes_copy.out
aten::squeeze_
aten::squeeze_.dim
aten::squeeze_.dims
diff --git a/test/test_mps.py b/test/test_mps.py
index 57eca9e..b0e64b3 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -297,6 +297,7 @@
'slice',
'split',
'split_with_sizes',
+ 'split_with_sizes_copy',
'splitlist_args',
'squeeze',
'squeezemultiple',
