| .. |
|
ao/sparsity
|
|
|
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
[3/N] Remove inclusion of c10/util/string_utils.h (#128504)
|
2024-06-15 06:38:40 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
|
|
|
custom_backend
|
|
|
|
custom_operator
|
[custom ops] convert string type annotation to real type (#128809)
|
2024-06-18 00:55:50 +00:00 |
|
distributed
|
[Horace's PR #126446] Prevent partitioner from ever saving views (#129039)
|
2024-06-19 23:21:16 +00:00 |
|
distributions
|
|
|
|
dynamo
|
[Horace's PR #126446] Prevent partitioner from ever saving views (#129039)
|
2024-06-19 23:21:16 +00:00 |
|
dynamo_expected_failures
|
re-enable jacrev/jacfwd/hessian after #128028 landed (#128622)
|
2024-06-18 17:08:58 +00:00 |
|
dynamo_skips
|
Dynamo testing: add some skips (#128734)
|
2024-06-14 20:53:30 +00:00 |
|
edge
|
|
|
|
error_messages
|
|
|
|
expect
|
Revert "Short-term fix to preserve NJT metadata cache in torch.compile (#122836)"
|
2024-06-19 00:28:53 +00:00 |
|
export
|
Revert "[export] experimental joint graph API. (#128847)"
|
2024-06-19 19:04:36 +00:00 |
|
forward_backward_compatibility
|
Remove amax return from _scaled_mm (#128683)
|
2024-06-17 16:48:00 +00:00 |
|
functorch
|
re-enable jacrev/jacfwd/hessian after #128028 landed (#128622)
|
2024-06-18 17:08:58 +00:00 |
|
fx
|
Save DOT file of graph instead of SVG for GraphTranformObserver (#128634)
|
2024-06-14 00:54:22 +00:00 |
|
higher_order_ops
|
[Inductor] Fix the High Order Op layout issue (#128275)
|
2024-06-15 00:33:21 +00:00 |
|
inductor
|
[inductor] Fix logging for run_and_get_cpp_code (#128794)
|
2024-06-19 21:32:34 +00:00 |
|
jit
|
[Caffe2] [2/N] Remove Caffe2 from tests (#128911)
|
2024-06-19 00:05:50 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
|
|
|
mobile
|
|
|
|
nn
|
Revert "Make nn.Module state_dict load_state_dict pre-hook and state_dict post hook public (#126704)"
|
2024-06-11 17:45:20 +00:00 |
|
onnx
|
[ONNX] Add onnx::Gelu support for version 20 (#128773)
|
2024-06-19 15:39:02 +00:00 |
|
optim
|
Quick Fix on #126854, deepcopy lr and other possible base_parameters (#127190)
|
2024-06-03 18:06:31 +00:00 |
|
package
|
|
|
|
profiler
|
[RELAND2][dynamo][nn-modules] Trace through nn.Module dunder methods for UnspecializedNNModule (#126578)
|
2024-06-12 04:09:23 +00:00 |
|
quantization
|
[Caffe2] [2/N] Remove Caffe2 from tests (#128911)
|
2024-06-19 00:05:50 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
torch_np
|
Revert "[dynamo] Support ndarray.dtype attribute access (#124490)"
|
2024-06-06 14:21:29 +00:00 |
|
typing
|
Flip default value for mypy disallow_untyped_defs [final] (#127836)
|
2024-06-12 15:28:42 +00:00 |
|
xpu
|
|
|
|
_test_bazel.py
|
|
|
|
allowlist_for_publicAPI.json
|
[pipelining] Rename to stage.py and schedules.py (#128278)
|
2024-06-08 04:42:35 +00:00 |
|
conftest.py
|
[BE]: Enable ruff TCH rules and autofixes for better imports (#127688)
|
2024-06-06 16:55:58 +00:00 |
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
hi.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
minioptest_failures_dict.json
|
|
|
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
pytest_shard_custom.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
[TD] Test removal on sm86 (#127131)
|
2024-06-07 20:19:18 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
[BE] wrap deprecated function/class with typing_extensions.deprecated (#127689)
|
2024-06-02 12:30:43 +00:00 |
|
test_autograd.py
|
[checkpoint] Clean up selective activation checkpoint and make public (#125795)
|
2024-06-18 18:18:50 +00:00 |
|
test_autograd_fallback.py
|
|
|
|
test_binary_ufuncs.py
|
Update fake tensor error checks for bool tensor subtraction (#128492)
|
2024-06-17 13:41:15 +00:00 |
|
test_bundled_images.py
|
Flip default value for mypy disallow_untyped_defs [1/11] (#127838)
|
2024-06-08 18:16:33 +00:00 |
|
test_bundled_inputs.py
|
Flip default value for mypy disallow_untyped_defs [1/11] (#127838)
|
2024-06-08 18:16:33 +00:00 |
|
test_ci_sanity_check_fail.py
|
|
|
|
test_comparison_utils.py
|
|
|
|
test_compile_benchmark_util.py
|
|
|
|
test_complex.py
|
Flip default value for mypy disallow_untyped_defs [1/11] (#127838)
|
2024-06-08 18:16:33 +00:00 |
|
test_content_store.py
|
|
|
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
|
|
|
test_cpp_extensions_jit.py
|
|
|
|
test_cpp_extensions_mtia_backend.py
|
|
|
|
test_cpp_extensions_open_device_registration.py
|
|
|
|
test_cpp_extensions_stream_and_event.py
|
|
|
|
test_cuda.py
|
[BE][Eazy] remove torch.torch.xxx usages (#127800)
|
2024-06-05 21:53:49 +00:00 |
|
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
|
Rename impl_abstract to register_fake, part 2/2 (#123938)
|
2024-06-14 14:37:24 +00:00 |
|
test_dataloader.py
|
add multiprocessing checks in test_dataloader.py (#128244)
|
2024-06-15 01:32:55 +00:00 |
|
test_datapipe.py
|
[BE][Ez]: Apply PYI059 - Generic always come last (#127685)
|
2024-06-02 13:38:58 +00:00 |
|
test_decomp.py
|
|
|
|
test_deploy.py
|
|
|
|
test_determination.py
|
[Caffe2] [2/N] Remove Caffe2 from tests (#128911)
|
2024-06-19 00:05:50 +00:00 |
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
|
|
|
test_dynamic_shapes.py
|
Do not generate -1* in SymPy expressions when canonicalising (#128411)
|
2024-06-13 16:49:59 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
[fake tensor] Set _is_param for base fake tensors for views (#127823)
|
2024-06-05 20:26:52 +00:00 |
|
test_flop_counter.py
|
Revert "[cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343)"
|
2024-06-12 18:47:52 +00:00 |
|
test_foreach.py
|
Improve the debugging message for when foreach mta_called (#128991)
|
2024-06-19 00:25:09 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
|
|
|
test_functionalization_of_rng_ops.py
|
[dynamo][aot autograd] Silently disable default saved tensor hooks during tracing (#123196)
|
2024-06-14 20:28:08 +00:00 |
|
test_futures.py
|
Flip default value for mypy disallow_untyped_defs [1/11] (#127838)
|
2024-06-08 18:16:33 +00:00 |
|
test_fx.py
|
|
|
|
test_fx_experimental.py
|
Revert "Add aten._unsafe_masked_index (#116491)"
|
2024-06-03 15:51:50 +00:00 |
|
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
|
[1/N] [Caffe2] Remove caffe2_aten_fallback code (#128675)
|
2024-06-17 21:25:59 +00:00 |
|
test_jit_autocast.py
|
|
|
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Enable UFMT on test/test_jit_fuser_te.py (#127759)
|
2024-06-04 16:56:03 +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_legacy_vmap.py
|
|
|
|
test_license.py
|
Fix manual licensing (#128630)
|
2024-06-14 00:12:09 +00:00 |
|
test_linalg.py
|
[ROCm] TunableOp improvements (#124362)
|
2024-06-03 22:30:11 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
|
|
|
test_matmul_cuda.py
|
Enable fp8 rowwise scaling kernel on cuda, TAKE 2: #125204 (#128989)
|
2024-06-19 04:49:39 +00:00 |
|
test_meta.py
|
|
|
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
|
|
|
test_mkldnn_fusion.py
|
|
|
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
[BE] enable ruff rule Q from flake8-quotes (#127713)
|
2024-06-02 23:25:26 +00:00 |
|
test_model_dump.py
|
|
|
|
test_model_exports_to_core_aten.py
|
|
|
|
test_module_tracker.py
|
|
|
|
test_modules.py
|
Revert "Default meta device to use swap_tensors in nn.Module._apply (.to_empty and .to('meta')) (#126819)"
|
2024-06-01 18:46:16 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
[MPS] Fused Adam & AdamW (#127242)
|
2024-06-18 19:59:50 +00:00 |
|
test_multiprocessing.py
|
expose set_thread_name to Python and set thread names (#128448)
|
2024-06-13 16:38:23 +00:00 |
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
|
|
|
test_namedtuple_return_api.py
|
|
|
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
|
|
|
test_nestedtensor.py
|
Revert "Short-term fix to preserve NJT metadata cache in torch.compile (#122836)"
|
2024-06-19 00:28:53 +00:00 |
|
test_nn.py
|
[easy] Move state_dict hooks tests to test_module_hooks and decorate tests that call load_state_dict with swap (#126906)
|
2024-06-10 21:50:17 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
[dynamo][numpy] Add unsigned integer dtypes (#125717)
|
2024-06-05 14:33:47 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
Add fake impl for aten.unique_dim (#126561)
|
2024-06-01 04:03:10 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
[MPS] Fused Adam & AdamW (#127242)
|
2024-06-18 19:59:50 +00:00 |
|
test_out_dtype_op.py
|
|
|
|
test_overrides.py
|
Properly detect nested torch function args (#127496)
|
2024-06-02 03:43:22 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
[BE] wrap deprecated function/class with typing_extensions.deprecated (#127689)
|
2024-06-02 12:30:43 +00:00 |
|
test_proxy_tensor.py
|
Introduce int_oo (#127693)
|
2024-06-13 04:08:20 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
[Caffe2] [2/N] Remove Caffe2 from tests (#128911)
|
2024-06-19 00:05:50 +00:00 |
|
test_python_dispatch.py
|
|
|
|
test_pytree.py
|
[BE] wrap deprecated function/class with typing_extensions.deprecated (#127689)
|
2024-06-02 12:30:43 +00:00 |
|
test_quantization.py
|
|
|
|
test_reductions.py
|
|
|
|
test_scatter_gather_ops.py
|
|
|
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
Remove tensor subclass detection logic from weights_only unpickler (#127808)
|
2024-06-05 02:14:30 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
Enable UFMT on test_shape_ops.py test_show_pickle.py test_sort_and_select.py (#127165)
|
2024-06-05 14:31:26 +00:00 |
|
test_show_pickle.py
|
Enable UFMT on test_shape_ops.py test_show_pickle.py test_sort_and_select.py (#127165)
|
2024-06-05 14:31:26 +00:00 |
|
test_sort_and_select.py
|
Remove global variable SIZE, and fix linter warning (#128559)
|
2024-06-13 22:09:51 +00:00 |
|
test_sparse.py
|
|
|
|
test_sparse_csr.py
|
|
|
|
test_sparse_semi_structured.py
|
[CUDA][Sparse] Change comparison function of test_sparse_semi_structured.py and bump tolerances for sp24_matmuls (#128553)
|
2024-06-13 06:58:07 +00:00 |
|
test_spectral_ops.py
|
|
|
|
test_stateless.py
|
[BE] wrap deprecated function/class with typing_extensions.deprecated (#127689)
|
2024-06-02 12:30:43 +00:00 |
|
test_static_runtime.py
|
[BE] enable ruff rule Q from flake8-quotes (#127713)
|
2024-06-02 23:25:26 +00:00 |
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
Introduce int_oo (#127693)
|
2024-06-13 04:08:20 +00:00 |
|
test_tensor_creation_ops.py
|
Fix Storage.filename to not track the filename when storage was mmap-ed with MAP_PRIVATE (#128725)
|
2024-06-17 18:55:47 +00:00 |
|
test_tensorboard.py
|
[Caffe2] [2/N] Remove Caffe2 from tests (#128911)
|
2024-06-19 00:05:50 +00:00 |
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
Retire torch.distributed.pipeline (#127354)
|
2024-06-07 08:11:58 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
[Caffe2] [2/N] Remove Caffe2 from tests (#128911)
|
2024-06-19 00:05:50 +00:00 |
|
test_transformers.py
|
Revert "[cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343)"
|
2024-06-12 18:47:52 +00:00 |
|
test_type_hints.py
|
Flip default value for mypy disallow_untyped_defs [1/11] (#127838)
|
2024-06-08 18:16:33 +00:00 |
|
test_type_info.py
|
Flip default value for mypy disallow_untyped_defs [1/11] (#127838)
|
2024-06-08 18:16:33 +00:00 |
|
test_type_promotion.py
|
|
|
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
Revert "[dynamo] Support ndarray.dtype attribute access (#124490)"
|
2024-06-06 14:21:29 +00:00 |
|
test_utils.py
|
Flip default value for mypy disallow_untyped_defs [final] (#127836)
|
2024-06-12 15:28:42 +00:00 |
|
test_view_ops.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
|
test_xpu.py
|
Clean up xpu ut to make CI happy (#128383)
|
2024-06-13 07:06:41 +00:00 |
