已合并
[test]fix _test_alltoall_2p_size test_mgr.load #32520
AACAES创建于 3月27日
[test]fix _test_alltoall_2p_size test_mgr.load #32520
已合并
共 2 个文件变更+1-2
| @@ -34,8 +34,6 @@ class TestMgr: | |||
| 34 | def load(self, modify_files, world_size): | 34 | def load(self, modify_files, world_size): |
| 35 | with open(modify_files) as f: | 35 | with open(modify_files) as f: |
| 36 | for line in f: | 36 | for line in f: |
| 37 | - if world_size != 0 and ("test/distributed/" in line or "test/_inductor/" in line): | ||
| 38 | - continue | ||
| 39 | line = line.strip() | 37 | line = line.strip() |
| 40 | self.modify_files.append(line) | 38 | self.modify_files.append(line) |
| 41 | 39 | ||
| @@ -37,6 +37,7 @@ class HcclAlltoAllTest(TestCase): | |||
| 37 | 37 | ||
| 38 | 38 | ||
| 39 | def _test_alltoall_2p_size(cls, rank, world_size, init_pg, c2p, p2c): | 39 | def _test_alltoall_2p_size(cls, rank, world_size, init_pg, c2p, p2c): |
| 40 | + torch_npu.npu.config.allow_internal_format = True | ||
| 40 | pg = init_pg(rank, world_size) | 41 | pg = init_pg(rank, world_size) |
| 41 | input_list = [(torch.zeros(rank + 1, 1) + rank).float().npu() for i in range(2)] | 42 | input_list = [(torch.zeros(rank + 1, 1) + rank).float().npu() for i in range(2)] |
| 42 | output_list = [torch.empty(i + 1, 1).float().npu() for i in range(2)] | 43 | output_list = [torch.empty(i + 1, 1).float().npu() for i in range(2)] |