Skip to content

[rayorch] integrate RayOrch for parallel data processing with new operators and storage classes#501

Merged
SunnyHaze merged 5 commits intomainfrom
dev/rayorch
Apr 9, 2026
Merged

[rayorch] integrate RayOrch for parallel data processing with new operators and storage classes#501
SunnyHaze merged 5 commits intomainfrom
dev/rayorch

Conversation

@SunnyHaze
Copy link
Copy Markdown
Collaborator

  • Added RayAcceleratedOperator for transparent data-parallel execution of DataFlow operators.
  • Introduced InMemoryStorage for efficient in-memory data handling within Ray actors.
  • Created dummy operators for testing and validation of the Ray integration.
  • Updated requirements.txt to include rayorch dependency.
  • Added comprehensive tests for both serial and parallel execution scenarios.

This implementation enhances the DataFlow framework by enabling efficient multi-GPU processing without modifying existing operators.

…rators and storage classes

- Added `RayAcceleratedOperator` for transparent data-parallel execution of DataFlow operators.
- Introduced `InMemoryStorage` for efficient in-memory data handling within Ray actors.
- Created dummy operators for testing and validation of the Ray integration.
- Updated `requirements.txt` to include `rayorch` dependency.
- Added comprehensive tests for both serial and parallel execution scenarios.

This implementation enhances the DataFlow framework by enabling efficient multi-GPU processing without modifying existing operators.
@SunnyHaze
Copy link
Copy Markdown
Collaborator Author

@haolpku 先不要合并,这个还是等试用期各自体验下,再进主分支。

- Added logic to automatically release actor resources after each stage in PipelineABC, BatchedPipelineABC, and StreamBatchedPipelineABC.
- Updated documentation in README files to clarify the necessity of calling shutdown() for multiple RayAcceleratedOperators.
- Introduced tests to verify the auto-shutdown functionality across different pipeline types.

This enhancement ensures efficient resource management and prevents GPU resource blocking during pipeline execution.
@SunnyHaze SunnyHaze merged commit 19a408a into main Apr 9, 2026
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant