pytorch/torch/testing/_internal
dilililiwhy 6e5dddba64 Use accelerator API in common_dtensor (#163498)
Fixes #ISSUE_NUMBER

Try to unify the device checking in common_dtensor (testing module) by accelerator API

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163498
Approved by: https://github.com/albanD, https://github.com/H-Huang
2025-09-23 16:30:20 +00:00
..
codegen
data
distributed Use accelerator API in common_dtensor (#163498) 2025-09-23 16:30:20 +00:00
generated
opinfo
optests [doc]: Small typos (#162982) 2025-09-16 17:42:19 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py [ROCm] Remove HIPBLASLT_ALLOW_TF32 from codebase (#162998) 2025-09-18 13:53:48 +00:00
common_device_type.py [BE] Improve pytest summary display for OpInfo tests (#162961) 2025-09-15 19:58:19 +00:00
common_dist_composable.py
common_distributed.py Use functools.cache on has_efa (#163439) 2025-09-23 05:03:03 +00:00
common_dtype.py
common_fsdp.py
common_jit.py
common_methods_invocations.py Bugfix for doing negative padding (#161639) 2025-09-19 20:57:05 +00:00
common_mkldnn.py
common_modules.py [BE] Delete skipIfMPSOnMacOS13 (#163515) 2025-09-22 21:10:22 +00:00
common_mps.py [MPS] Add embedding_bag forward pass (#163012) 2025-09-17 19:00:47 +00:00
common_nn.py
common_optimizers.py remove tolerance override for dynamo test_mixed_device_dtype in SGD (#163088) 2025-09-17 18:17:23 +00:00
common_pruning.py
common_quantization.py Replace export_for_training with export (#162396) 2025-09-10 14:19:34 +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 [BE] Delete skipIfMPSOnMacOS13 (#163515) 2025-09-22 21:10:22 +00:00
composite_compliance.py
custom_op_db.py
custom_tensor.py
dist_utils.py
dynamo_test_failures.py type misc init and tools for dynamo (#161293) 2025-08-26 17:38:49 +00:00
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
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
quantization_torch_package_models.py
static_module.py
subclasses.py
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