已合并
skip test_schedule_multiproc #40664
我应该是一阵风创建于 7月10日
skip test_schedule_multiproc #40664
已合并
共 1 个文件变更+2-3
| @@ -5,7 +5,7 @@ import logging | |||
| 5 | import os | 5 | import os |
| 6 | import sys | 6 | import sys |
| 7 | import tempfile | 7 | import tempfile |
| 8 | - | 8 | +import unittest |
| 9 | from model_registry import ModelWithKwargs, MultiMLP, MultiMLPWithDw | 9 | from model_registry import ModelWithKwargs, MultiMLP, MultiMLPWithDw |
| 10 | from schedule_registry import ( | 10 | from schedule_registry import ( |
| 11 | ScheduleUnbalanced, | 11 | ScheduleUnbalanced, |
| @@ -49,7 +49,7 @@ device_type = "npu" | |||
| 49 | 49 | ||
| 50 | torch.manual_seed(0) | 50 | torch.manual_seed(0) |
| 51 | 51 | ||
| 52 | - | 52 | +@unittest.skip("Skip: test not adapted") |
| 53 | class ScheduleTest(MultiProcContinuousTest): | 53 | class ScheduleTest(MultiProcContinuousTest): |
| 54 | world_size = int(os.getenv("WORLD_SIZE", 2)) | 54 | world_size = int(os.getenv("WORLD_SIZE", 2)) |
| 55 | 55 | ||
| @@ -67,7 +67,6 @@ class ScheduleTest(MultiProcContinuousTest): | |||
| 67 | super().setUpClass() | 67 | super().setUpClass() |
| 68 | dev_id = cls.rank % torch.npu.device_count() | 68 | dev_id = cls.rank % torch.npu.device_count() |
| 69 | cls.device = torch.device(f"npu:{dev_id}") | 69 | cls.device = torch.device(f"npu:{dev_id}") |
| 70 | - | ||
| 71 | 70 | ||
| 72 | def device(self) -> torch.device: | 71 | def device(self) -> torch.device: |
| 73 | return torch.device(device_type, self.rank) | 72 | return torch.device(device_type, self.rank) |