| .. |
|
_nvfuser
|
|
|
|
ao/sparsity
|
enable ASAN on some tests (#103647)
|
2023-06-28 02:17:14 +00:00 |
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
DDP + C10D sparse all_reduce changes (#103916) (#104256)
|
2023-06-28 00:37:52 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
[functorch] disable C++ Function under functorch transforms (#103957)
|
2023-06-23 11:01:44 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[PTD][TP] Add BWD support for colwise embedding sharding (#104820)
|
2023-07-10 22:33:20 +00:00 |
|
distributions
|
Fix Dirichlet.log_prob() when x=0 and alpha=1 (#103605)
|
2023-06-15 16:16:50 +00:00 |
|
dynamo
|
Fix mod semantics for Z3Ops. (#104827)
|
2023-07-10 23:35:04 +00:00 |
|
edge
|
[Specialized Kernel] Update yaml syntax to use kernel instead of dispatch (#104070)
|
2023-06-27 09:53:20 +00:00 |
|
error_messages
|
|
|
|
expect
|
[inductor] addmm + ReLU / GELU fusion pass (#104132)
|
2023-07-10 16:44:14 +00:00 |
|
export
|
[export][serde] Hookup export upgrader with TorchScript upgrader entries (#104227)
|
2023-07-06 16:57:36 +00:00 |
|
forward_backward_compatibility
|
DDP + C10D sparse all_reduce changes (#103916) (#104256)
|
2023-06-28 00:37:52 +00:00 |
|
functorch
|
[HigherOrderOp] Remove _deprecated_global_ns from cond (#104380)
|
2023-07-07 17:13:09 +00:00 |
|
fx
|
|
|
|
inductor
|
Enable running compiled optimizers in CI (#104888)
|
2023-07-10 23:45:41 +00:00 |
|
jit
|
[jit] Fix inspect.get_annotations usage in python >= 3.10 (#104485)
|
2023-07-06 00:37:47 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
|
|
|
mobile
|
|
|
|
nn
|
Added option to always call nn.Module global/non-global forward hooks (#104278)
|
2023-07-10 18:58:07 +00:00 |
|
onnx
|
[ONNX] Fix exported onnx initializer name (#104741)
|
2023-07-08 04:36:33 +00:00 |
|
onnx_caffe2
|
|
|
|
optim
|
[foreach][AdamW] Minimize intermediates to save peak memory (#104898)
|
2023-07-10 23:40:52 +00:00 |
|
package
|
[fx] make fx.wrap idempotent (#104838)
|
2023-07-09 20:57:46 +00:00 |
|
profiler
|
Revert "[optim] use lerp whenever possible (#104796)"
|
2023-07-10 09:36:41 +00:00 |
|
quantization
|
[quant] Skip some x86 quantizer tests for now due to time out (#104666)
|
2023-07-06 17:34:13 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
typing
|
|
|
|
_test_bazel.py
|
|
|
|
allowlist_for_publicAPI.json
|
|
|
|
conftest.py
|
Do not collect and skip non-disabled tests when rerunning disabled tests (#102107)
|
2023-05-27 12:10:36 +00:00 |
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
Re-land: Turn translation validation on for tests and accuracy runs by default. (#104467)
|
2023-07-05 19:01:50 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
Add autocast support for IPU (#103890)
|
2023-06-22 15:38:45 +00:00 |
|
test_autograd.py
|
[autograd Function] Add private API to not materialize grads for non-differentiable outputs (#104291)
|
2023-07-08 14:53:54 +00:00 |
|
test_binary_ufuncs.py
|
Enable bitwise shift operations tests (#97150)
|
2023-07-06 15:32:57 +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
|
|
|
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
Use random inputs for mps extension tests (#104597)
|
2023-07-06 07:14:56 +00:00 |
|
test_cpp_extensions_jit.py
|
Use random inputs for mps extension tests (#104597)
|
2023-07-06 07:14:56 +00:00 |
|
test_cpp_extensions_open_device_registration.py
|
add generic func to get function defined in custom device module (#99048)
|
2023-06-30 20:02:44 +00:00 |
|
test_cuda.py
|
Refactor multigpu tests to test_cuda_multigpu (#104059)
|
2023-06-27 05:32:05 +00:00 |
|
test_cuda_expandable_segments.py
|
|
|
|
test_cuda_multigpu.py
|
Refactor multigpu tests to test_cuda_multigpu (#104059)
|
2023-06-27 05:32:05 +00:00 |
|
test_cuda_nvml_based_avail.py
|
|
|
|
test_cuda_primary_ctx.py
|
[BE] Use TEST_MULTIGPU from common_cuda.py (#103982)
|
2023-06-22 00:07:44 +00:00 |
|
test_cuda_sanitizer.py
|
|
|
|
test_cuda_trace.py
|
|
|
|
test_custom_op_testing.py
|
[operator_compile_check] Add FakeTensor testing (#103595)
|
2023-06-16 16:55:51 +00:00 |
|
test_dataloader.py
|
|
|
|
test_datapipe.py
|
|
|
|
test_decomp.py
|
[decomp] Add test tracking core ATen operators (#104262)
|
2023-07-04 16:41:44 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
add additional stream priority for cuda streams (#101956)
|
2023-05-27 02:36:16 +00:00 |
|
test_dynamic_shapes.py
|
Support bit shifting SymInts (#104318)
|
2023-07-05 18:35:57 +00:00 |
|
test_expanded_weights.py
|
tf32 context fixes for various tests (#103137)
|
2023-06-15 02:33:12 +00:00 |
|
test_fake_tensor.py
|
Fix broken meta registration for torch.full (#104451)
|
2023-06-30 05:14:52 +00:00 |
|
test_flop_counter.py
|
De-register forward hooks upon exiting flop counter context (#103744)
|
2023-06-20 19:34:02 +00:00 |
|
test_foreach.py
|
Revert "Add forward mode AD to out-place foreach functions (#102409)"
|
2023-06-30 22:46:57 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
Upgrade LoggingTensor mode and add traceback collection (#103734)
|
2023-06-21 22:04:30 +00:00 |
|
test_functionalization_of_rng_ops.py
|
|
|
|
test_futures.py
|
Enable mypy allow redefinition (#102046)
|
2023-05-24 07:05:30 +00:00 |
|
test_fx.py
|
Preserve original co_filename when FX symbolic_trace (#103885)
|
2023-07-05 22:00:05 +00:00 |
|
test_fx_experimental.py
|
|
|
|
test_fx_passes.py
|
|
|
|
test_fx_reinplace_pass.py
|
|
|
|
test_hub.py
|
|
|
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
[Dynamo] Fix lineinfo generation on PY3.11+ (#103525)
|
2023-06-14 05:41:43 +00:00 |
|
test_itt.py
|
|
|
|
test_jit.py
|
|
|
|
test_jit_autocast.py
|
|
|
|
test_jit_cuda_fuser.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
|
[BE] Deprecate has_XYZ attributes (#103279)
|
2023-06-10 05:17:17 +00:00 |
|
test_jit_profiling.py
|
|
|
|
test_jit_simple.py
|
|
|
|
test_jit_string.py
|
|
|
|
test_jiterator.py
|
|
|
|
test_kernel_launch_checks.py
|
|
|
|
test_legacy_vmap.py
|
Make index_add_ error if input source shape is wrong (#100321)
|
2023-06-08 06:51:10 +00:00 |
|
test_license.py
|
|
|
|
test_linalg.py
|
Revert "[ROCm] reduce tolerance for triangular solve with well_conditioned set to True (#104425)"
|
2023-07-03 04:18:04 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
Tighten FakeTensor reentrancy asserts, add debugging (#102091)
|
2023-05-24 05:37:51 +00:00 |
|
test_matmul_cuda.py
|
|
|
|
test_meta.py
|
[pt2] add metas for multilabel_margin_loss ops (#104388)
|
2023-07-05 13:42:22 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
[BE] Deprecate has_XYZ attributes (#103279)
|
2023-06-10 05:17:17 +00:00 |
|
test_mkldnn_fusion.py
|
inductor: reduce compile time for cpu backend by reducing weight conversion (#104402)
|
2023-07-06 13:44:50 +00:00 |
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_module_init.py
|
|
|
|
test_modules.py
|
Add partial derivative unit tests (#103809)
|
2023-06-25 00:36:10 +00:00 |
|
test_monitor.py
|
|
|
|
test_mps.py
|
Fix activation checkpoint for mps (#104787)
|
2023-07-08 14:57:05 +00:00 |
|
test_multiprocessing.py
|
|
|
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
|
|
|
test_namedtuple_return_api.py
|
|
|
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
|
|
|
test_nestedtensor.py
|
|
|
|
test_nn.py
|
Use int64_t accordingly in cunn_SoftMaxBackward to avoid int overflow (#104270)
|
2023-06-30 21:39:46 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
|
|
|
test_nvfuser_dynamo.py
|
|
|
|
test_nvfuser_frontend.py
|
|
|
|
test_openmp.py
|
|
|
|
test_ops.py
|
enable ASAN on some tests (#103647)
|
2023-06-28 02:17:14 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
[custom_op] explicit autograd API (#101824)
|
2023-05-23 18:31:29 +00:00 |
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
|
|
|
test_overrides.py
|
|
|
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
Upgrade LoggingTensor mode and add traceback collection (#103734)
|
2023-06-21 22:04:30 +00:00 |
|
test_proxy_tensor.py
|
Re-land: Turn translation validation on for tests and accuracy runs by default. (#104467)
|
2023-07-05 19:01:50 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
[AMP] Support XLA:TPU (#96370)
|
2023-06-23 19:46:42 +00:00 |
|
test_python_dispatch.py
|
Upgrade LoggingTensor mode and add traceback collection (#103734)
|
2023-06-21 22:04:30 +00:00 |
|
test_pytree.py
|
Simulate treespec flattening/unflattening (#101896)
|
2023-06-23 10:53:15 +00:00 |
|
test_quantization.py
|
[Quant][PT2E]Enable x86 inductor quantizer (#98730)
|
2023-06-17 06:10:23 +00:00 |
|
test_reductions.py
|
|
|
|
test_scatter_gather_ops.py
|
Revert "Optimize scatter_add/scatter_reduce in BFloat16/Half data type in CPU backend (#103427)"
|
2023-07-06 17:02:03 +00:00 |
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
Enable more tests (#104437)
|
2023-06-30 16:26:11 +00:00 |
|
test_serialization.py
|
Add functions to get and set default endianness in load() functions (#101973)
|
2023-07-06 20:12:56 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
Make 1D integer sorting work in parallel (#100081)
|
2023-06-02 07:41:28 +00:00 |
|
test_sparse.py
|
sparse_mask: backward support for sparse lhs (take 2) (#104341)
|
2023-07-03 14:12:44 +00:00 |
|
test_sparse_csr.py
|
Revert "Optimize scatter_add/scatter_reduce in BFloat16/Half data type in CPU backend (#103427)"
|
2023-07-06 17:02:03 +00:00 |
|
test_sparse_semi_structured.py
|
[core][pruning][sparse][feature] SparseSemiStructured tensor subclass (#102135)
|
2023-06-27 19:21:06 +00:00 |
|
test_spectral_ops.py
|
|
|
|
test_stateless.py
|
|
|
|
test_static_runtime.py
|
|
|
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
|
|
|
test_tensor_creation_ops.py
|
|
|
|
test_tensorboard.py
|
|
|
|
test_tensorexpr.py
|
Generate nearbyint for Round in tensorexpr llvm codegen, match torch.round result (#104430)
|
2023-07-07 01:47:46 +00:00 |
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
Make torch._dynamo lazy-importable (#104368)
|
2023-06-29 00:51:59 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Add deterministic path for Tensor.resize_ (#104300)
|
2023-07-07 00:22:13 +00:00 |
|
test_transformers.py
|
REDO of dropout support for mem eff #102038 (#103704)
|
2023-06-26 23:05:03 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
|
|
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
enable ASAN on some tests (#103647)
|
2023-06-28 02:17:14 +00:00 |
|
test_utils.py
|
Add torch._utils.render_call, improve printoptions (#102623)
|
2023-05-31 22:08:04 +00:00 |
|
test_view_ops.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|