Skip to content

ci(checkpoint): mount shared model-cache PVC in DynamoCheckpoint tests - #12400

Merged
galletas1712 merged 1 commit into
mainfrom
ci/checkpoint-model-cache-pvc
Jul 30, 2026
Merged

ci(checkpoint): mount shared model-cache PVC in DynamoCheckpoint tests#12400
galletas1712 merged 1 commit into
mainfrom
ci/checkpoint-model-cache-pvc

Conversation

@galletas1712

@galletas1712 galletas1712 commented Jul 29, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Wire DynamoCheckpoint CI to the same Azure model-cache PVC path regular deploy tests already use (setup-dynamo-operator + --model-cache-pvc).
  • Honor --model-cache-pvc in test_dynamocheckpoint.py, skipping TRTLLM container HF_HOME so it cannot shadow the shared cache.

Validation

  • CI DynamoCheckpoint jobs mount model-cache and stop cold-fetching Qwen via HF/ModelExpress when the cache is populated.

@galletas1712
galletas1712 requested review from a team as code owners July 29, 2026 22:44

@devin-ai-integration devin-ai-integration Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

✅ Devin Review: No Issues Found

Devin Review analyzed this PR and found no bugs or issues to report.

Open in Devin Review

@github-actions github-actions Bot added ci Issues/PRs that reference CI build/test actions labels Jul 29, 2026
@coderabbitai

coderabbitai Bot commented Jul 29, 2026

Copy link
Copy Markdown
Contributor

Review Change Stack

Walkthrough

Checkpoint deployment specifications and CI workflows now support an optional shared model-cache server, path, and PVC. The test helper mounts the PVC and adjusts HF_HOME handling, while vLLM, SGLang, and TRTLLM jobs conditionally enable the cache.

Changes

Shared model cache integration

Layer / File(s) Summary
Checkpoint specification cache mounting
tests/deploy/test_dynamocheckpoint.py
_new_checkpoint_spec accepts cache PVC and mount options, mounts the PVC when provided, skips container-level HF_HOME, and receives the new CLI values.
Workflow cache configuration
.github/workflows/pr.yaml
Checkpoint operator and snapshot-agent jobs pass cache server/path variables, and deploy tests conditionally pass the model-cache PVC for all three backends.

Estimated code review effort: 3 (Moderate) | ~20 minutes

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name Status Explanation Resolution
Description check ⚠️ Warning The description is missing required template sections: Overview, Details, reviewer start, and Related Issues. Add the missing template sections and include either a linked issue or the required no-issue confirmation.
✅ Passed checks (4 passed)
Check name Status Explanation
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check ✅ Passed Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check ✅ Passed Check skipped because no linked issues were found for this pull request.
Title check ✅ Passed The title clearly and concisely describes the main change: mounting a shared model-cache PVC for DynamoCheckpoint tests.

Comment @coderabbitai help to get the list of available commands.

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🧹 Nitpick comments (1)
tests/deploy/test_dynamocheckpoint.py (1)

276-279: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Add a regression test for cache environment precedence.

Verify that a PVC removes container-level HF_HOME, sets deployment-level HF_HOME to the mount path, and adds the PVC mount; also verify that the no-PVC path preserves TRTLLM’s snapshot HF_HOME.

