| .. |
|
ao/sparsity
|
|
|
|
autograd
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
[caffe2] fix build failures in optimized builds under clang
|
2022-02-22 22:31:47 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
|
|
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[ZeRO] (Reland) Add ctor support for multiple param groups (#72932)
|
2022-02-22 16:29:55 +00:00 |
|
distributions
|
Fix discrete sampler test to correctly run Chi2 test (#73251)
|
2022-02-23 02:31:42 +00:00 |
|
error_messages
|
|
|
|
expect
|
Refactor FX codegen into extensible Codegen object (#72566)
|
2022-02-11 18:13:29 +00:00 |
|
forward_backward_compatibility
|
scatter_reduce documentation (#73125)
|
2022-02-22 19:33:46 +00:00 |
|
fx
|
[test_fx_const_fold] Remove dependencies on acc_* (#72810)
|
2022-02-14 20:46:54 +00:00 |
|
jit
|
Fix asserts in tests (#72864)
|
2022-02-16 18:35:16 +00:00 |
|
jit_hooks
|
|
|
|
mobile
|
[TensorExpr] Move AOT compilation logic from aot_compiler.cpp to NNC's to_backend (#70375)
|
2022-02-02 02:34:55 +00:00 |
|
onnx
|
[ONNX] Add Squeeze/Unsqueeze dynamic dimensions support when opset >= 13 (#71158)
|
2022-02-23 06:41:15 +00:00 |
|
package
|
[torch.package] add test case for repackaging parent module (#72367)
|
2022-02-07 21:49:36 +00:00 |
|
quantization
|
[quant] Add ConvTranspose reference module - Reland #73031 (#73094)
|
2022-02-23 02:31:42 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
run_test.py
|
add dry run option and improve test list printing
|
2022-02-22 20:45:41 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
|
|
|
test_autograd.py
|
no longer coalesce sparse COO tensors before comparison (#69751)
|
2022-02-17 02:33:08 +00:00 |
|
test_binary_ufuncs.py
|
only compare attributes for meta tensors (#72508)
|
2022-02-17 02:33:08 +00:00 |
|
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_cuda.py
|
no longer coalesce sparse COO tensors before comparison (#69751)
|
2022-02-17 02:33:08 +00:00 |
|
test_cuda_primary_ctx.py
|
remove hasPrimaryContext workaround on ROCm (#71146)
|
2022-01-25 20:32:12 +00:00 |
|
test_dataloader.py
|
Fix persistent worker exits before pin_memory thread (#71579)
|
2022-02-01 23:57:17 +00:00 |
|
test_datapipe.py
|
[DataPipe] Extend FileLister to support load multiple directories (#72260)
|
2022-02-04 07:55:00 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
Revert D34034848: free up dispatch key space (in C++)
|
2022-02-14 23:29:00 +00:00 |
|
test_expanded_weights.py
|
expanded weights without fast rules (#70140)
|
2022-02-22 20:35:16 +00:00 |
|
test_foreach.py
|
use torch.testing.assert_equal in TestCase.assertEqual (#67796)
|
2022-01-27 08:33:55 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
fix python version error (#71021)
|
2022-01-21 01:05:53 +00:00 |
|
test_functional_optim.py
|
[Opt Overlap] Clean up code in _OptimizerHookState (#71620)
|
2022-01-26 19:33:49 +00:00 |
|
test_functionalization.py
|
Fix test ownership lint (#71554)
|
2022-01-21 18:24:42 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
Add docstrings to native_channel_shuffle (#72919)
|
2022-02-17 02:33:08 +00:00 |
|
test_fx_experimental.py
|
Remove some unnecessary python functional wrappers (#61608)
|
2022-02-01 16:59:26 +00:00 |
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
|
|
|
test_jit.py
|
(2/2) Make TorchScript Preserve Fully Qualified Class Name for Python Exceptions: frontend change (#72899)
|
2022-02-16 03:45:15 +00:00 |
|
test_jit_autocast.py
|
|
|
|
test_jit_cuda_fuser.py
|
Nvfuser code bump 2_1_2022 (#72127)
|
2022-02-15 00:43:16 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Implement Tanh Gelu Approximation (#61439)
|
2022-02-14 03:40:32 +00:00 |
|
test_jit_legacy.py
|
|
|
|
test_jit_profiling.py
|
|
|
|
test_jit_simple.py
|
|
|
|
test_jit_string.py
|
|
|
|
test_kernel_launch_checks.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
Fix SVD error code handling for OpenBLAS 0.3.15+ and MKL 2022+ (again) (#72357)
|
2022-02-07 21:36:30 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
Fix test ownership lint (#71554)
|
2022-01-21 18:24:42 +00:00 |
|
test_metal.py
|
|
|
|
test_mkldnn.py
|
|
|
|
test_mobile_optimizer.py
|
[PyTorchEdge] backport v8 to v7 to support promoted ops as instruction (#71662)
|
2022-02-15 03:47:39 +00:00 |
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
|
|
|
test_modules.py
|
only compare attributes for meta tensors (#72508)
|
2022-02-17 02:33:08 +00:00 |
|
test_monitor.py
|
torch/monitor: merge Interval and FixedCount stats (#72009)
|
2022-01-30 23:21:59 +00:00 |
|
test_multiprocessing.py
|
Allow forking until a worker thread is created in autograd engine (#72689)
|
2022-02-12 01:52:57 +00:00 |
|
test_multiprocessing_spawn.py
|
Make test_multiprocessing_spawn.py compatible with pytest (#50408)
|
2022-02-10 16:18:46 +00:00 |
|
test_namedtensor.py
|
use torch.testing.assert_equal in TestCase.assertEqual (#67796)
|
2022-01-27 08:33:55 +00:00 |
|
test_namedtuple_return_api.py
|
Rewrite svd and linalg.svd as structured kernels (#69827)
|
2022-01-27 18:38:30 +00:00 |
|
test_native_functions.py
|
|
|
|
test_nn.py
|
Add simple correctness check for native MHA (#72941)
|
2022-02-19 00:31:45 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
Skip superfluous storage allocations while constructing meta tensors (#65331)
|
2022-02-11 12:55:11 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
[ROCm] Enable skipped ROCm unit tests (#67706)
|
2022-02-11 22:14:54 +00:00 |
|
test_optim.py
|
Uncomment step no-op test in test_optim (#70953)
|
2022-02-15 18:02:08 +00:00 |
|
test_overrides.py
|
Add autograd tests for addmm, addmv, mm, mv and CSR matrix input (#71949)
|
2022-02-11 23:14:02 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
Add torch.ops per overload API (#72206)
|
2022-02-11 17:19:06 +00:00 |
|
test_profiler.py
|
|
|
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
|
|
|
test_python_dispatch.py
|
Revert D34318185: [pytorch][PR] Ensure that call before redispatch work well for PythonTLSSnapshot
|
2022-02-22 18:31:13 +00:00 |
|
test_pytree.py
|
|
|
|
test_quantization.py
|
|
|
|
test_reductions.py
|
optimize dim reduce performance on norm, argmax and argmin (#72083)
|
2022-02-10 19:39:31 +00:00 |
|
test_scatter_gather_ops.py
|
Kill the test_torch.py mixin and creates test_scatter_gather_ops (#71691)
|
2022-01-24 09:32:32 +00:00 |
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
Rename Typed/UntypedStorage to _Typed/_UntypedStorage (#72540)
|
2022-02-15 23:53:01 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
Revert D34152115: [pytorch][PR] [ROCm] Enable sort operator BF16 support
|
2022-02-15 15:23:29 +00:00 |
|
test_sparse.py
|
Back out "[pytorch][PR] add BFloat16 sparse operators on CPU: copy, coalesce, sparse_mask, ad…"
|
2022-02-21 18:31:51 +00:00 |
|
test_sparse_csr.py
|
Enable Half/BFloat16 support for to_dense and coalesce methods. (#72397)
|
2022-02-17 02:54:23 +00:00 |
|
test_spectral_ops.py
|
fft: Fix invalid shape error for complex-to-real transforms (#73012)
|
2022-02-23 03:33:26 +00:00 |
|
test_stateless.py
|
Add lightweight reparametrization for _stateless calls (#68969)
|
2022-01-28 14:38:45 +00:00 |
|
test_static_runtime.py
|
Fix quicklints (#72256)
|
2022-02-03 15:17:35 +00:00 |
|
test_tensor_creation_ops.py
|
only compare attributes for meta tensors (#72508)
|
2022-02-17 02:33:08 +00:00 |
|
test_tensorboard.py
|
|
|
|
test_tensorexpr.py
|
[JIT] Opinfo tests for nnc fusion - retry (#72486)
|
2022-02-09 19:01:22 +00:00 |
|
test_tensorexpr_pybind.py
|
[TensorExpr] Delet DimArg class. (#72390)
|
2022-02-11 01:21:59 +00:00 |
|
test_testing.py
|
only compare attributes for meta tensors (#72508)
|
2022-02-17 02:33:08 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
scatter_reduce documentation (#73125)
|
2022-02-22 19:33:46 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
only compare attributes for meta tensors (#72508)
|
2022-02-17 02:33:08 +00:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
|
|
|
test_utils.py
|
|
|
|
test_view_ops.py
|
only compare attributes for meta tensors (#72508)
|
2022-02-17 02:33:08 +00:00 |
|
test_vmap.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
