Bump torch version to 2.7.1 (#1562)
### What this PR does / why we need it?
Bump torch version to 2.7.1, and cleanup infer schema patch
https://github.com/vllm-project/vllm-ascend/commit/857f489
(https://github.com/vllm-project/vllm-ascend/pull/837), this patch
depends on also: https://github.com/vllm-project/vllm-ascend/pull/1974
### Does this PR introduce any user-facing change?
No
#### How was this patch tested?
CI passed
torch-npu 2.7.1rc1 install guide:
https://gitee.com/ascend/pytorch/tree/v2.7.1/
install depending:
```
pip3 install pyyaml
pip3 install setuptools
```
install torch-npu:
Closes: https://github.com/vllm-project/vllm-ascend/issues/1866
Closes: https://github.com/vllm-project/vllm-ascend/issues/1390
- vLLM version: v0.10.0
- vLLM main:
9af654cc38
---------
Signed-off-by: Yikun Jiang <yikunkero@gmail.com>
Signed-off-by: leo-pony <nengjunma@outlook.com>
Co-authored-by: Yikun Jiang <yikunkero@gmail.com>
This commit is contained in:
@@ -543,13 +543,9 @@ class TestNPUPlatform(TestBase):
|
||||
|
||||
@patch("torch.distributed.is_hccl_available", return_value=True)
|
||||
@patch("torch_npu._C._distributed_c10d.ProcessGroupHCCL")
|
||||
@patch("torch_npu._C._distributed_c10d.ProcessGroupHCCL.Options")
|
||||
@patch("torch.distributed.ProcessGroup")
|
||||
def test_successful_initialization(self, mock_pg, mock_options_cls,
|
||||
mock_pg_hccl, _):
|
||||
def test_successful_initialization(self, mock_pg, mock_pg_hccl, _):
|
||||
mock_prefix = MagicMock(spec=PrefixStore)
|
||||
mock_options = MagicMock(spec=ProcessGroup.Options)
|
||||
mock_options_cls.return_value = mock_options
|
||||
mock_backend = MagicMock()
|
||||
mock_pg_hccl.return_value = mock_backend
|
||||
group_rank = 0
|
||||
@@ -574,8 +570,7 @@ class TestNPUPlatform(TestBase):
|
||||
timeout=timedelta(seconds=30),
|
||||
)
|
||||
|
||||
mock_pg.assert_called_once_with(mock_prefix, group_rank, group_size,
|
||||
unittest.mock.ANY)
|
||||
mock_pg.assert_called_once_with(mock_prefix, group_rank, group_size)
|
||||
mock_pg_hccl.assert_called_once_with(mock_prefix, group_rank,
|
||||
group_size, unittest.mock.ANY)
|
||||
mock_backend._set_sequence_number_for_group.assert_called_once()
|
||||
|
||||
Reference in New Issue
Block a user