pytorch/test
Tristan Rice ffaea656b5 WorkerServer: add support for binding to TCP (#127986)
This adds support for the WorkerServer binding to TCP as well as the existing unix socket support.

```py
server = _WorkerServer("", 1234)
```

Test plan:

Added unit test

```
python test/distributed/elastic/test_control_plane.py
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/127986
Approved by: https://github.com/c-p-i-o
2024-06-05 22:56:32 +00:00
..
ao/sparsity
autograd
backends/xeon
benchmark_utils
bottleneck_test
cpp Remove unused functions (#127881) 2024-06-05 05:21:24 +00:00
cpp_api_parity
cpp_extensions Support torch.dtype as parameter in pybind11 cpp extension. (#126865) 2024-05-29 23:19:32 +00:00
custom_backend [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
custom_operator [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
distributed WorkerServer: add support for binding to TCP (#127986) 2024-06-05 22:56:32 +00:00
distributions [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
dynamo [BE][Eazy] remove torch.torch.xxx usages (#127800) 2024-06-05 21:53:49 +00:00
dynamo_expected_failures [dynamo][numpy] Add unsigned integer dtypes (#125717) 2024-06-05 14:33:47 +00:00
dynamo_skips
edge [BE][CMake] Use FindPython module (#124613) 2024-05-29 13:17:35 +00:00
error_messages
expect Lift jagged -> padded dense forward / backward kernels from fbgemm_gpu (#125946) 2024-06-03 23:41:54 +00:00
export Handle unpacking during TorchScript to ExportedProgram conversion (#127419) 2024-06-05 15:27:13 +00:00
forward_backward_compatibility [SDPA/memeff] Backport changes from xFormers to PT (#127090) 2024-06-05 07:33:27 +00:00
functorch [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
fx Implement Graph Transform Observer (#127427) 2024-06-02 06:49:47 +00:00
higher_order_ops
inductor [inductor] Take absolute value of strides when picking loop order (#127425) 2024-06-05 16:48:22 +00:00
jit [reland] Refresh OpOverloadPacket if a new OpOverload gets added (#128000) 2024-06-05 17:57:09 +00:00
jit_hooks
lazy [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
mobile [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
nn [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
onnx [ONNX] Single function for torch.onnx.export and torch.onnx.dynamo_export (#127974) 2024-06-05 21:27:46 +00:00
onnx_caffe2 [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
optim Quick Fix on #126854, deepcopy lr and other possible base_parameters (#127190) 2024-06-03 18:06:31 +00:00
package [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
profiler [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
quantization [CUDNN] Remove defunct cuDNN V8 API build flag (#120006) 2024-06-03 22:42:05 +00:00
scripts [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_img
torch_np [dynamo] Support ndarray.dtype attribute access (#124490) 2024-06-05 17:20:01 +00:00
typing [1/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort stdlib (#127122) 2024-05-25 08:25:50 +00:00
xpu
_test_bazel.py
allowlist_for_publicAPI.json Revert "Retire torch.distributed.pipeline (#127354)" 2024-06-04 18:19:31 +00:00
conftest.py
create_dummy_torchscript_model.py
delete.py
hi.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
minioptest_failures_dict.json
mkl_verbose.py
mkldnn_verbose.py
pytest_shard_custom.py
run_doctests.sh
run_test.py [CI] disable td for xpu ci test by default (#127611) 2024-06-04 17:15:10 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_autocast.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
test_autograd.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
test_autograd_fallback.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_binary_ufuncs.py [dynamo] Support ndarray.dtype attribute access (#124490) 2024-06-05 17:20:01 +00:00
test_bundled_images.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_bundled_inputs.py
test_ci_sanity_check_fail.py
test_comparison_utils.py
test_compile_benchmark_util.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_complex.py
test_content_store.py
test_cpp_api_parity.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_cpp_extensions_aot.py Support torch.dtype as parameter in pybind11 cpp extension. (#126865) 2024-05-29 23:19:32 +00:00
test_cpp_extensions_jit.py
test_cpp_extensions_mtia_backend.py
test_cpp_extensions_open_device_registration.py Fix map_location for wrapper subclass and device tensors that go through numpy (#126728) 2024-05-24 16:39:30 +00:00
test_cpp_extensions_stream_and_event.py
test_cuda.py [BE][Eazy] remove torch.torch.xxx usages (#127800) 2024-06-05 21:53:49 +00:00
test_cuda_expandable_segments.py
test_cuda_multigpu.py Deprecate device-specific GradScaler autocast API (#126527) 2024-05-25 06:41:34 +00:00
test_cuda_nvml_based_avail.py
test_cuda_primary_ctx.py
test_cuda_sanitizer.py
test_cuda_trace.py
test_custom_ops.py [reland] Refresh OpOverloadPacket if a new OpOverload gets added (#128000) 2024-06-05 17:57:09 +00:00
test_dataloader.py
test_datapipe.py [BE][Ez]: Apply PYI059 - Generic always come last (#127685) 2024-06-02 13:38:58 +00:00
test_decomp.py [decomp] Fix default values missing from inplace rrelu decomposition (#126978) 2024-05-26 23:49:40 +00:00
test_deploy.py
test_determination.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_dispatch.py
test_dlpack.py
test_dynamic_shapes.py Revert "Complete revamp of float/promotion sympy handling (#126905)" 2024-06-05 03:57:58 +00:00
test_expanded_weights.py
test_fake_tensor.py [fake tensor] Set _is_param for base fake tensors for views (#127823) 2024-06-05 20:26:52 +00:00
test_flop_counter.py
test_foreach.py Allow slow foreach to run for any backend, not just CPU (#127412) 2024-06-01 13:58:18 +00:00
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py
test_functionalization.py
test_functionalization_of_rng_ops.py
test_futures.py
test_fx.py
test_fx_experimental.py Revert "Add aten._unsafe_masked_index (#116491)" 2024-06-03 15:51:50 +00:00
test_fx_passes.py
test_fx_reinplace_pass.py
test_hub.py Enable UFMT on test/test_hub.py (#127155) 2024-05-25 18:23:24 +00:00
test_import_stats.py
test_indexing.py [dynamo] wrap GraphModule exceptions in dynamo-wrapped tests (#126341) 2024-05-29 05:18:04 +00:00
test_itt.py
test_jit.py [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
test_jit_autocast.py
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py Enable UFMT on test/test_jit_fuser_te.py (#127759) 2024-06-04 16:56:03 +00:00
test_jit_legacy.py
test_jit_llga_fuser.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_legacy_vmap.py
test_license.py
test_linalg.py [ROCm] TunableOp improvements (#124362) 2024-06-03 22:30:11 +00:00
test_logging.py
test_masked.py
test_maskedtensor.py
test_matmul_cuda.py FP8 rowwise scaling (#125204) 2024-06-05 15:46:40 +00:00
test_meta.py Correct error message for aten::_local_scalar_dense on meta tensor (#124554) 2024-05-30 00:50:29 +00:00
test_metal.py
test_mkl_verbose.py
test_mkldnn.py [dynamo] wrap GraphModule exceptions in dynamo-wrapped tests (#126341) 2024-05-29 05:18:04 +00:00
test_mkldnn_fusion.py
test_mkldnn_verbose.py
test_mobile_optimizer.py [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
test_model_dump.py
test_model_exports_to_core_aten.py
test_module_tracker.py [dynamo] wrap GraphModule exceptions in dynamo-wrapped tests (#126341) 2024-05-29 05:18:04 +00:00
test_modules.py Revert "Default meta device to use swap_tensors in nn.Module._apply (.to_empty and .to('meta')) (#126819)" 2024-06-01 18:46:16 +00:00
test_monitor.py
test_mps.py Restore MPS testing on MacOS 13 and m2 metal (#127853) 2024-06-05 14:44:00 +00:00
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtensor.py
test_namedtuple_return_api.py
test_native_functions.py
test_native_mha.py
test_nestedtensor.py Lift jagged -> padded dense forward / backward kernels from fbgemm_gpu (#125946) 2024-06-03 23:41:54 +00:00
test_nn.py Default XLA to use swap_tensors path in nn.Module._apply (#126814) 2024-06-04 21:40:49 +00:00
test_nnapi.py Enable ufmt format on test files (#126845) 2024-05-28 01:42:07 +00:00
test_numba_integration.py Enable ufmt format on test files (#126845) 2024-05-28 01:42:07 +00:00
test_numpy_interop.py [dynamo][numpy] Add unsigned integer dtypes (#125717) 2024-06-05 14:33:47 +00:00
test_openmp.py Enable ufmt format on test files (#126845) 2024-05-28 01:42:07 +00:00
test_ops.py Add fake impl for aten.unique_dim (#126561) 2024-06-01 04:03:10 +00:00
test_ops_fwd_gradients.py
test_ops_gradients.py
test_ops_jit.py
test_optim.py Revert "[BE] wrap deprecated function/class with typing_extensions.deprecated (#126898)" 2024-05-31 19:47:24 +00:00
test_out_dtype_op.py
test_overrides.py Properly detect nested torch function args (#127496) 2024-06-02 03:43:22 +00:00
test_package.py
test_per_overload_api.py
test_prims.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
test_proxy_tensor.py Revert "Complete revamp of float/promotion sympy handling (#126905)" 2024-06-05 03:57:58 +00:00
test_pruning_op.py
test_public_bindings.py Revert "Retire torch.distributed.pipeline (#127354)" 2024-06-04 18:19:31 +00:00
test_python_dispatch.py
test_pytree.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
test_quantization.py
test_reductions.py Remove calls of deprecated _aminmax (#127182) 2024-05-28 03:51:45 +00:00
test_scatter_gather_ops.py
test_schema_check.py [dynamo] wrap GraphModule exceptions in dynamo-wrapped tests (#126341) 2024-05-29 05:18:04 +00:00
test_segment_reductions.py
test_serialization.py Remove tensor subclass detection logic from weights_only unpickler (#127808) 2024-06-05 02:14:30 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py Enable UFMT on test_shape_ops.py test_show_pickle.py test_sort_and_select.py (#127165) 2024-06-05 14:31:26 +00:00
test_show_pickle.py Enable UFMT on test_shape_ops.py test_show_pickle.py test_sort_and_select.py (#127165) 2024-06-05 14:31:26 +00:00
test_sort_and_select.py Enable UFMT on test_shape_ops.py test_show_pickle.py test_sort_and_select.py (#127165) 2024-06-05 14:31:26 +00:00
test_sparse.py [traced-graph][sparse] propagate sparsity metadata into traced graph (#117907) 2024-05-23 22:46:46 +00:00
test_sparse_csr.py
test_sparse_semi_structured.py
test_spectral_ops.py
test_stateless.py [BE] wrap deprecated function/class with typing_extensions.deprecated (#127689) 2024-06-02 12:30:43 +00:00
test_static_runtime.py [BE] enable ruff rule Q from flake8-quotes (#127713) 2024-06-02 23:25:26 +00:00
test_subclass.py
test_sympy_utils.py Revert "Complete revamp of float/promotion sympy handling (#126905)" 2024-06-05 03:57:58 +00:00
test_tensor_creation_ops.py [dynamo] wrap GraphModule exceptions in dynamo-wrapped tests (#126341) 2024-05-29 05:18:04 +00:00
test_tensorboard.py
test_tensorexpr.py
test_tensorexpr_pybind.py
test_testing.py Revert "Retire torch.distributed.pipeline (#127354)" 2024-06-04 18:19:31 +00:00
test_throughput_benchmark.py
test_torch.py [BE]: Apply PLR1736 fixes (unnecessary index lookup) (#127716) 2024-06-03 17:22:13 +00:00
test_transformers.py
test_type_hints.py
test_type_info.py
test_type_promotion.py [dynamo] wrap GraphModule exceptions in dynamo-wrapped tests (#126341) 2024-05-29 05:18:04 +00:00
test_typing.py
test_unary_ufuncs.py [dynamo] Support ndarray.dtype attribute access (#124490) 2024-06-05 17:20:01 +00:00
test_utils.py
test_view_ops.py [5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126) 2024-05-27 14:49:57 +00:00
test_vulkan.py
test_weak.py
test_xnnpack_integration.py
test_xpu.py