Skip to content

Commit cf2b5dd

Browse files
committed
split CI
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/)! ghstack-source-id: 314025102 Pull Request resolved: #1428
1 parent e5ca0a9 commit cf2b5dd

File tree

1 file changed

+26
-2
lines changed

1 file changed

+26
-2
lines changed

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

Lines changed: 26 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,8 +52,32 @@ jobs:
5252
# pyre currently does not check these assertions
5353
pyright python/tests/test_python_actors.py
5454
55-
# Run GPU Python tests
56-
LC_ALL=C pytest python/tests/ -s -v -m "not oss_skip"
55+
# Run GPU Python tests split into 10 groups sequentially
56+
# Each group runs separately with process cleanup in between
57+
for GROUP in {1..10}; do
58+
echo "Running test group $GROUP of 10..."
59+
60+
# Kill any existing Python processes to ensure clean state
61+
echo "Cleaning up Python processes before group $GROUP..."
62+
pkill -9 python || true
63+
pkill -9 pytest || true
64+
65+
# Wait a moment for processes to terminate
66+
sleep 2
67+
68+
# Run tests for this group
69+
LC_ALL=C pytest python/tests/ -s -v -m "not oss_skip" \
70+
--dist=no \
71+
--group=$GROUP \
72+
--splits=10 || {
73+
echo "Test group $GROUP failed with exit code $?"
74+
exit 1
75+
}
76+
77+
echo "Completed test group $GROUP of 10"
78+
done
79+
80+
echo "All test groups completed successfully!"
5781
# TODO(meriksen): temporarily disabled to unblock lands while debugging
5882
# mock CUDA issues on the OSS setup
5983
# python python/tests/test_mock_cuda.py

0 commit comments

Comments
 (0)