| .. |
|
ao/sparsity
|
|
|
|
autograd
|
|
|
|
backends/xeon
|
|
|
|
benchmark_utils
|
|
|
|
bottleneck_test
|
|
|
|
cpp
|
Fix functional::smooth_l1_loss signatures to not override beta (#109798)
|
2023-10-11 21:37:37 +00:00 |
|
cpp_api_parity
|
|
|
|
cpp_extensions
|
|
|
|
custom_backend
|
|
|
|
custom_operator
|
Add torch.ops.import_module (#110090)
|
2023-09-27 13:56:47 +00:00 |
|
distributed
|
fix: do not reshard parameters twice (#110948)
|
2023-10-12 15:09:33 +00:00 |
|
distributions
|
[BE]: Enable some basic pytest style rules (#110362)
|
2023-10-05 21:40:43 +00:00 |
|
dynamo
|
Move export.constrain_as_* to torch._constrain_as_* (#110757)
|
2023-10-12 05:37:44 +00:00 |
|
edge
|
[executorch][kernel reg] Allow kernel manual registration (#110086)
|
2023-09-27 16:04:20 +00:00 |
|
error_messages
|
|
|
|
expect
|
Add CUTLASS-based support for mixed dtypes matrix multiplication (#110981)
|
2023-10-11 21:47:52 +00:00 |
|
export
|
Revert "[export] Fix graph signature data model to list of specs. (#111017)"
|
2023-10-12 09:52:33 +00:00 |
|
forward_backward_compatibility
|
Flash Attention v2 (#105602)
|
2023-09-13 13:59:05 +00:00 |
|
functorch
|
AOTAutograd: Go down inference path if no outputs require grad (#111011)
|
2023-10-11 20:59:47 +00:00 |
|
fx
|
[fx][subgraph_matcher] Add a matcher that supports name to node map (#110743)
|
2023-10-10 22:21:24 +00:00 |
|
inductor
|
Move export.constrain_as_* to torch._constrain_as_* (#110757)
|
2023-10-12 05:37:44 +00:00 |
|
jit
|
update type() calling to not use unneeded device (#110163)
|
2023-09-28 17:34:46 +00:00 |
|
jit_hooks
|
|
|
|
lazy
|
[LTC] correct AdaptiveAvgPool3d channel dim index for shape inference (#109822)
|
2023-09-29 22:54:12 +00:00 |
|
mobile
|
Add a workflow to release Android binaries (#110976)
|
2023-10-11 00:19:33 +00:00 |
|
nn
|
Revert "[ROCm] Remove PYTORCH_MIOPEN_SUGGEST_NHWC flag (#90725)" (#110319)
|
2023-10-03 19:14:47 +00:00 |
|
onnx
|
[ONNX] Create onnxscript-torchlib specific xfails/skips for fx tests (#110536)
|
2023-10-05 00:39:05 +00:00 |
|
onnx_caffe2
|
Revert "[ONNX] Remove the depreacated function _export (#109763)"
|
2023-09-25 17:47:21 +00:00 |
|
optim
|
feat(optim): add SGD sparse multitensor to testing path (#110562)
|
2023-10-06 07:48:25 +00:00 |
|
package
|
|
|
|
profiler
|
[profiler] Show shapes for lists of tensors in chrome traces #109263 (#109751)
|
2023-09-26 01:03:54 +00:00 |
|
quantization
|
[quant][pt2e][be] Move xnnpack quantizer tests to separate file (#111004)
|
2023-10-12 01:16:05 +00:00 |
|
scripts
|
|
|
|
test_img
|
|
|
|
torch_np
|
[BE]: Enable some basic pytest style rules (#110362)
|
2023-10-05 21:40:43 +00:00 |
|
typing
|
Improve type annotations for jit.script (#108782)
|
2023-09-13 19:20:25 +00:00 |
|
_test_bazel.py
|
|
|
|
allowlist_for_publicAPI.json
|
Fix public binding check to check all submodules (#110601)
|
2023-10-08 00:36:31 +00:00 |
|
conftest.py
|
|
|
|
create_dummy_torchscript_model.py
|
|
|
|
delete.py
|
|
|
|
HowToWriteTestsUsingFileCheck.md
|
|
|
|
linear.py
|
|
|
|
load_torchscript_model.py
|
|
|
|
minioptest_failures_dict.json
|
[generate_opcheck_tests] Enable using same failures_dict for multiple testclasses (#110164)
|
2023-09-28 17:56:45 +00:00 |
|
mkl_verbose.py
|
|
|
|
mkldnn_verbose.py
|
|
|
|
run_doctests.sh
|
|
|
|
run_test.py
|
Add test_jit_cuda_fuser to ROCM_BLOCKLIST (#110440)
|
2023-10-06 08:47:15 +00:00 |
|
simulate_nccl_errors.py
|
|
|
|
test_ao_sparsity.py
|
|
|
|
test_autocast.py
|
|
|
|
test_autograd.py
|
Allow specifiying inputs as GradientEdge in autograd APIs (#110867)
|
2023-10-12 04:08:44 +00:00 |
|
test_autograd_fallback.py
|
|
|
|
test_binary_ufuncs.py
|
|
|
|
test_bundled_images.py
|
|
|
|
test_bundled_inputs.py
|
|
|
|
test_comparison_utils.py
|
|
|
|
test_compile_benchmark_util.py
|
|
|
|
test_complex.py
|
Don't fastpath conj copy when conj/neg bit mismatch (#108881)
|
2023-09-08 20:44:43 +00:00 |
|
test_content_store.py
|
|
|
|
test_cpp_api_parity.py
|
|
|
|
test_cpp_extensions_aot.py
|
add torch.device to python type (#108116)
|
2023-09-18 02:20:30 +00:00 |
|
test_cpp_extensions_jit.py
|
fix TEST_ROCM definition to disable test_jit_cudnn_extension on rocm (#110385)
|
2023-10-04 20:02:02 +00:00 |
|
test_cpp_extensions_open_device_registration.py
|
|
|
|
test_cuda.py
|
Fix typo under test directory (#110826)
|
2023-10-08 20:52: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
|
[generate_opcheck_tests] Add some reasonable defaults (#110977)
|
2023-10-11 14:28:05 +00:00 |
|
test_dataloader.py
|
StackDataset batched sampling (#110694)
|
2023-10-10 22:05:51 +00:00 |
|
test_datapipe.py
|
Disable compile for massive data pipe test (#109063)
|
2023-09-12 00:15:52 +00:00 |
|
test_decomp.py
|
Revert "[core ATen IR] Add decompositions for max, min, var_mean (#110906)"
|
2023-10-11 11:41:21 +00:00 |
|
test_deploy.py
|
|
|
|
test_determination.py
|
|
|
|
test_dispatch.py
|
|
|
|
test_dlpack.py
|
Add DLPack bool support (#108486)
|
2023-09-08 17:55:33 +00:00 |
|
test_dynamic_shapes.py
|
Make NestedTensor compilable with eager backend (#109171)
|
2023-10-11 04:47:10 +00:00 |
|
test_expanded_weights.py
|
|
|
|
test_fake_tensor.py
|
Define the public API for torch.distributed.fsdp (#109922)
|
2023-09-28 02:15:58 +00:00 |
|
test_flop_counter.py
|
Add option to flop counter formula registration to get raw values (#110591)
|
2023-10-05 21:14:41 +00:00 |
|
test_foreach.py
|
[foreach] Fix 0-size handling for real for real (#109402)
|
2023-09-26 17:38:20 +00:00 |
|
test_function_schema.py
|
|
|
|
test_functional_autograd_benchmark.py
|
|
|
|
test_functional_optim.py
|
|
|
|
test_functionalization.py
|
Make FunctionalTensor subclass to be more like functorch (interaction with ZeroTensor + Conjugate key) (#109023)
|
2023-09-22 07:09:04 +00:00 |
|
test_functionalization_of_rng_ops.py
|
|
|
|
test_futures.py
|
|
|
|
test_fx.py
|
Replace node.meta source_fn with source_fn_stack (#108595)
|
2023-09-28 18:18:36 +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
|
Add torch.unravel_index (#110580)
|
2023-10-12 00:55:51 +00:00 |
|
test_itt.py
|
|
|
|
test_jit.py
|
|
|
|
test_jit_autocast.py
|
|
|
|
test_jit_disabled.py
|
|
|
|
test_jit_fuser.py
|
|
|
|
test_jit_fuser_legacy.py
|
|
|
|
test_jit_fuser_te.py
|
[jit] Skip NNAPI, test_ivalue, CPU NNC tests in fbcode (#108937)
|
2023-09-11 22:42:30 +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
|
|
|
|
test_linalg.py
|
Revise def of contiguity in bmm (#110811)
|
2023-10-10 06:48:08 +00:00 |
|
test_logging.py
|
|
|
|
test_masked.py
|
|
|
|
test_maskedtensor.py
|
|
|
|
test_matmul_cuda.py
|
Add CUTLASS-based support for mixed dtypes matrix multiplication (#110981)
|
2023-10-11 21:47:52 +00:00 |
|
test_meta.py
|
Add meta kernel for ctc_loss.intList (#107949)
|
2023-10-09 16:35:14 +00:00 |
|
test_metal.py
|
|
|
|
test_mkl_verbose.py
|
|
|
|
test_mkldnn.py
|
add fp16 support for gemm (#99498)
|
2023-09-28 01:03:50 +00:00 |
|
test_mkldnn_fusion.py
|
|
|
|
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
|
|
|
|
test_mps.py
|
Add torch.unravel_index (#110580)
|
2023-10-12 00:55:51 +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
|
Make NestedTensor compilable with eager backend (#109171)
|
2023-10-11 04:47:10 +00:00 |
|
test_nn.py
|
add Half support for BatchNorm on CPU (#102070)
|
2023-09-19 10:43:33 +00:00 |
|
test_nnapi.py
|
|
|
|
test_numba_integration.py
|
|
|
|
test_numpy_interop.py
|
Implement numpy(force=True) (#109636)
|
2023-09-20 20:06:13 +00:00 |
|
test_openmp.py
|
|
|
|
test_ops.py
|
[inductor] Re-enable more fixed tests (#110798)
|
2023-10-09 04:36:51 +00:00 |
|
test_ops_fwd_gradients.py
|
|
|
|
test_ops_gradients.py
|
|
|
|
test_ops_jit.py
|
|
|
|
test_optim.py
|
|
|
|
test_out_dtype_op.py
|
|
|
|
test_overrides.py
|
Disabled UserWarnings for some public functions in torch.overrides (#109890)
|
2023-09-23 20:40:04 +00:00 |
|
test_package.py
|
|
|
|
test_per_overload_api.py
|
|
|
|
test_prims.py
|
fix infinite loop with primtorch and .to(meta) (#109632)
|
2023-09-22 07:09:04 +00:00 |
|
test_proxy_tensor.py
|
Move export.constrain_as_* to torch._constrain_as_* (#110757)
|
2023-10-12 05:37:44 +00:00 |
|
test_pruning_op.py
|
|
|
|
test_public_bindings.py
|
Add a test to make sure all modules in the codebase are importable (#110598)
|
2023-10-08 03:52:30 +00:00 |
|
test_python_dispatch.py
|
Rename torch._C._TensorBase to TensorBase (#109940)
|
2023-09-25 19:10:22 +00:00 |
|
test_pytree.py
|
[pytree] Extract reusable generic tests for pytree (#110395)
|
2023-10-04 23:40:50 +00:00 |
|
test_quantization.py
|
[quant][pt2e][be] Move xnnpack quantizer tests to separate file (#111004)
|
2023-10-12 01:16:05 +00:00 |
|
test_reductions.py
|
Fix typo under test directory (#110826)
|
2023-10-08 20:52:38 +00:00 |
|
test_scatter_gather_ops.py
|
|
|
|
test_schema_check.py
|
|
|
|
test_segment_reductions.py
|
|
|
|
test_serialization.py
|
|
|
|
test_set_default_mobile_cpu_allocator.py
|
|
|
|
test_shape_ops.py
|
|
|
|
test_show_pickle.py
|
|
|
|
test_sort_and_select.py
|
|
|
|
test_sparse.py
|
|
|
|
test_sparse_csr.py
|
Move has_triton to top level triton utils so that dynamo can also access (#109832)
|
2023-09-22 19:33:41 +00:00 |
|
test_sparse_semi_structured.py
|
Revert "[sparse] Add i8i8->i32 support for cuSPARSELt (#110499)"
|
2023-10-11 16:14:59 +00:00 |
|
test_spectral_ops.py
|
Feature/stft no window warn (#110695)
|
2023-10-07 20:24:36 +00:00 |
|
test_stateless.py
|
|
|
|
test_static_runtime.py
|
|
|
|
test_subclass.py
|
|
|
|
test_sympy_utils.py
|
[reland] Add symbolic singleton int (#110674)
|
2023-10-10 19:37:17 +00:00 |
|
test_tensor_creation_ops.py
|
|
|
|
test_tensorboard.py
|
Add tensorboard to pip requirements (#109349)
|
2023-09-15 10:39:48 +00:00 |
|
test_tensorexpr.py
|
|
|
|
test_tensorexpr_pybind.py
|
|
|
|
test_testing.py
|
[Inductor CUTLASS backend] Step 1: Inductor config for cuda / cutlass, util functions. (#107802)
|
2023-09-12 17:44:32 +00:00 |
|
test_throughput_benchmark.py
|
|
|
|
test_torch.py
|
Added decorator skipRocmIfTorchInductor and skipped failing tests (#107760)
|
2023-10-12 16:00:35 +00:00 |
|
test_transformers.py
|
Adding Backward Support for NestedTensors and FlashAttention (#97485)
|
2023-10-10 18:08:17 +00:00 |
|
test_type_hints.py
|
|
|
|
test_type_info.py
|
Add finfo properties for float8 dtypes (#109744)
|
2023-09-21 03:41:48 +00:00 |
|
test_type_promotion.py
|
Fix type promotion of float8_e5m2 and float8_e4m3fn (#110279)
|
2023-10-05 01:28:48 +00:00 |
|
test_typing.py
|
|
|
|
test_unary_ufuncs.py
|
|
|
|
test_utils.py
|
Clean up test_external_module_register (#110254)
|
2023-09-29 17:02:13 +00:00 |
|
test_view_ops.py
|
Implement numpy(force=True) (#109636)
|
2023-09-20 20:06:13 +00:00 |
|
test_vulkan.py
|
|
|
|
test_weak.py
|
|
|
|
test_xnnpack_integration.py
|
|
|
