| .. |
|
ao/sparsity
|
|
|
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
[2/N] Cleanup header inclusions in torch_cpu by iwyu (#109964)
|
2023-11-19 20:56:32 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
Revert "Add function to materialize COW storages (#113396)"
|
2023-11-20 10:26:01 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
Update impl_abstract_pystub to be less boilerplatey (#113182)
|
2023-11-08 00:39:00 +00:00 |
|
distributed
|
[dtensor] support convolution ops (#113123)
|
2023-11-20 21:01:28 +00:00 |
|
distributions
|
Add Half support for poisson and use float for Half cumulative distribution on CPU (#112124)
|
2023-11-05 16:10:27 +00:00 |
|
dynamo
|
Insert deferred runtime asserts into Dynamo FX graph (#113958)
|
2023-11-20 21:25:11 +00:00 |
|
edge
|
|
|
|
error_messages
|
|
|
|
expect
|
Revert "Add torch._lazy_clone to create COW tensors (#113397)"
|
2023-11-20 10:21:09 +00:00 |
|
export
|
Fix filename typo 'funtionalized' (#114132)
|
2023-11-20 19:19:25 +00:00 |
|
forward_backward_compatibility
|
expose mem-eff to autograd (#110495)
|
2023-11-13 17:47:40 +00:00 |
|
functorch
|
[HigherOrderOp] Move _map.py to _higher_order_ops (#111152)
|
2023-11-16 03:04:12 +00:00 |
|
fx
|
Fix typo under test directory (#112346)
|
2023-11-03 07:53:33 +00:00 |
|
inductor
|
Don't DCE item nodes if they're float (#114135)
|
2023-11-20 20:42:27 +00:00 |
|
jit
|
Revert "Add support for torch.Generator type in TorchScript (#110413)"
|
2023-11-15 00:51:23 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
Revert "Add support for torch.Generator type in TorchScript (#110413)"
|
2023-11-15 00:51:23 +00:00 |
|
mobile
|
|
|
|
nn
|
add Half support for AdaptiveAvgPool2d and AdaptiveMaxPool2d on CPU (#102079)
|
2023-11-20 03:01:00 +00:00 |
|
onnx
|
Extend _TestONNXRuntime to reuses all tests for new model format (#112289)
|
2023-11-18 00:27:56 +00:00 |
|
onnx_caffe2
|
|
|
|
optim
|
Clean up optimizer imports in test_optim (#113971)
|
2023-11-18 03:52:01 +00:00 |
|
package
|
|
|
|
profiler
|
[BE][profiler] add test for EventList (#113764)
|
2023-11-16 00:49:29 +00:00 |
|
quantization
|
Disallow fp8 type promotion (#113975)
|
2023-11-20 19:47:43 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
torch_np
|
[Dynamo][6.1/N] Refactor out TorchInGraphFunctionVariable and improve heuristic (#113432)
|
2023-11-17 23:42:00 +00:00 |
|
typing
|
|
|
|
_test_bazel.py
|
|
|
|
allowlist_for_publicAPI.json
|
Remove dead tensor_has_hints. (#114071)
|
2023-11-20 16:02:24 +00:00 |
|
conftest.py
|
|
|
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
minioptest_failures_dict.json
|
|
|
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
[TD] Fix metric emission for split test files (#113789)
|
2023-11-16 23:19:40 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
|
|
|
test_autograd.py
|
Fix autograd engine callback error propagation from device thread (#113702)
|
2023-11-17 20:17:02 +00:00 |
|
test_autograd_fallback.py
|
|
|
|
test_binary_ufuncs.py
|
Add Half for aten2, logaddexp, logaddexp2, hypot, and nextafter on CPU (#112138)
|
2023-11-06 06:01:29 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_comparison_utils.py
|
|
|
|
test_compile_benchmark_util.py
|
|
|
|
test_complex.py
|
|
|
|
test_content_store.py
|
Fix recompilation issue with content store (#113533)
|
2023-11-13 23:58:13 +00:00 |
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
|
|
|
test_cpp_extensions_jit.py
|
|
|
|
test_cpp_extensions_open_device_registration.py
|
new_qtensor support privateuseone allocator. (#111464)
|
2023-11-01 05:16:58 +00:00 |
|
test_cuda.py
|
|
|
|
test_cuda_expandable_segments.py
|
|
|
|
test_cuda_multigpu.py
|
|
|
|
test_cuda_nvml_based_avail.py
|
|
|
|
test_cuda_primary_ctx.py
|
|
|
|
test_cuda_sanitizer.py
|
|
|
|
test_cuda_trace.py
|
|
|
|
test_custom_ops.py
|
Grandfather in built-in TorchScript ops to being pt2_compliant (#113061)
|
2023-11-09 02:35:33 +00:00 |
|
test_dataloader.py
|
[BE]: Apply FURB145 to make code more readable and idiomatic. (#112990)
|
2023-11-06 13:15:04 +00:00 |
|
test_datapipe.py
|
|
|
|
test_decomp.py
|
[decomp] Fix _scaled_dot_product_flash_attention decomposition bug (#113102)
|
2023-11-08 21:47:37 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
|
|
|
test_dynamic_shapes.py
|
Explicitly enumerate all method to operator mappings (#113968)
|
2023-11-18 05:43:39 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
[BE] Package dynamic_dims/constraint_dims into CreateSymbolicPolicy (#113802)
|
2023-11-17 18:22:46 +00:00 |
|
test_flop_counter.py
|
|
|
|
test_foreach.py
|
Add 0dim Tensor overload for _foreach_div (#113688)
|
2023-11-15 20:59:32 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
Skip aliasing correction for lift_fresh. (#112202)
|
2023-11-03 20:46:30 +00:00 |
|
test_functionalization_of_rng_ops.py
|
|
|
|
test_futures.py
|
|
|
|
test_fx.py
|
|
|
|
test_fx_experimental.py
|
|
|
|
test_fx_passes.py
|
|
|
|
test_fx_reinplace_pass.py
|
|
|
|
test_hub.py
|
|
|
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
|
|
|
test_itt.py
|
|
|
|
test_jit.py
|
[BE]: ruff - enable PIE804 (#113951)
|
2023-11-17 21:23:02 +00:00 |
|
test_jit_autocast.py
|
|
|
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
|
|
|
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_legacy_vmap.py
|
|
|
|
test_license.py
|
|
|
|
test_linalg.py
|
cholesky_solve_backward: speed up using output_mask (#112981)
|
2023-11-16 18:30:57 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
|
|
|
test_matmul_cuda.py
|
[CUDA][cuBLAS] Separate reduced precision reductions on/off for addmm tests (#112545)
|
2023-11-07 19:09:29 +00:00 |
|
test_meta.py
|
Add OpInfo test that tests meta functions binary ufuncs with different dtypes (#113674)
|
2023-11-16 19:09:12 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
|
|
|
test_mkldnn_fusion.py
|
|
|
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_model_exports_to_core_aten.py
|
Build and test ExecuTorch on PyTorch (#113364)
|
2023-11-16 02:19:58 +00:00 |
|
test_module_init.py
|
|
|
|
test_modules.py
|
|
|
|
test_monitor.py
|
|
|
|
test_mps.py
|
[MPS] Fix and refactor unary/binary ops with non-zero offset or non-contiguous output (#97085)
|
2023-11-14 22:03:21 +00:00 |
|
test_multiprocessing.py
|
|
|
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
Add some checks about Device and Layout when create/convert named tensor (#113628)
|
2023-11-14 20:40:27 +00:00 |
|
test_namedtuple_return_api.py
|
|
|
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
|
|
|
test_nestedtensor.py
|
[Dynamo][6.1/N] Refactor out TorchInGraphFunctionVariable and improve heuristic (#113432)
|
2023-11-17 23:42:00 +00:00 |
|
test_nn.py
|
[ROCm] fix test_softmax_forward_64bit_indexing_cuda OOM (#113093)
|
2023-11-07 03:00:37 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
|
|
|
test_openmp.py
|
|
|
|
test_ops.py
|
BUG: trace frames with numpy scalar -> ndarray functions (#112959)
|
2023-11-17 03:00:24 +00:00 |
|
test_ops_fwd_gradients.py
|
s390x: skip tests relying on specific openblas precision (#112843)
|
2023-11-07 17:18:20 +00:00 |
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
|
|
|
test_out_dtype_op.py
|
|
|
|
test_overrides.py
|
|
|
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
|
|
|
test_proxy_tensor.py
|
Replace expect_int with guard_int (#113921)
|
2023-11-20 21:27:48 +00:00 |
|
test_pruning_op.py
|
[dynamo] Eagerly install guards (#111415)
|
2023-11-07 19:55:19 +00:00 |
|
test_public_bindings.py
|
Improves comparison of state dicts for Checkpoint E2E Tests (#113181)
|
2023-11-15 20:48:45 +00:00 |
|
test_python_dispatch.py
|
Make numel/sym_numel PyInterpreter work symmetrically to others (#113065)
|
2023-11-08 17:44:29 +00:00 |
|
test_pytree.py
|
Revert "[pytree] reorganize submodule structure for C++ and Python pytree (#112278)"
|
2023-11-10 16:30:36 +00:00 |
|
test_quantization.py
|
[quant][pt2] Support conv1d-bn QAT fusion (#113714)
|
2023-11-17 22:09:30 +00:00 |
|
test_reductions.py
|
Add Half support for cummax, cummin, cumprod, logcumsumexp, and prod on CPU (#112132)
|
2023-11-05 12:31:38 +00:00 |
|
test_scatter_gather_ops.py
|
|
|
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
[BE] Do not warn when safely loading legacy dicts (#113614)
|
2023-11-14 22:09:10 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
|
|
|
test_sparse.py
|
Add sparse tensors support to dataloader. (#112842)
|
2023-11-19 16:05:27 +00:00 |
|
test_sparse_csr.py
|
[dynamo] Eagerly install guards (#111415)
|
2023-11-07 19:55:19 +00:00 |
|
test_sparse_semi_structured.py
|
|
|
|
test_spectral_ops.py
|
|
|
|
test_stateless.py
|
|
|
|
test_static_runtime.py
|
Skip test_fork_wait_4 and test_fork_wait_4_async (#112743)
|
2023-11-02 20:46:29 +00:00 |
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
Run sympy expressions with Python values / FX tracing (#113978)
|
2023-11-20 21:25:11 +00:00 |
|
test_tensor_creation_ops.py
|
Remove unecessary warning when getting storage.filename (#113212)
|
2023-11-08 02:09:59 +00:00 |
|
test_tensorboard.py
|
|
|
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
robustify parametrize default name (#113856)
|
2023-11-16 23:25:04 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Revert "Add torch._lazy_clone to create COW tensors (#113397)"
|
2023-11-20 10:21:09 +00:00 |
|
test_transformers.py
|
expose sdpa helpers to python (#110496)
|
2023-11-15 07:34:34 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
Disallow fp8 type promotion (#113975)
|
2023-11-20 19:47:43 +00:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
|
|
|
test_utils.py
|
Move test_utils.py back to MYPY (#113745)
|
2023-11-16 01:57:58 +00:00 |
|
test_view_ops.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
