pytorch/torch/testing/_internal
2024-07-18 17:42:05 +00:00
..
codegen
data
distributed [Easy] Fix argument name collision in dispatched functions (#129562) 2024-07-17 14:39:56 +00:00
generated
opinfo Revert "Increase tolerance for tensorsolve tests (#130620)" 2024-07-15 17:35:04 +00:00
optests Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)" 2024-06-10 20:44:42 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py [ROCm] Add int4 support (#129710) 2024-07-09 19:49:12 +00:00
common_device_type.py The device_suffix in a test_name is "privateuse1" sometimes. (#130091) 2024-07-12 02:51:40 +00:00
common_dist_composable.py
common_distributed.py Revert "Use inductor TestCase for test_replicate_with_compiler.py (#129494)" 2024-07-18 17:42:05 +00:00
common_dtype.py Enable UFMT on common_device_type.py and common_dtype.py (#128490) 2024-06-15 00:07:42 +00:00
common_fsdp.py [FSDP2] Added APIs for explicit fwd/bwd prefetching (#128884) 2024-06-18 13:32:57 +00:00
common_jit.py
common_methods_invocations.py [TEST] Fix _scaled_mm tests (#130897) 2024-07-18 02:15:00 +00:00
common_mkldnn.py
common_modules.py
common_nn.py [9/N] Replace c10::optional with std::optional (#130674) 2024-07-15 00:48:43 +00:00
common_optimizers.py Add testing regarding SparseAdam state_dicts (#130645) 2024-07-16 11:29:22 +00:00
common_pruning.py
common_quantization.py Change BN to eval before QAT Convert phase (#130598) 2024-07-12 16:03:56 +00:00
common_quantized.py
common_subclass.py
common_utils.py Support for expandable segments with cuda graph trees (#128068) 2024-07-15 23:23:23 +00:00
composite_compliance.py
custom_op_db.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
custom_tensor.py Nested tensor subclass support (#127431) 2024-06-26 04:45:22 +00:00
dist_utils.py
dynamo_test_failures.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
hop_db.py Removing some cruff and updating signatures for consistency (#130871) 2024-07-18 13:32:11 +00:00
hypothesis_utils.py
inductor_utils.py Learn a heuristic to decide whether to pad before mm (#128643) 2024-07-15 23:04:06 +00:00
jit_metaprogramming_utils.py
jit_utils.py
logging_tensor.py
logging_utils.py
quantization_torch_package_models.py
static_module.py Flip default value for mypy disallow_untyped_defs [9/11] (#127846) 2024-06-08 18:50:06 +00:00
torchbind_impls.py Allow kwargs in _remove_effect_tokens_pass (#130491) 2024-07-11 19:03:19 +00:00
triton_utils.py [Inductor Intel GPU backend Upstream] Reuse inductor test for Intel GPU (PART 2) (#124147) 2024-06-16 08:07:05 +00:00
two_tensor.py Add recursive metadata guard test (#131002) 2024-07-18 08:24:43 +00:00