Skip to content

Validate checkpoint GPU UUID inputs early#2086

Open
aryanputta wants to merge 1 commit into
NVIDIA:mainfrom
aryanputta:fix-checkpoint-gpu-uuid-validation
Open

Validate checkpoint GPU UUID inputs early#2086
aryanputta wants to merge 1 commit into
NVIDIA:mainfrom
aryanputta:fix-checkpoint-gpu-uuid-validation

Conversation

@aryanputta
Copy link
Copy Markdown

Summary

  • reject non-UUID gpu_mapping values in cuda.core.checkpoint before they reach driver-facing restore args
  • normalize malformed UUID string failures to the documented ValueError path
  • add source-level checkpoint helper tests that run without importing the built cuda.core package

Testing

  • python3 -m pytest cuda_core/tests/test_checkpoint_helpers.py --noconftest

Signed-off-by: Aryan <aryansputta@gmail.com>
@copy-pr-bot
Copy link
Copy Markdown
Contributor

copy-pr-bot Bot commented May 14, 2026

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.

@github-actions github-actions Bot added the cuda.core Everything related to the cuda.core module label May 14, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cuda.core Everything related to the cuda.core module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant