| .. |
|
ao/sparsity
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
autograd
|
Autograd attaches logging hooks only in debug level (#116522)
|
2024-01-02 20:06:18 +00:00 |
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
Initial torchbind support in PT2 (#117697)
|
2024-01-19 06:28:20 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
Add function to materialize COW storages (#117053)
|
2024-01-10 15:34:16 +00:00 |
|
custom_backend
|
|
|
|
custom_operator
|
|
|
|
distributed
|
[c10d] Remove health check (#117699)
|
2024-01-18 02:14:49 +00:00 |
|
distributions
|
Add bfloat16 CUDA support to binomial distribution (#116932)
|
2024-01-07 19:50:10 +00:00 |
|
dynamo
|
[Inductor] Use codegen reference for buffer to string (#117838)
|
2024-01-19 20:18:53 +00:00 |
|
edge
|
|
|
|
error_messages
|
|
|
|
expect
|
Revert "[CUDNN][SDPA] Experimental cuDNN Flash Attention v2 Inference (#115663)"
|
2024-01-18 23:40:30 +00:00 |
|
export
|
[export] Support preserving submodule callling convention in non-strict export (#117796)
|
2024-01-19 17:16:45 +00:00 |
|
forward_backward_compatibility
|
Add out_dtype support for sparse semi-structured CUTLASS back-end (#116519)
|
2024-01-03 16:23:17 +00:00 |
|
functorch
|
[HigherOrderOp] Add while_loop support (#116823)
|
2024-01-19 00:35:26 +00:00 |
|
fx
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
inductor
|
Log stack trace of mutated idx (#117720)
|
2024-01-19 21:38:44 +00:00 |
|
jit
|
Run test_jit with PYTORCH_TEST_WITH_DYNAMO=1 in CI (#117765)
|
2024-01-19 13:42:41 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
Replaced CHECK with TORCH_CHECK in order to not abort, but throw a Ru… (#117653)
|
2024-01-18 07:47:22 +00:00 |
|
mobile
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
nn
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
onnx
|
[export] Error on not pytree-flattened nodes (#117598)
|
2024-01-19 17:13:39 +00:00 |
|
onnx_caffe2
|
|
|
|
optim
|
[mta] Fused SGD (#116585)
|
2024-01-16 23:54:38 +00:00 |
|
package
|
|
|
|
profiler
|
accelerate binary_cross_entropy_with_logits by using log_sigmoid operator (#115539)
|
2024-01-19 14:56:43 +00:00 |
|
quantization
|
[Quant] Enable QConv2d with hardswish post op (#117487)
|
2024-01-19 13:24:06 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
torch_np
|
Add super().setUp() to TestFFT1D (#117329)
|
2024-01-12 07:47:01 +00:00 |
|
typing
|
|
|
|
_test_bazel.py
|
|
|
|
allowlist_for_publicAPI.json
|
Initial torchbind support in PT2 (#117697)
|
2024-01-19 06:28:20 +00:00 |
|
conftest.py
|
Revert "Reduce pytest prints (#117069)"
|
2024-01-19 23:07:51 +00:00 |
|
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
|
Revert "Reduce pytest prints (#117069)"
|
2024-01-19 23:07:51 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
Remove incorrect usages of skipIfTorchDynamo (#117114)
|
2024-01-10 22:25:31 +00:00 |
|
test_autograd.py
|
[codemod] markDynamoStrictTest batch 22 (#117729)
|
2024-01-18 16:59:26 +00:00 |
|
test_autograd_fallback.py
|
|
|
|
test_binary_ufuncs.py
|
[codemod] markDynamoStrictTest batch 14 (#117133)
|
2024-01-11 04:28:57 +00:00 |
|
test_bundled_images.py
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
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
|
|
|
|
test_cpp_extensions_jit.py
|
|
|
|
test_cpp_extensions_open_device_registration.py
|
|
|
|
test_cuda.py
|
[mta] Fused SGD (#116585)
|
2024-01-16 23:54:38 +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
|
Add config toggle suggestions for data-dependent/dynamic output shape (#114337)
|
2024-01-05 14:01:01 +00:00 |
|
test_dataloader.py
|
[BE]: Enable RUFF PERF402 and apply fixes (#115505)
|
2023-12-20 18:01:24 +00:00 |
|
test_datapipe.py
|
Remove incorrect usages of skipIfTorchDynamo (#117114)
|
2024-01-10 22:25:31 +00:00 |
|
test_decomp.py
|
[PyTorch] SDPA decomp: actually use attn_mask (#117579)
|
2024-01-17 10:26:43 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
Improve uint{16,32,64} dlpack/numpy compatibility (#116808)
|
2024-01-11 17:01:54 +00:00 |
|
test_dynamic_shapes.py
|
[fx][inductor] Add statically_known_true utility for SymBool (#117359)
|
2024-01-15 18:01:10 +00:00 |
|
test_expanded_weights.py
|
[Dynamo] Use isinstance rather than istype when check if python module type (#117022)
|
2024-01-15 23:25:30 +00:00 |
|
test_fake_tensor.py
|
Initial implementation of FakeTensor caching (#113873)
|
2024-01-17 20:38:54 +00:00 |
|
test_flop_counter.py
|
Re-add initial Flash Attention support on ROCM (#115981)
|
2024-01-04 22:21:31 +00:00 |
|
test_foreach.py
|
Refactor some tests by using TEST_CUDA & TEST_MULTIGPU instead (#116083)
|
2024-01-03 08:53:59 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
Enable reverse view_funcs by default for python subclasses (#116512)
|
2024-01-05 16:48:12 +00:00 |
|
test_functionalization_of_rng_ops.py
|
|
|
|
test_futures.py
|
|
|
|
test_fx.py
|
[codemod] markDynamoStrictTest batch 22 (#117729)
|
2024-01-18 16:59:26 +00:00 |
|
test_fx_experimental.py
|
Revert "[codemod] markDynamoStrictTest batch 16 (#117218)"
|
2024-01-12 03:06:20 +00:00 |
|
test_fx_passes.py
|
|
|
|
test_fx_reinplace_pass.py
|
|
|
|
test_hub.py
|
|
|
|
test_import_stats.py
|
|
|
|
test_indexing.py
|
[BE]: Apply FURB118 (prev): replaces unnecessary lambdas with operator. (#116027)
|
2023-12-20 19:35:08 +00:00 |
|
test_itt.py
|
|
|
|
test_jit.py
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
test_jit_autocast.py
|
|
|
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
[BE]: Enable F821 and fix bugs (#116579)
|
2024-01-01 08:40:46 +00:00 |
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
[BE]: Enable F821 and fix bugs (#116579)
|
2024-01-01 08:40:46 +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
|
Skip some slow tests (under Dynamo) (#117389)
|
2024-01-12 22:18:07 +00:00 |
|
test_license.py
|
|
|
|
test_linalg.py
|
Workaround a cusolver bug on CUDA < 12.1 in triangular_solve (#117636)
|
2024-01-19 12:42:37 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
|
|
|
test_matmul_cuda.py
|
|
|
|
test_meta.py
|
Added meta function for _upsample_bicubic2d_aa (#117347)
|
2024-01-16 23:33:55 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
[BE]: Enable RUFF PERF402 and apply fixes (#115505)
|
2023-12-20 18:01:24 +00:00 |
|
test_mkldnn_fusion.py
|
[codemod] markDynamoStrictTest batch 22 (#117729)
|
2024-01-18 16:59:26 +00:00 |
|
test_mkldnn_verbose.py
|
|
|
|
test_mobile_optimizer.py
|
|
|
|
test_model_dump.py
|
|
|
|
test_model_exports_to_core_aten.py
|
|
|
|
test_module_init.py
|
|
|
|
test_modules.py
|
|
|
|
test_monitor.py
|
Change dynamo_test_failures.py to silently run skipped tests (#117401)
|
2024-01-17 02:48:19 +00:00 |
|
test_mps.py
|
add Half support for interpolate operators on CPU (#105648)
|
2024-01-18 09:07:16 +00:00 |
|
test_multiprocessing.py
|
|
|
|
test_multiprocessing_spawn.py
|
|
|
|
test_namedtensor.py
|
|
|
|
test_namedtuple_return_api.py
|
Revert "[CUDNN][SDPA] Experimental cuDNN Flash Attention v2 Inference (#115663)"
|
2024-01-18 23:40:30 +00:00 |
|
test_native_functions.py
|
|
|
|
test_native_mha.py
|
Re-add initial Flash Attention support on ROCM (#115981)
|
2024-01-04 22:21:31 +00:00 |
|
test_nestedtensor.py
|
[Nested Tensor]Support SDPA math fallback for jagged layout nested tensor (#116445)
|
2024-01-12 17:30:40 +00:00 |
|
test_nn.py
|
Add Half support for masked_softmax on CPU (#117028)
|
2024-01-18 08:59:20 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
Improve uint{16,32,64} dlpack/numpy compatibility (#116808)
|
2024-01-11 17:01:54 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
Initial implementation of FakeTensor caching (#113873)
|
2024-01-17 20:38:54 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
Migrate the rest of state_dict testing to OptimizerInfo (#117186)
|
2024-01-12 22:32:37 +00:00 |
|
test_out_dtype_op.py
|
|
|
|
test_overrides.py
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
|
|
|
test_proxy_tensor.py
|
Always accept 0-d scalar tensors as int, even if __index__ fails (#117451)
|
2024-01-14 14:39:46 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
[ONNX][dynamo_export] Decomposition skips using custom operator (#117314)
|
2024-01-18 22:19:28 +00:00 |
|
test_python_dispatch.py
|
[BE]: Update flake8 to v6.1.0 and fix lints (#116591)
|
2024-01-03 06:04:44 +00:00 |
|
test_pytree.py
|
[pytree] add key path api (#116786)
|
2024-01-17 07:24:35 +00:00 |
|
test_quantization.py
|
|
|
|
test_reductions.py
|
Improve uint{16,32,64} dlpack/numpy compatibility (#116808)
|
2024-01-11 17:01:54 +00:00 |
|
test_scatter_gather_ops.py
|
|
|
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
Support pathlib.Path as input to torch.load when mmap=True (#116104)
|
2023-12-28 22:54:11 +00:00 |
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
Removed an internal assertion for the optional stable value and inste… (#117414)
|
2024-01-17 02:25:21 +00:00 |
|
test_sparse.py
|
Add batched sparse CSR/CSC/BSR/BSC to sparse COO conversion support (#116206)
|
2024-01-07 19:42:02 +00:00 |
|
test_sparse_csr.py
|
[codemod] markDynamoStrictTest batch 14 (#117133)
|
2024-01-11 04:28:57 +00:00 |
|
test_sparse_semi_structured.py
|
Add out_dtype support for sparse semi-structured CUTLASS back-end (#116519)
|
2024-01-03 16:23:17 +00:00 |
|
test_spectral_ops.py
|
FFT: Handle noop fftn calls gracefully (#117368)
|
2024-01-12 20:16:50 +00:00 |
|
test_stateless.py
|
|
|
|
test_static_runtime.py
|
|
|
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
|
|
|
test_tensor_creation_ops.py
|
Enable some uint{16,32,64} tests that are working (#116809)
|
2024-01-15 02:25:21 +00:00 |
|
test_tensorboard.py
|
|
|
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
Back out "[Kineto] Initialize libkineto profilers during torch init process during pybind set-up (#112623)" (#116201)
|
2023-12-21 16:32:19 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
add _amp_foreach_non_finite_check_and_unscale_cpu_ and _amp_update_scale_cpu_ kernels on CPU (#109281)
|
2024-01-16 15:25:08 +00:00 |
|
test_transformers.py
|
Revert "[CUDNN][SDPA] Experimental cuDNN Flash Attention v2 Inference (#115663)"
|
2024-01-18 23:40:30 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
|
|
|
test_type_promotion.py
|
Improve uint{16,32,64} dlpack/numpy compatibility (#116808)
|
2024-01-11 17:01:54 +00:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
Remove incorrect usages of skipIfTorchDynamo (#117114)
|
2024-01-10 22:25:31 +00:00 |
|
test_utils.py
|
|
|
|
test_view_ops.py
|
|
|
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
