split CI

Pull Request resolved: #1429


ghstack-source-id: 314035861
@exported-using-ghexport

Differential Revision: [D83884590](https://our.internmc.facebook.com/intern/diff/D83884590/)

**NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D83884590/)!

Author: zdevito

.github/workflows/test-gpu-python.yml

@@ -52,8 +52,49 @@ jobs:
         # pyre currently does not check these assertions
         pyright python/tests/test_python_actors.py
 
-        # Run GPU Python tests
-        LC_ALL=C pytest python/tests/ -s -v -m "not oss_skip"
+        # Run GPU Python tests split into 10 groups sequentially
+        # Each group runs separately with process cleanup in between
+        pip install pytest-split
+        FAILED_GROUPS=()
+
+        for GROUP in {1..10}; do
+          echo "Running test group $GROUP of 10..."
+
+          # Kill any existing Python processes to ensure clean state
+          echo "Cleaning up Python processes before group $GROUP..."
+          pkill -9 python || true
+          pkill -9 pytest || true
+
+          # Wait a moment for processes to terminate
+          sleep 2
+
+          # Run tests for this group
+          if LC_ALL=C pytest python/tests/ -s -v -m "not oss_skip" \
+            --ignore-glob="**/meta/**" \
+            --dist=no \
+            --group=$GROUP \
+            --splits=10; then
+            echo "✓ Test group $GROUP completed successfully"
+          else
+            FAILED_GROUPS+=($GROUP)
+            echo "✗ Test group $GROUP failed with exit code $?"
+          fi
+
+        done
+
+        # Final cleanup after all groups
+        echo "Final cleanup of Python processes..."
+        pkill -9 python || true
+        pkill -9 pytest || true
+
+        # Check if any groups failed and exit with appropriate code
+        if [ ${#FAILED_GROUPS[@]} -eq 0 ]; then
+          echo "✓ All test groups completed successfully!"
+        else
+          echo "✗ The following test groups failed: ${FAILED_GROUPS[*]}"
+          echo "Failed groups count: ${#FAILED_GROUPS[@]}/10"
+          exit 1
+        fi
         # TODO(meriksen): temporarily disabled to unblock lands while debugging
         # mock CUDA issues on the OSS setup
         # python python/tests/test_mock_cuda.py