Also applies to: 293-296

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@tests/deploy/test_dynamocheckpoint.py` around lines 276 - 279, Extend the
deployment tests around the container environment filtering and
mount_model_cache_pvc flow to cover both cache modes: with a model cache PVC,
assert container-level HF_HOME is removed, deployment-level HF_HOME points to
the mount path, and the PVC mount is present; without a PVC, assert TRTLLM’s
snapshot HF_HOME remains unchanged.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Nitpick comments:
In `@tests/deploy/test_dynamocheckpoint.py`:
- Around line 276-279: Extend the deployment tests around the container
environment filtering and mount_model_cache_pvc flow to cover both cache modes:
with a model cache PVC, assert container-level HF_HOME is removed,
deployment-level HF_HOME points to the mount path, and the PVC mount is present;
without a PVC, assert TRTLLM’s snapshot HF_HOME remains unchanged.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 98c98466-4bd2-4626-9b3e-6bc21560cac9

📥 Commits

Reviewing files that changed from the base of the PR and between 642ba9c and 5980355.

📒 Files selected for processing (2)
  • .github/workflows/pr.yaml
  • tests/deploy/test_dynamocheckpoint.py

@datadog-official

This comment has been minimized.

@galletas1712
galletas1712 force-pushed the ci/checkpoint-model-cache-pvc branch from 5980355 to 4c11947 Compare July 30, 2026 00:42
@galletas1712

Copy link
Copy Markdown
Contributor Author

/ok to test 4c11947

@galletas1712
galletas1712 enabled auto-merge (squash) July 30, 2026 00:45
@galletas1712

Copy link
Copy Markdown
Contributor Author

/ok to test 4c11947

@galletas1712
galletas1712 force-pushed the ci/checkpoint-model-cache-pvc branch from 4c11947 to 10a25da Compare July 30, 2026 01:52
@galletas1712
galletas1712 requested a review from a team as a code owner July 30, 2026 01:52
@pull-request-size pull-request-size Bot added size/L and removed size/M labels Jul 30, 2026
@galletas1712

Copy link
Copy Markdown
Contributor Author

/ok to test 10a25da

@galletas1712
galletas1712 force-pushed the ci/checkpoint-model-cache-pvc branch from 10a25da to 1884688 Compare July 30, 2026 02:37
@galletas1712
galletas1712 requested review from a team as code owners July 30, 2026 02:37
@galletas1712 galletas1712 changed the title ci(checkpoint): mount shared model-cache PVC in DynamoCheckpoint tests fix(checkpoint): stabilize model cache and vLLM capture Jul 30, 2026
@galletas1712

Copy link
Copy Markdown
Contributor Author

/ok to test 1884688

@github-actions github-actions Bot added fix backend::vllm Relates to the vllm backend and removed ci Issues/PRs that reference CI build/test labels Jul 30, 2026
@galletas1712

Copy link
Copy Markdown
Contributor Author

/ok to test 1884688

@galletas1712
galletas1712 force-pushed the ci/checkpoint-model-cache-pvc branch from 1884688 to 3c7936f Compare July 30, 2026 06:40
@galletas1712 galletas1712 changed the title fix(checkpoint): stabilize model cache and vLLM capture ci(checkpoint): mount shared model-cache PVC in DynamoCheckpoint tests Jul 30, 2026
@galletas1712

Copy link
Copy Markdown
Contributor Author

/ok to test 3c7936f

@github-actions github-actions Bot added ci Issues/PRs that reference CI build/test and removed fix labels Jul 30, 2026

@nv-tusharma nv-tusharma left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

Checkpoint deploy CI was downloading Qwen via ModelExpress/HF on every
run while regular deploy tests already used the Azure model-cache PVC.
Wire the same provision + mount path through, and honor --model-cache-pvc
in the checkpoint test harness (including skipping TRTLLM's container
HF_HOME override so it cannot shadow the shared cache).

Signed-off-by: Schwinn Saereesitthipitak <schwinns@nvidia.com>
@galletas1712
galletas1712 force-pushed the ci/checkpoint-model-cache-pvc branch from 3c7936f to f2fbb78 Compare July 30, 2026 19:13
@copy-pr-bot

copy-pr-bot Bot commented Jul 30, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@galletas1712

Copy link
Copy Markdown
Contributor Author

/ok to test f2fbb78

@galletas1712
galletas1712 merged commit 4bd68c3 into main Jul 30, 2026
112 checks passed
@galletas1712
galletas1712 deleted the ci/checkpoint-model-cache-pvc branch July 30, 2026 20:25
pvijayakrish pushed a commit that referenced this pull request Aug 4, 2026
…ckpoint tests (#12400) (#12593)

Signed-off-by: Schwinn Saereesitthipitak <schwinns@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

actions backend::vllm Relates to the vllm backend ci Issues/PRs that reference CI build/test size/M

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants