pytorch/torch/testing/_internal
2024-07-24 18:22:30 +00:00
..
codegen
data
distributed
generated
opinfo Add hooks for execution on intel gaudi devices - 1 (#128584) 2024-07-20 05:03:36 +00:00
optests
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py
common_device_type.py Add hooks for execution on intel gaudi devices - 1 (#128584) 2024-07-20 05:03:36 +00:00
common_dist_composable.py
common_distributed.py Use fake PG for test_compute_comm_reordering.py unit tests (#131415) 2024-07-23 22:53:23 +00:00
common_dtype.py
common_fsdp.py
common_jit.py
common_methods_invocations.py Add decomposition for t_copy (#130939) 2024-07-23 08:29:19 +00:00
common_mkldnn.py
common_modules.py Add hooks for execution on intel gaudi devices - 1 (#128584) 2024-07-20 05:03:36 +00:00
common_nn.py
common_optimizers.py [Optim] Support tensor lr for all optimizers and check it is 1-element (#131065) 2024-07-23 04:27:05 +00:00
common_pruning.py
common_quantization.py Set correct output dtype for dequantize op during convert_pt2e in decomposed mode (#128953) 2024-07-19 04:58:02 +00:00
common_quantized.py
common_subclass.py
common_utils.py [3.13] support 3.13 multiline traces in munge_exc (#131207) 2024-07-24 18:22:30 +00:00
composite_compliance.py
custom_op_db.py [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
custom_tensor.py
dist_utils.py
dynamo_test_failures.py
hop_db.py [Autograd] Cond Higher-Order Operation (#126911) 2024-07-22 23:18:19 +00:00
hypothesis_utils.py
inductor_utils.py
jit_metaprogramming_utils.py
jit_utils.py
logging_tensor.py
logging_utils.py
quantization_torch_package_models.py
static_module.py
torchbind_impls.py
triton_utils.py Add flag to ignore unsupported @triton.autotune args in user-written kernel compilation (#131431) 2024-07-24 05:37:09 +00:00
two_tensor.py