pytorch/torch/testing/_internal
Tristan Rice df4e5294a6 Reapply "ProcessGroupGloo: support lazy_init (#150801)" (#151031)
This reverts commit 73f3d6d9aa.

Reapplies #150801

Test plan:

See #150801

submodule

Pull Request resolved: https://github.com/pytorch/pytorch/pull/151031
Approved by: https://github.com/fduwjj
2025-04-11 01:58:35 +00:00
..
codegen
data
distributed [Reland] Launch kernel on current stream & remove record_stream entirely (#150398) 2025-04-01 16:46:07 +00:00
generated
opinfo Avoid overflow in vector_norm for scalar input (#144073) 2025-04-07 17:10:10 +00:00
optests [opcheck] Improve error reporting; allow atol/rtol overrides (#146488) 2025-02-05 21:25:06 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_cuda.py [BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257) 2025-03-18 00:46:07 +00:00
common_device_type.py Revert "Add torch._scaled_mm for CPU (#150410)" 2025-04-04 06:52:54 +00:00
common_dist_composable.py Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
common_distributed.py Reapply "ProcessGroupGloo: support lazy_init (#150801)" (#151031) 2025-04-11 01:58:35 +00:00
common_dtype.py [BE]: Enable ruff SLOT checks (#146276) 2025-02-04 19:18:23 +00:00
common_fsdp.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
common_jit.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_methods_invocations.py Support negative values for fill with uint tensors (#144458) 2025-04-09 21:08:06 +00:00
common_mkldnn.py [BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257) 2025-03-18 00:46:07 +00:00
common_modules.py Remove outdated skipCUDAIfCudnnVersionLessThan decoration (#148940) 2025-03-13 18:02:50 +00:00
common_nn.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_optimizers.py Remove code for Python < 3.9 (#147097) 2025-02-14 03:22:49 +00:00
common_pruning.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_quantization.py [Codemod][AddExplicitStrictExportForTrainingInferenceArg] caffe2/ (#149595) 2025-04-03 23:50:13 +00:00
common_quantized.py wire torch._scaled_mm with fp4 operands to the cublas nvfp4 kernel (#148792) 2025-03-27 17:32:20 +00:00
common_subclass.py
common_utils.py Code Clean: Using the new builtin function provides by python 3.8 later (#150839) 2025-04-10 01:17:39 +00:00
composite_compliance.py [pytree] add APIs to determine a class is a namedtuple or PyStructSequence (#113257) 2025-04-01 10:40:43 +00:00
custom_op_db.py Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
custom_tensor.py
dist_utils.py Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
dynamo_test_failures.py [BE] Remove test_ops from FIXME_inductor_dont_reset_dynamo (#145307) 2025-01-27 18:12:39 +00:00
fake_config_module.py config: Support str env variables (#145980) 2025-01-30 00:13:02 +00:00
fake_config_module2.py Add multi env variable support to configs (#145288) 2025-01-24 10:04:24 +00:00
fake_config_module3.py [minifier] Fix config generator for callables (#144518) 2025-01-14 17:18:13 +00:00
hop_db.py Support torch.compile rng selective activation checkpointing with cudagraph (#146878) 2025-02-28 00:47:03 +00:00
hypothesis_utils.py
inductor_utils.py cpp_wrapper: Fix even more tests (#147225) 2025-04-07 14:20:06 +00:00
jit_metaprogramming_utils.py [BE]: Enable ruff SLOT checks (#146276) 2025-02-04 19:18:23 +00:00
jit_utils.py Fix linter F821 error (#146665) 2025-02-08 07:19:37 +00:00
logging_tensor.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
logging_utils.py Update ruff linter for PEP585 (#147540) 2025-02-22 04:45:17 +00:00
quantization_torch_package_models.py
static_module.py
subclasses.py [subclass] testing WrapperSubclass respect outer_size, outer_stride (#146897) 2025-02-13 15:21:19 +00:00
torchbind_impls.py Fakify torchbind objects in compile_fx and add tests for SigridTransformsInstanceTorchBind (#149529) 2025-03-21 18:58:28 +00:00
triton_utils.py [AOTInductor] Fix autotuning code's codegen (#150522) 2025-04-03 00:08:19 +00:00
two_tensor.py Support subclass constructor capturing in export (#147014) 2025-03-16 18:19:19 +00:00