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/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',
