|
34 | 34 | 'test_cuda_primary_ctx',
|
35 | 35 | 'test_dataloader',
|
36 | 36 | 'distributed/test_data_parallel',
|
37 |
| - 'distributed/test_distributed', |
| 37 | + 'distributed/test_distributed_fork', |
38 | 38 | 'distributed/test_distributed_spawn',
|
39 | 39 | 'test_distributions',
|
40 | 40 | 'test_expecttest',
|
|
96 | 96 | 'distributed/rpc/test_faulty_agent',
|
97 | 97 | 'distributed/rpc/test_process_group_agent',
|
98 | 98 | 'distributed/rpc/test_tensorpipe_agent',
|
99 |
| - 'distributed/test_distributed', |
| 99 | + 'distributed/test_distributed_fork', |
100 | 100 | 'distributed/test_distributed_spawn',
|
101 | 101 | ]
|
102 | 102 |
|
|
140 | 140 | 'test_jit_profiling',
|
141 | 141 | 'test_torch',
|
142 | 142 | 'distributed/nn/jit/test_instantiator',
|
143 |
| - 'distributed/test_distributed', |
| 143 | + 'distributed/test_distributed_fork', |
144 | 144 | 'distributed/rpc/test_process_group_agent',
|
145 | 145 | 'distributed/rpc/test_tensorpipe_agent',
|
146 | 146 | 'distributed/algorithms/ddp_comm_hooks/test_ddp_hooks',
|
@@ -319,7 +319,7 @@ def test_distributed(test_module, test_directory, options):
|
319 | 319 | os.environ['INIT_METHOD'] = 'env://'
|
320 | 320 | os.environ.update(env_vars)
|
321 | 321 | if with_init_file:
|
322 |
| - if test_module in ["test_distributed", "test_distributed_spawn"]: |
| 322 | + if test_module in ["test_distributed_fork", "test_distributed_spawn"]: |
323 | 323 | init_method = 'file://{}/'.format(tmp_dir)
|
324 | 324 | else:
|
325 | 325 | init_method = 'file://{}/shared_init_file'.format(tmp_dir)
|
@@ -351,7 +351,7 @@ def test_distributed(test_module, test_directory, options):
|
351 | 351 | 'test_cuda_primary_ctx': test_cuda_primary_ctx,
|
352 | 352 | 'test_cpp_extensions_aot_no_ninja': test_cpp_extensions_aot_no_ninja,
|
353 | 353 | 'test_cpp_extensions_aot_ninja': test_cpp_extensions_aot_ninja,
|
354 |
| - 'distributed/test_distributed': test_distributed, |
| 354 | + 'distributed/test_distributed_fork': test_distributed, |
355 | 355 | 'distributed/test_distributed_spawn': test_distributed,
|
356 | 356 | }
|
357 | 357 |
|
|
0 commit comments