pytorch/torch/testing/_internal
Wei Wang d7e275d4b4 [CI][CUDA] Add periodic b200 distributed job (#159323)
1. Run distributed job with B200 runner, periodically.
2. discovered generic distributed test issue that certain unit test hard-coded ranks, calling for require_exact_world_size(world_size) API instead of require_world_size(world_size).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159323
Approved by: https://github.com/eqy

Co-authored-by: Aidyn-A <aidyn.b.aitzhan@gmail.com>
2025-10-16 21:54:04 +00:00
..
codegen
data
distributed [CI][CUDA] Add periodic b200 distributed job (#159323) 2025-10-16 21:54:04 +00:00
generated
opinfo [1/N] Use "is" in python type comparison (#165037) 2025-10-10 12:36:50 +00:00
optests add the option to disable functionalization in AOTDispatcher (#164577) 2025-10-16 15:44:11 +00:00
test_module
__init__.py
autocast_test_lists.py [2/N] Use "is" in python type comparison (#165142) 2025-10-10 15:36:44 +00:00
autograd_function_db.py
check_kernel_launches.py
common_cuda.py MX: Remove redundant PLATFORM_SUPPORTS_MX_GEMM constant (#164320) 2025-10-02 23:30:56 +00:00
common_device_type.py Fix IValue from SymBool on big-endian system (#163647) 2025-10-14 15:07:48 +00:00
common_dist_composable.py
common_distributed.py [1/N] Use "is" in python type comparison (#165037) 2025-10-10 12:36:50 +00:00
common_dtype.py
common_fsdp.py [FSDP2] provide public API to share cuda streams across roots (#165024) 2025-10-14 17:50:46 +00:00
common_jit.py
common_methods_invocations.py [2/N] More ruff SIM fixes (#165031) 2025-10-14 14:22:54 +00:00
common_mkldnn.py
common_modules.py Fix the shape check inside gnll loss (#147522) 2025-09-30 18:40:15 +00:00
common_mps.py [MPS] Add backward pass for embedding_bag (#163931) 2025-10-03 00:48:38 +00:00
common_nn.py More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
common_optimizers.py Make Adam, AdamW work with nonzero-dim Tensor betas (#149939) 2025-10-06 22:03:25 +00:00
common_pruning.py
common_quantization.py [1/N] Use "is" in python type comparison (#165037) 2025-10-10 12:36:50 +00:00
common_quantized.py MXFP8 grouped GEMM support for torch._scaled_grouped_mm + submodule bump (#162209) 2025-09-06 15:25:30 +00:00
common_subclass.py
common_utils.py [AOTI] skip Windows XPU crashed UTs. (#165393) 2025-10-14 23:45:14 +00:00
composite_compliance.py [6/N] Apply ruff UP035 rule (#164438) 2025-10-03 00:15:32 +00:00
custom_op_db.py
custom_tensor.py
dist_utils.py
dynamo_test_failures.py
fake_config_module.py
fake_config_module2.py
fake_config_module3.py
hop_db.py [dynamo][hop] Introduce Local Map HOP (#161458) 2025-09-17 09:32:38 +00:00
hypothesis_utils.py [Tools] Adapting the Hypothesis library (version 5.x) for use with the PyTorch framework (#163748) 2025-09-25 16:41:50 +00:00
inductor_utils.py [AOTI-FX] Support registering custom FX backends (#162317) 2025-09-06 07:32:03 +00:00
jit_metaprogramming_utils.py
jit_utils.py
logging_tensor.py
logging_utils.py [6/N] Apply ruff UP035 rule (#164438) 2025-10-03 00:15:32 +00:00
quantization_torch_package_models.py
static_module.py
subclasses.py [1/N] Use "is" in python type comparison (#165037) 2025-10-10 12:36:50 +00:00
torchbind_impls.py
triton_utils.py [AMD] [Reland] Fix AMD User Defined Kernel Autotune (#161521) 2025-09-04 08:41:18 +00:00
two_tensor.py Add view support for library custom Function (#164520) 2025-10-09 16:17:48 +00:00