| .. |
|
ao/sparsity
|
|
|
|
autograd
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
[PyTorchEdge] Extend flatbuffer to support extra files map (#72951)
|
2022-02-24 19:39:32 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
Revert D33992795: Add BUILD_LAZY_CUDA_LINALG option
|
2022-02-25 18:37:31 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[FSDP] full_state_dict impl (#73324)
|
2022-02-27 19:32:22 +00:00 |
|
distributions
|
Add SoftplusTransform (#52300)
|
2022-02-25 02:30:03 +00:00 |
|
error_messages
|
|
|
|
expect
|
Refactor FX codegen into extensible Codegen object (#72566)
|
2022-02-11 18:13:29 +00:00 |
|
forward_backward_compatibility
|
Optimize grid sample 3d
|
2022-02-23 19:25:17 +00:00 |
|
fx
|
[test_fx_const_fold] Remove dependencies on acc_* (#72810)
|
2022-02-14 20:46:54 +00:00 |
|
jit
|
Prepare for an update to the XNNPACK submodule (#72642)
|
2022-02-25 00:39:15 +00:00 |
|
jit_hooks
|
|
|
|
mobile
|
Add dynamic shape support to AOT driver & compiler (#72995)
|
2022-02-24 04:30:48 +00:00 |
|
onnx
|
Revert "Improve logging on ONNX symbolic helpers"
|
2022-02-25 16:58:51 +00:00 |
|
package
|
[torch.package] add test case for repackaging parent module (#72367)
|
2022-02-07 21:49:36 +00:00 |
|
quantization
|
[quant][fx] Add root_node_getter in backend_config_dict (#73345)
|
2022-02-26 06:34:22 +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
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_complex.py
|
|
|
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
No-batch-dim support for ConvNd (#70506)
|
2022-01-06 16:53:50 -08:00 |
|
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
|
[DataPipe] Fix issue with DataPipe serialization with dill (#72896)
|
2022-02-23 16:31:20 +00:00 |
|
test_datapipe.py
|
[DataPipe] Adding serialization test at different stages of reading for IterDataPipes (#73119)
|
2022-02-23 16:31:21 +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
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +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
|
Add forward compatability tests in CI (#64139)
|
2022-01-05 23:40:06 -08:00 |
|
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
|
[fx] Improve support for tuple subclasses such as NamedTuple (#73198)
|
2022-02-23 11:31:10 +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
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +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
|
[PyTorch] Move prim string ops to JIT op registry (#70501)
|
2022-01-06 18:26:28 -08:00 |
|
test_kernel_launch_checks.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
Testing of masked reductions on mixed layout inputs. (#72398)
|
2022-02-25 05:32:47 +00:00 |
|
test_metal.py
|
|
|
|
test_mkldnn.py
|
|
|
|
test_mobile_optimizer.py
|
Prepare for an update to the XNNPACK submodule (#72642)
|
2022-02-25 00:39:15 +00:00 |
|
test_model_dump.py
|
Prepare for an update to the XNNPACK submodule (#72642)
|
2022-02-25 00:39:15 +00:00 |
|
test_module_init.py
|
[quant] fix dropout in FX graph mode quantization (#71043)
|
2022-01-13 15:59:59 -08:00 |
|
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
|
Optimize grid sample 3d
|
2022-02-23 19:25:17 +00:00 |
|
test_nnapi.py
|
NNAPI: quant logistic fix (#70847)
|
2022-01-07 13:36:33 -08:00 |
|
test_numba_integration.py
|
Remove unwanted comma (#71193)
|
2022-01-13 19:09:05 -08:00 |
|
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
|
[Profiler] Fix memory profile type from recent refactor (#71417)
|
2022-01-18 22:18:11 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
|
|
|
test_python_dispatch.py
|
Fix serialization and deepcopying for wrapper subclasses
|
2022-02-24 18:21:25 +00:00 |
|
test_pytree.py
|
|
|
|
test_quantization.py
|
dbr quant: break up test class into multiple classes (#70246)
|
2022-01-05 06:36:41 -08:00 |
|
test_reductions.py
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30: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
|
Fix serialization and deepcopying for wrapper subclasses
|
2022-02-24 18:21:25 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
test_sparse.py
|
Fix test_Sparse_to_Sparse_copy__cuda_bfloat16 failure (#73157)
|
2022-02-28 05:33:50 +00:00 |
|
test_sparse_csr.py
|
Sparse CSR CPU: implement addmm(dense, sparse, sparse) -> dense (#73076)
|
2022-02-26 01:08:45 +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
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +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
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +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
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
test_utils.py
|
|
|
|
test_view_ops.py
|
align signature of make_tensor with other creation ops (#72702)
|
2022-02-25 06:30:31 +00:00 |
|
test_vmap.py
|
update example in classification losses (#69816)
|
2021-12-21 02:46:48 -08:00 |
|
test_vulkan.py
|
|
|
|
test_xnnpack_integration.py
|
Update XNNPACK (#73467)
|
2022-02-28 06:09:26 +00:00 |
a22d079abd