Skip to content

[ci] Update publish workflow and dependency ranges #60

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.

Sign up for GitHub

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

Merged
KaisennHu merged 3 commits into from
May 30, 2026
Merged

[ci] Update publish workflow and dependency ranges #60

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
0c43b9b
Update publish workflow and dependency ranges
May 30, 2026
b8b57ff
update "gpu object" in error message to "tensor object", since it can…
May 30, 2026
789514f
fix typo
May 30, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 5 additions & 13 deletions .github/workflows/publish.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,6 @@ on:
push:
tags:
- "v*"
workflow_dispatch:
inputs:
version:
description: 'Version to publish (without v prefix, e.g. 0.1.0)'
required: true
type: string

jobs:
build-package:
Expand All @@ -32,13 +26,8 @@ jobs:

- name: Update version
run: |
if [ "${{ github.event_name }}" = "workflow_dispatch" ]; then
echo "Using manual version: ${{ inputs.version }}"
python${{ matrix.python-version }} ci/update_version.py ${{ inputs.version }}
else
echo "Using tag version: ${GITHUB_REF#refs/tags/v}"
python${{ matrix.python-version }} ci/update_version.py
fi
echo "Using tag version: ${GITHUB_REF#refs/tags/v}"
python${{ matrix.python-version }} ci/update_version.py

- name: Build package for Python ${{ matrix.python-version }}
run: |
Expand Down Expand Up @@ -68,10 +57,12 @@ jobs:
with:
pattern: python-packages-*
path: dist/
merge-multiple: true

- name: Create GitHub Release
uses: softprops/action-gh-release@a06a81a03ee405af7f2048a818ed3f03bbf83c7b # v2.5.0
with:
tag_name: ${{ github.ref_name }}
files: dist/*.whl
generate_release_notes: true
env:
Expand All @@ -89,6 +80,7 @@ jobs:
with:
pattern: python-packages-*
path: dist/
merge-multiple: true

- name: Publish to PyPI
uses: pypa/gh-action-pypi-publish@release/v1
Expand Down
7 changes: 4 additions & 3 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,10 @@ For performance benchmarks, see the [Performance Benchmark Report](./docs/develo
- **OS Kernel**: Linux
- **Python Dependencies**:
- python >= 3.10, <= 3.11
- CANN == 8.2.rc1
- torch == 2.7.1, torch-npu == 2.7.1.post1
- Ray (same version as ray-ascend)
- CANN >= 8.2.rc1
- torch >= 2.7.1; torch-npu >= 2.7.1.post2
- torch and torch-npu versions must be compatible with each other.
- ray >= 2.55.0

## Version

Expand Down
6 changes: 4 additions & 2 deletions docs/user_guide/index.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,10 @@ of ray-ascend's key features:

Optional dependencies for specific features:

- **CANN == 8.2.rc1**: Required for NPU features (HCCL, NPU tensor transport)
- **torch == 2.7.1, torch-npu == 2.7.1.post1**: Required for PyTorch NPU support
- **CANN >= 8.2.rc1**: Required for NPU features (HCCL, NPU tensor transport)
- **torch >= 2.7.1; torch-npu >= 2.7.1.post2**: Required for PyTorch NPU support
- **torch and torch-npu version compatibility**: Use matching torch and torch-npu
versions.

## Quick Start

Expand Down
12 changes: 6 additions & 6 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,18 +32,18 @@ version = {attr = "ray_ascend.__version__"}

[project.optional-dependencies]
test = [
"torch==2.7.1+cpu; platform_machine == 'x86_64'",
"torch==2.7.1; platform_machine == 'aarch64'",
"torch-npu==2.7.1.post2",
"torch>=2.7.1; platform_machine == 'x86_64'",
"torch>=2.7.1; platform_machine == 'aarch64'",
"torch-npu>=2.7.1.post2",
Comment on lines +35 to +37
Copy link
Copy Markdown
Collaborator

@dpj135 dpj135 May 30, 2026

Choose a reason for hiding this comment

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

we better add notes about the versions of torch and torch-npu to user_guide and developer_guide if we cannot make sure the versions of torch and torch-npu are aligned when pip install ray-ascend.

"pytest>=7.0.0",
"pytest-cov",
]

yr = [
"openyuanrong-datasystem>=0.8.0",
"torch==2.7.1+cpu; platform_machine == 'x86_64'",
"torch==2.7.1; platform_machine == 'aarch64'",
"torch-npu==2.7.1.post2",
"torch>=2.7.1; platform_machine == 'x86_64'",
"torch>=2.7.1; platform_machine == 'aarch64'",
"torch-npu>=2.7.1.post2",
"requests",
]

Expand Down
6 changes: 3 additions & 3 deletions ray_ascend/direct_transport/yr_tensor_transport.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ class YRCommunicatorMetadata(CommunicatorMetadata):

@dataclass
class YRTransportMetadata(TensorTransportMetadata):
"""Metadata for tensors stored in the GPU object store for YR transport.
"""Metadata for tensors stored in the CPU/NPU object store for YR transport.
Args:
ds_serialized_keys: Serialized tensor keys for YR transport.
"""
Expand Down Expand Up @@ -158,9 +158,9 @@ def extract_tensor_transport_metadata(

tensor_meta = []
if not gpu_object:
raise ValueError("GPU object list is empty.")
raise ValueError("Tensor object list is empty.")
serialized_keys = self.get_ds_metadata(gpu_object)
# We assume all tensors in one GPU object have the same device type.
# We assume all tensors in one NPU/CPU object have the same device type.
device = gpu_object[0].device
for t in gpu_object:
if t.device.type != device.type:
Expand Down
Loading