-
Notifications
You must be signed in to change notification settings - Fork 62
Migrate test_utils, test_schema_check for xpu and add checkpoint.py f… #2259
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull Request Overview
This PR migrates test utilities and schema checking functionality to support Intel XPU (GPU) devices by porting existing CUDA-specific test files. The migration follows a pattern of making device-agnostic changes while preserving original code structure.
- Creates device-agnostic test infrastructure for Intel XPU support
- Introduces a checkpoint utility module for XPU test environments
- Enables device type detection using accelerator helpers instead of hardcoded CUDA references
Reviewed Changes
Copilot reviewed 3 out of 3 changed files in this pull request and generated 9 comments.
| File | Description |
|---|---|
| test/xpu/test_utils.py | Migrates utility tests from CUDA to device-agnostic approach using HAS_GPU and device_type variables |
| test/xpu/test_schema_check.py | Ports schema validation tests to XPU with device type instantiation for cpu/cuda/xpu |
| test/xpu/checkpoint.py | Adds comprehensive checkpoint functionality supporting XPU device detection and state management |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
Description
Fixes #114850, we will port test utils and schema check to Intel GPU
We could enable Intel GPU with following methods and try the best to keep the original code styles:
Changes