pytorch/test
2022-07-21 19:48:56 +00:00
..
ao/sparsity [ao][sparsity] Vectorized WeightNormSparsifier (#80059) 2022-07-12 19:16:44 +00:00
autograd
benchmark_utils
bottleneck_test
cpp Lower randint default dtype to the C++ API (#81410) 2022-07-21 16:42:49 +00:00
cpp_api_parity
cpp_extensions add open device registration test with cpp extensions (#80477) 2022-07-12 01:46:16 +00:00
custom_backend
custom_operator
distributed Add cpu/gloo tests for sharded tensor distributed checkpoint (#80997) 2022-07-20 23:02:52 +00:00
distributions Setting validation flag for Distributions tests to work with TorchDynamo (#80081) 2022-06-24 04:51:51 +00:00
error_messages
expect [Reland] Recursively print graph module and its submodule (#81639) 2022-07-21 16:58:25 +00:00
forward_backward_compatibility Adding docs on how to run backwards compatability test (#81431) 2022-07-14 21:08:56 +00:00
fx bmm and contiguous constraints (#81527) 2022-07-21 19:21:50 +00:00
jit [JIT] Modify SchemaCheckMode to utilize SchemaInfo and fix training op tests accordingly (#81783) 2022-07-20 22:09:56 +00:00
jit_hooks
lazy functionalization <> LTC integration (take 3) (#80251) 2022-06-26 23:10:21 +00:00
mobile [retake2][mobile] Fix lightweight dispatch OOM error by introducing selective build (#80791) 2022-07-15 18:04:25 +00:00
onnx Apply ufmt linter to all py files under test/onnx (#81335) 2022-07-15 18:51:38 +00:00
package Allow torch._C to be recognized a module in torch.package (#80917) 2022-07-12 18:10:22 +00:00
quantization Add native impl for group norm on quantized CPU for channels-last inputs (#70520) 2022-07-21 00:37:25 +00:00
scripts
test_img
typing
allowlist_for_publicAPI.json Modules under migration in the public binding test (#81314) 2022-07-12 20:49:04 +00:00
conftest.py pytest to run test_ops, test_ops_gradients, test_ops_jit in non linux cuda environments (#79898) 2022-07-19 19:50:57 +00:00
create_dummy_torchscript_model.py
defs.bzl
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
load_torchscript_model.py
profiler_utils_mock_events.json [Profiler] Add heuristic to rank events base on computed metrics (#80094) 2022-07-05 22:06:02 +00:00
run_test.py [test stats] use published test stats for sharding (#81116) 2022-07-12 04:50:19 +00:00
simulate_nccl_errors.py
test_ao_sparsity.py [ao][sparsity] removing leading '.' from fqn in utils (#79774) 2022-06-30 00:00:56 +00:00
test_autocast.py
test_autograd.py [forward ad] Fix codegen to ignore undefined outputs (#81114) 2022-07-11 15:01:39 +00:00
test_binary_ufuncs.py
test_bundled_images.py
test_bundled_inputs.py
test_complex.py
test_cpp_api_parity.py
test_cpp_extensions_aot.py
test_cpp_extensions_jit.py
test_cpp_extensions_open_device_registration.py add open device registration test with cpp extensions (#80477) 2022-07-12 01:46:16 +00:00
test_cuda.py Revert "[CUDA graphs] Clear autocast amp cache (#81558)" 2022-07-21 12:46:36 +00:00
test_cuda_primary_ctx.py
test_dataloader.py Choose test affinity based on current affinity (#80327) 2022-07-20 16:45:57 +00:00
test_datapipe.py [DataPipe] Fixes various warnings, exceptions, and clean up testing (#81833) 2022-07-21 18:59:40 +00:00
test_decomp.py Add prelu and relu6 refs missing from __all__ and decomp db (#81420) 2022-07-19 20:50:04 +00:00
test_deploy.py
test_determination.py
test_dispatch.py
test_dynamic_shapes.py Added a couple more symint magic methods + symbolic shape infra (#81086) 2022-07-16 23:47:58 +00:00
test_expanded_weights.py [Expanded Weights] add ability to not specify batch size (#80944) 2022-07-19 13:43:46 +00:00
test_fake_tensor.py normalize cuda device (#81739) 2022-07-21 19:48:56 +00:00
test_foreach.py
test_function_schema.py
test_functional_autograd_benchmark.py
test_functional_optim.py Enable Zero1's ddp_with_overlap for hpu backend (#80438) 2022-07-18 15:05:27 +00:00
test_functionalization.py functionalization: fix for mutable ops with different type promotion rules (#81702) 2022-07-19 18:52:35 +00:00
test_futures.py
test_fx.py Follow GitHub relabeling of oncall: fx for test owners (#81821) 2022-07-21 01:50:06 +00:00
test_fx_backends.py Follow GitHub relabeling of oncall: fx for test owners (#81821) 2022-07-21 01:50:06 +00:00
test_fx_experimental.py Follow GitHub relabeling of oncall: fx for test owners (#81821) 2022-07-21 01:50:06 +00:00
test_fx_passes.py Follow GitHub relabeling of oncall: fx for test owners (#81821) 2022-07-21 01:50:06 +00:00
test_hub.py
test_import_stats.py
test_indexing.py
test_jit.py Lower randint default dtype to the C++ API (#81410) 2022-07-21 16:42:49 +00:00
test_jit_autocast.py
test_jit_cuda_fuser.py [WIP] Upstream push 0627 (#80355) 2022-07-13 19:34:31 +00:00
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py Add OpInfo for torch.equal and fix support for non-standard bools 2022-06-20 23:48:39 +00:00
test_jit_legacy.py
test_jit_llga_fuser.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_jiterator.py
test_kernel_launch_checks.py
test_license.py
test_linalg.py matrix_exp: Make sure _compute_linear_combinations result preserves dim of the input. (#81330) 2022-07-12 21:21:34 +00:00
test_logging.py
test_masked.py
test_meta.py Add prelu and relu6 refs missing from __all__ and decomp db (#81420) 2022-07-19 20:50:04 +00:00
test_metal.py
test_mkldnn.py
test_mobile_optimizer.py
test_model_dump.py
test_module_init.py Add prelu op and module for quantized CPU backend (#73491) 2022-07-20 07:48:15 +00:00
test_modules.py Fix ModuleInfo skip logic (#80471) 2022-07-08 19:52:15 +00:00
test_monitor.py
test_mps.py [MPS] Fixes for MPS testConsistency (#81735) 2022-07-20 16:31:44 +00:00
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtensor.py Reland: Enable dim=None for torch.sum (#79881) 2022-07-09 00:54:42 +00:00
test_namedtuple_return_api.py Expose linalg.solve_ex (#80073) 2022-07-01 16:09:23 +00:00
test_native_functions.py
test_native_mha.py
test_nestedtensor.py Introduce discontinuity to nested tensor (#80981) 2022-07-21 17:17:25 +00:00
test_nn.py Simplify torch.nn.grad by calling into aten::convolution_backward (#81839) 2022-07-21 19:34:27 +00:00
test_nnapi.py
test_numba_integration.py
test_numpy_interop.py
test_openmp.py
test_ops.py Revert "Revert "Refactored prim utils into _prims_utils folder (#81746) 2022-07-20 23:43:57 +00:00
test_ops_gradients.py
test_ops_jit.py Minor fix in jit tests to pass TorchDynamo (#79903) 2022-06-21 00:42:02 +00:00
test_optim.py Revert "Adding maximize to ASGD (#80323)" 2022-07-08 13:35:31 +00:00
test_overrides.py Return mode object from __enter__ (#80998) 2022-07-12 23:22:26 +00:00
test_package.py
test_per_overload_api.py
test_prims.py Add partitioned nvFuser executor with ATen fallbacks (#81043) 2022-07-20 19:51:20 +00:00
test_profiler.py [TorchTidy] Adding Layout support and exposing TensorMetadata (#81155) 2022-07-21 18:49:20 +00:00
test_profiler_tree.py [Profiler][Trivial] Update test_profiler_tree 2022-06-22 22:30:33 +00:00
test_proxy_tensor.py Follow GitHub relabeling of oncall: fx for test owners (#81821) 2022-07-21 01:50:06 +00:00
test_pruning_op.py
test_public_bindings.py Add nondeterministic tags in tags.yaml and add the nondeterministic_seeded tag to all functions in native_functions.yaml defined as nondeterministic by alias_analysis.cpp (#81440) 2022-07-19 21:03:38 +00:00
test_python_dispatch.py Make proxy tensor support item() calls on torch.tensor constants (#81192) 2022-07-15 03:53:40 +00:00
test_pytree.py
test_quantization.py [ao] Created Skeleton for ModelReportVisualizer class (#81523) 2022-07-20 02:39:14 +00:00
test_reductions.py
test_scatter_gather_ops.py Add integer support to scatter_reduce (#80324) 2022-06-29 21:10:26 +00:00
test_segment_reductions.py
test_serialization.py Revert "[fix] allow saving python attr on Tensor and Parameter via torch.save (#81616)" 2022-07-21 10:46:24 +00:00
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py Setup for running PyTorch tests with TorchDynamo and skips for known failing tests (#80106) 2022-07-07 18:57:33 +00:00
test_show_pickle.py
test_sort_and_select.py
test_sparse.py Add spdiags sparse matrix initialization (#78439) 2022-07-01 01:11:54 +00:00
test_sparse_csr.py Revert "[Py3.10] Allow floats to be imported as Long (#81372)" 2022-07-18 14:55:13 +00:00
test_spectral_ops.py Fix istft default output length (#80031) 2022-07-08 23:43:29 +00:00
test_stateless.py Make nn.stateless correctly reset parameters if the forward pass fails (#81262) 2022-07-12 13:54:23 +00:00
test_static_runtime.py [Static Runtime] test case for staticRuntime::runAsync() API (#80407) 2022-07-05 23:40:53 +00:00
test_subclass.py
test_tensor_creation_ops.py Setup for running PyTorch tests with TorchDynamo and skips for known failing tests (#80106) 2022-07-07 18:57:33 +00:00
test_tensorboard.py
test_tensorexpr.py
test_tensorexpr_pybind.py
test_testing.py [ao][sparsity] Post training data sparsifier callback for lightning (#80370) 2022-07-21 16:39:13 +00:00
test_throughput_benchmark.py
test_torch.py Fix test_doc_template regex (#81755) 2022-07-21 16:28:29 +00:00
test_transformers.py disable src mask for transformer and multiheadattention fastpath (#81277) 2022-07-15 20:55:17 +00:00
test_type_hints.py
test_type_info.py
test_type_promotion.py Make addcmul and addcdiv support different dtypes (#74234) 2022-07-06 10:20:33 +00:00
test_typing.py
test_unary_ufuncs.py
test_utils.py Revert "To add hipify_torch as a submodule in pytorch/third_party (#74704)" 2022-06-21 23:54:00 +00:00
test_view_ops.py Setup for running PyTorch tests with TorchDynamo and skips for known failing tests (#80106) 2022-07-07 18:57:33 +00:00
test_vmap.py
test_vulkan.py
test_xnnpack_integration.py