| .. |
|
ao/sparsity
|
[pruning][core][feature] LSTM Structured Pruning prune_functions + pattern (#90801)
|
2023-02-01 19:29:03 +00:00 |
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
Remove deprecated torch.symeig (#70988)
|
2023-01-31 11:59:11 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
|
|
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[dtensor] add split_with_sizes op (#93957)
|
2023-02-03 04:16:30 +00:00 |
|
distributions
|
[Dynamo] Fix calling UserDefinedObject.func should pass self object (#92050)
|
2023-01-21 05:47:01 +00:00 |
|
dynamo
|
Tag dynamo backends as debug/experimental (#93878)
|
2023-02-04 00:50:51 +00:00 |
|
edge
|
[torchgen] Add CI job to cover custom ops registration for Executorch (#91291)
|
2023-01-14 02:30:54 +00:00 |
|
error_messages
|
|
|
|
expect
|
[pt2] Fix arange to match ATen behavior (#93353)
|
2023-02-03 00:44:32 +00:00 |
|
forward_backward_compatibility
|
retire sparse_mask_helper (#91714)
|
2023-02-02 13:53:02 +00:00 |
|
functorch
|
Enable Python dispatcher on inference-only aot_dispatch_base (#94118)
|
2023-02-04 06:10:21 +00:00 |
|
fx
|
Match get_attr when compare node (#91657)
|
2023-01-09 08:10:55 +00:00 |
|
inductor
|
To vectorize long datatype as mask index (#91076)
|
2023-02-05 03:36:22 +00:00 |
|
jit
|
[jit] jit._drop fun modifier to allow in jit class non-jit decl funs (#93012)
|
2023-02-01 09:02:05 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
|
|
|
mobile
|
[BE][CI] change references from .jenkins to .ci (#92624)
|
2023-01-30 22:50:07 +00:00 |
|
nn
|
correct use of torch.backends.cudnn.flags() (#93182)
|
2023-01-28 06:50:06 +00:00 |
|
onnx
|
[ONNX] Fix Gather replacement in RNN peephole (#93120)
|
2023-02-01 06:29:17 +00:00 |
|
onnx_caffe2
|
|
|
|
package
|
[package] Add better debugging for torch.package (#92939)
|
2023-01-26 02:11:12 +00:00 |
|
profiler
|
[nn] zero_grad() set_to_none default True (#92731)
|
2023-01-26 01:04:28 +00:00 |
|
quantization
|
ns: add compare_weights API with a single model (#92058)
|
2023-02-03 01:17:19 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
allowlist_for_publicAPI.json
|
retire sparse_mask_helper (#91714)
|
2023-02-02 13:53:02 +00:00 |
|
conftest.py
|
|
|
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
Enable xdoctest runner in CI for real this time (#83816)
|
2022-12-29 05:32:42 +00:00 |
|
run_test.py
|
[BE][CI] change references from .jenkins to .ci (#92624)
|
2023-01-30 22:50:07 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
[pruning][core][feature] Add in SaliencyPruner to pruner._experimental (#91814)
|
2023-01-10 04:04:55 +00:00 |
|
test_autocast.py
|
|
|
|
test_autograd.py
|
Remove deprecated torch.symeig (#70988)
|
2023-01-31 11:59:11 +00:00 |
|
test_binary_ufuncs.py
|
Vectorized more stable complex division (#93277)
|
2023-02-03 11:48:20 +00:00 |
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_comparison_utils.py
|
|
|
|
test_complex.py
|
|
|
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
Set min supported Python version to 3.8 (#93155)
|
2023-01-29 18:28:46 +00:00 |
|
test_cpp_extensions_jit.py
|
Fix test_jit_cuda_archflags on Windows (#93332)
|
2023-02-02 02:49:27 +00:00 |
|
test_cpp_extensions_open_device_registration.py
|
|
|
|
test_cuda.py
|
FusedAdam(W) should take OptState into account before unscaling grads (#94060)
|
2023-02-04 05:20:13 +00:00 |
|
test_cuda_nvml_based_avail.py
|
|
|
|
test_cuda_primary_ctx.py
|
|
|
|
test_cuda_sanitizer.py
|
|
|
|
test_cuda_trace.py
|
|
|
|
test_dataloader.py
|
[CI][ROCm] skip multiprocessing tests that trigger hangs (#92101)
|
2023-01-13 22:39:00 +00:00 |
|
test_datapipe.py
|
Added keep_key option to Grouper (#92532)
|
2023-01-25 20:58:21 +00:00 |
|
test_decomp.py
|
Add test tracking operators without decompositions (#90887)
|
2023-01-26 01:44:42 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
[BE][CI] change references from .jenkins to .ci (#92624)
|
2023-01-30 22:50:07 +00:00 |
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
|
|
|
test_dynamic_shapes.py
|
[fx] fix type promotion in binary_magic_impl (#91376)
|
2023-02-03 15:06:40 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
[Fake Tensor] Allow fake meta by default, delete unused ctor args (#93993)
|
2023-02-03 09:23:38 +00:00 |
|
test_foreach.py
|
[mta] Backward of unary foreach functions (#89591)
|
2023-01-23 08:28:06 +00:00 |
|
test_function_schema.py
|
[frontend] Expose real_type getter for torch.Argument (#91938)
|
2023-01-12 01:26:50 +00:00 |
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
[Reland] Refactor stack_trace preservation for node meta preservation (#90803) (#92400)
|
2023-01-30 23:30:43 +00:00 |
|
test_futures.py
|
|
|
|
test_fx.py
|
feat(fx): make_fx should be aware of functions wrapped with @fx.wrap (#93273)
|
2023-02-02 01:57:52 +00:00 |
|
test_fx_experimental.py
|
|
|
|
test_fx_passes.py
|
|
|
|
test_fx_reinplace_pass.py
|
*_scatter ops should preserve input stride/storage_offset (#91029)
|
2022-12-22 19:41:53 +00:00 |
|
test_hub.py
|
|
|
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
|
|
|
test_itt.py
|
|
|
|
test_jit.py
|
[jit] Support Awaitable type (#90863)
|
2023-01-30 17:38:59 +00:00 |
|
test_jit_autocast.py
|
|
|
|
test_jit_cuda_fuser.py
|
[CUDA] Drop CUDA < 11.0 test flags (#92605)
|
2023-01-24 04:34:06 +00:00 |
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
Revert "Disable torch_jit_fuser_te for dynamo CI (#92945)"
|
2023-01-26 21:41:17 +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
|
Remove deprecated torch.symeig (#70988)
|
2023-01-31 11:59:11 +00:00 |
|
test_license.py
|
|
|
|
test_linalg.py
|
Remove deprecated torch.symeig (#70988)
|
2023-01-31 11:59:11 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
Support per-parameter test decoration (#91658)
|
2023-01-04 21:08:32 +00:00 |
|
test_maskedtensor.py
|
|
|
|
test_matmul_cuda.py
|
[CUDA] Drop CUDA < 11.0 test flags (#92605)
|
2023-01-24 04:34:06 +00:00 |
|
test_meta.py
|
Add meta registration for bucketize (#93893)
|
2023-02-02 21:03:08 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
Enable oneDNN implementation in LSTM op (#91158)
|
2023-01-18 04:41:18 +00:00 |
|
test_mkldnn_fusion.py
|
[Re-open 90266] [inductor] weight prepack for _convolution_transpose_pointwise (#91955)
|
2023-01-31 13:28:57 +00:00 |
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
[Reland] Clean Up MobileOptimizerType Rewrite Flags Public API and Documentation (#92081)
|
2023-01-14 17:06:00 +00:00 |
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
|
|
|
test_modules.py
|
Revert "Update Module.__setattr__ to respect property setters (#92044)"
|
2023-01-21 02:39:21 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
[nn] zero_grad() set_to_none default True (#92731)
|
2023-01-26 01:04:28 +00:00 |
|
test_multiprocessing.py
|
Increase test multiprocessing waiting time (#93183)
|
2023-01-28 07:59:59 +00:00 |
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
|
|
|
test_namedtuple_return_api.py
|
[SDPA] Wire up FlashAttention's backward (#92917)
|
2023-02-02 04:02:30 +00:00 |
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
|
|
|
test_nestedtensor.py
|
[SDPA] Update SDPA API and make function Public (#92189)
|
2023-01-23 20:50:46 +00:00 |
|
test_nn.py
|
Added super init to Module (#91819)
|
2023-02-01 22:17:59 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
|
|
|
test_nvfuser_dynamo.py
|
Refactor dynamo training backends (#93409)
|
2023-02-03 03:07:15 +00:00 |
|
test_nvfuser_frontend.py
|
[NVFUSER] refactor nvfuser build (#89621)
|
2023-01-26 02:50:44 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
[Fake Tensor] Allow fake meta by default, delete unused ctor args (#93993)
|
2023-02-03 09:23:38 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
[mta][foreach] Implement fused adamw (#88015)
|
2023-02-01 19:32:29 +00:00 |
|
test_overrides.py
|
Retire unsafe sparse tensor constructors in Python API (#91331)
|
2023-01-18 08:55:22 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
[NVFUSER] refactor nvfuser build (#89621)
|
2023-01-26 02:50:44 +00:00 |
|
test_proxy_tensor.py
|
Add meta registration for bucketize (#93893)
|
2023-02-02 21:03:08 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
[jit] Support Awaitable type (#90863)
|
2023-01-30 17:38:59 +00:00 |
|
test_python_dispatch.py
|
Correctly restore pybind11 error_already_set (#93238)
|
2023-01-30 16:43:01 +00:00 |
|
test_pytree.py
|
|
|
|
test_quantization.py
|
[quant][pt2e] Add a test to confirm we can set qconfig according to module_name (#91977)
|
2023-01-12 21:59:02 +00:00 |
|
test_reductions.py
|
Logcumsumexp for CPU (#93153)
|
2023-01-27 22:29:33 +00:00 |
|
test_scatter_gather_ops.py
|
optimize gather performance for gnn usage on CPU (#87586)
|
2023-01-12 00:43:43 +00:00 |
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
Fix BC-breaking change introduced by #91499 (#93091)
|
2023-01-27 03:58:35 +00:00 |
|
test_serialization.py
|
[follow-up] Python Attr Serialization (#88913)
|
2023-01-13 17:38:51 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
Added vectorized horizontal flip path for channels last for NcHW (#91806)
|
2023-01-23 20:15:30 +00:00 |
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
fix sort crash when the input is expanded scalar (#91752)
|
2023-01-09 02:02:56 +00:00 |
|
test_sparse.py
|
sparse_mask: extend lhs to sparse COO tensors (#92248)
|
2023-02-01 09:00:07 +00:00 |
|
test_sparse_csr.py
|
torch.triangular_solve for CSR: materialize diagonal elements when unitriangular=True. (#93352)
|
2023-01-31 16:33:57 +00:00 |
|
test_spectral_ops.py
|
|
|
|
test_stateless.py
|
Deprecate torch.nn.utils.stateless.functional_call (#92280)
|
2023-01-18 14:26:25 +00:00 |
|
test_static_runtime.py
|
|
|
|
test_subclass.py
|
|
|
|
test_tensor_creation_ops.py
|
asarray: Add support for NumPy scalars (#90914)
|
2023-01-24 08:09:30 +00:00 |
|
test_tensorboard.py
|
|
|
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
Implement hybrid sparse to/from dense conversions. (#90177)
|
2023-01-12 03:31:30 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
cauchy_ few fixes (1) check gamma > 0 (2) better dtype error log (#93314)
|
2023-02-03 11:56:28 +00:00 |
|
test_transformers.py
|
[SDPA] Wire up FlashAttention's backward (#92917)
|
2023-02-02 04:02:30 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
|
|
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
|
|
|
test_utils.py
|
Add utility report_compile_source_on_error (#91069)
|
2023-01-11 22:54:46 +00:00 |
|
test_view_ops.py
|
Deprecate .mT,.T,.mH,.H on 0D tensors (#92143)
|
2023-01-17 16:54:35 +00:00 |
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
