Repository navigation
[ci]: Run GPU CI in Torch 2.13/CUDA 13.0 containers - #6463
Merged
Merged
Conversation
Allow the 2.13 Torch line while retaining the 2.5.1 baseline. Pin a dependency-only CI image, replace the legacy testmon VCS dependency, and cover both scaled_mm return contracts without changing the existing 2.13 eager fallback.
for more information, see https://pre-commit.ci
Probe legacy AMP and fused normalization separately without unused imports. Keep checkpoint coverage on native normalization when Apex kernels are unavailable, with dependency-free guard regression tests.
kiseki-zj
marked this pull request as ready for review
October 13, 2026 15:32
Remove the old Build on PR entrypoint and its workflow_run coverage consumer. Document the three-layer replacement and retirement boundaries, and guard against restoring stale workflows while retaining batch 3 AOT compilation.
richardoo-707
approved these changes
Oct 14, 2026
richardoo-707
left a comment
Collaborator
There was a problem hiding this comment.
Approved to proceed with maintainer CR. Layer 1 and Layer 2 checks are green. The documented container Layer 3 four-/eight-GPU validation gaps and skipped optional backends remain outstanding; this approval does not certify complete GPU coverage.
yuxuan-lou
approved these changes
Oct 14, 2026
JazyJiang
requested review from
richardoo-707
and removed request for
richardoo-707
October 16, 2026 01:52
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
📌 Checklist before creating the PR
🚨 Issue number
No tracking issue linked yet.
📝 What does this PR do?
Migrate the existing layered GPU CI from a developer-specific host
environment to disposable containers using a pinned Torch 2.13 /
CUDA 13.0 dependency image, and retire the legacy PR build entrypoints.
Dependency profile
torch>=2.2.0,<=2.5.1to
torch>=2.2.0,<2.14.dependencies and CUDA 13.0 build tools.
pytest-testmon==2.2.0.The installation constraint does not imply validation of every
Torch/CUDA combination. This PR does not upgrade Transformers to 5.x.
Container execution
no registry publication or automatic image pull is required.
container. The dependency image contains no ColossalAI checkout
or prebuilt ColossalAI extensions.
shared host lock.
Docker socket into test containers.
runtime metadata, and clean up only the current job's containers.
BUILD_EXT=1in batch 3;other batches compile required kernels on demand.
Workflow routing and retirement
4ausing the separate
COLOSSAL_CI_SMOKE_GPU_POOL.batches
2 4a 5.e1-gpu-on-pr.ymlbuild_on_pr.ymlreport_test_coverage.ymland unrelated example, compatibility, scheduled-build and release
workflows.
The retired coverage-comment workflow is not replaced by code-coverage
measurement: JUnit test reports are not coverage reports.
Regression coverage
_scaled_mmreturn contracts, FP16/BF16, bias and batched inputs.
checkpoint tests.
routing, GPU-pool configuration and retired workflow references.
The FP8 implementation compatibility changes already exist in main;
this PR adds regression coverage around that behavior.
✅ Validation
Current head:
9278d34bLayer 2 uses
fast_mode=true; batch 4a is unaffected by FAST_TEST.Results:
Partial Layer 3 validation:
31e3c352Run with
fast_mode=false:2 3 4a 4b 4c 4e 5completed successfully,with some optional cases skipped.
6 7 8 9 9qfailed before test execution because onlytwo idle GPUs were available when four were required.
4dand11were entirely skipped.10was not selected.This run does not establish full Layer 3 coverage. Four-GPU batches
still require validation with sufficient authorized resources.
Layer 3 report
backends are not included in this image; skipped paths remain
unvalidated.
not establish
torch.compilesupport.CPU-operation validation with this profile.
self-hosted runners; trusted-ref restrictions remain necessary.
required-check settings.
version.txt, publish a PyPI package,or publish a user-facing Docker image.
💥 Checklist before requesting a review
explicit maintainer-approved rollout exception documented