Files

.ci
.circleci
.ctags.d
.devcontainer
.github
.vscode
android
aten
benchmarks
binaries
c10
caffe2
cmake
docs
functorch
mypy_plugins
scripts
test
- ao
- autograd
- backends
- benchmark_utils
- bottleneck_test
- cpp
  - aoti_abi_check
  - aoti_inference
  - api
  - c10d
  - common
  - dist_autograd
  - jit
    - upgrader_models
    - CMakeLists.txt
    - README.md
    - __init__.py
    - script_module_v4.ptl
    - script_module_v5.ptl
    - script_module_v6.ptl
    - source_range_test.cpp
    - test_add_if_then_else.cpp
    - test_alias_analysis.cpp
    - test_argument_spec.cpp
    - test_autodiff.cpp
    - test_backend.cpp
    - test_backend_compiler_lib.cpp
    - test_backend_compiler_preprocess.cpp
    - test_backend_lib.cpp
    - test_class_import.cpp
    - test_class_parser.cpp
    - test_class_type.cpp
    - test_cleanup_passes.cpp
    - test_code_template.cpp
    - test_concat_opt.cpp
    - test_constant_pooling.cpp
    - test_create_autodiff_subgraphs.cpp
    - test_cs_debug_info_serialization.cpp
    - test_custom_class.cpp
    - test_custom_class_registrations.cpp
    - test_custom_class_registrations.h
    - test_custom_operators.cpp
    - test_dce.cpp
    - test_exception.cpp
    - test_file_format.cpp
    - test_flatbuffer.cpp
    - test_fuser.cpp
    - test_graph_executor.cpp
    - test_graph_iterator.cpp
    - test_inliner.cpp
    - test_interface.cpp
    - test_interpreter.cpp
    - test_interpreter_async.pt
    - test_ir.cpp
    - test_irparser.cpp
    - test_jit_logging_levels.cpp
    - test_jit_type.cpp
    - test_lite_interpreter.cpp
    - test_lite_interpreter_direct.cpp
    - test_lite_trainer.cpp
    - test_load_upgraders.cpp
    - test_memory_dag.cpp
    - test_misc.cpp
    - test_mobile_type_parser.cpp
    - test_module_api.cpp
    - test_op_replacement.cpp
    - test_peephole_optimize.cpp
    - test_qualified_name.cpp
    - test_save_load.cpp
    - test_schema_info.cpp
    - test_schema_matching.cpp
    - test_script_profile.cpp
    - test_shape_analysis.cpp
    - test_stack_opt.cpp
    - test_subgraph_matcher.cpp
    - test_subgraph_rewriter.cpp
    - test_subgraph_utils.cpp
    - test_union.cpp
    - test_upgrader_utils.cpp
    - test_utils.cpp
    - test_utils.h
    - tests_setup.py
    - torch_python_test.cpp
  - lazy
  - lite_interpreter_runtime
  - monitor
  - profiler
  - rpc
  - tensorexpr
  - __init__.py
- cpp_api_parity
- cpp_extensions
- custom_backend
- custom_operator
- distributed
- distributions
- dynamo
- dynamo_expected_failures
- dynamo_skips
- edge
- error_messages
- expect
- export
- forward_backward_compatibility
- functorch
- fx
- higher_order_ops
- inductor
- inductor_expected_failures
- inductor_skips
- jit
- jit_hooks
- lazy
- mobile
- nn
- onnx
- optim
- package
- profiler
- quantization
- scripts
- strobelight
- test_img
- torch_np
- typing
- xpu
- HowToWriteTestsUsingFileCheck.md
- _test_bazel.py
- allowlist_for_publicAPI.json
- bench_mps_ops.py
- conftest.py
- create_dummy_torchscript_model.py
- delete.py
- hi.py
- 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
- simulate_nccl_errors.py
- slow_tests.json
- test_accelerator.py
- test_ao_sparsity.py
- test_appending_byte_serializer.py
- test_autocast.py
- test_autograd.py
- test_autograd_fallback.py
- test_autoload.py
- test_binary_ufuncs.py
- test_bundled_images.py
- test_bundled_inputs.py
- test_ci_sanity_check_fail.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_mtia_backend.py
- test_cpp_extensions_open_device_registration.py
- test_cpp_extensions_stream_and_event.py
- test_cuda.py
- 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
- test_dataloader.py
- test_datapipe.py
- test_decomp.py
- test_deploy.py
- test_determination.py
- test_dispatch.py
- test_dlpack.py
- test_dynamic_shapes.py
- test_expanded_weights.py
- test_extension_utils.py
- test_fake_tensor.py
- test_file_check.py
- test_flop_counter.py
- test_foreach.py
- test_function_schema.py
- test_functional_autograd_benchmark.py
- test_functional_optim.py
- test_functionalization.py
- test_functionalization_of_rng_ops.py
- test_futures.py
- test_fx.py
- test_fx_experimental.py
- test_fx_passes.py
- test_fx_reinplace_pass.py
- test_hop_infra.py
- test_hub.py
- test_import_stats.py
- test_indexing.py
- 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
- 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
- test_logging.py
- test_masked.py
- test_maskedtensor.py
- test_matmul_cuda.py
- 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
- test_model_exports_to_core_aten.py
- test_module_tracker.py
- test_modules.py
- test_monitor.py
- test_mps.py
- 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
- test_nnapi.py
- test_numba_integration.py
- test_numpy_interop.py
- test_openmp.py
- test_ops.py
- test_ops_fwd_gradients.py
- test_ops_gradients.py
- test_ops_jit.py
- test_optim.py
- test_out_dtype_op.py
- test_overrides.py
- test_package.py
- test_per_overload_api.py
- test_prims.py
- test_proxy_tensor.py
- test_pruning_op.py
- test_public_bindings.py
- test_python_dispatch.py
- test_pytree.py
- test_quantization.py
- test_reductions.py
- 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
- test_sparse_semi_structured.py
- 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
- test_tensorexpr_pybind.py
- test_testing.py
- test_throughput_benchmark.py
- test_torch.py
- test_transformers.py
- test_transformers_privateuse1.py
- test_type_hints.py
- test_type_info.py
- test_type_promotion.py
- test_typing.py
- test_unary_ufuncs.py
- test_utils.py
- test_utils_config_module.py
- test_utils_filelock.py
- test_view_ops.py
- test_vulkan.py
- test_weak.py
- test_xnnpack_integration.py
- test_xpu.py
third_party
tools
torch
torchgen
.bazelignore
.bazelrc
.bazelversion
.clang-format
.clang-tidy
.cmakelintrc
.coveragerc
.dockerignore
.flake8
.gdbinit
.git-blame-ignore-revs
.gitattributes
.gitignore
.gitmodules
.lintrunner.toml
.lldbinit
BUCK.oss
BUILD.bazel
CITATION.cff
CMakeLists.txt
CODEOWNERS
CODE_OF_CONDUCT.md
CONTRIBUTING.md
Dockerfile
GLOSSARY.md
LICENSE
MANIFEST.in
Makefile
NOTICE
README.md
RELEASE.md
SECURITY.md
WORKSPACE
aten.bzl
buckbuild.bzl
build.bzl
build_variables.bzl
defs.bzl
docker.Makefile
mypy-strict.ini
mypy.ini
pt_ops.bzl
pt_template_srcs.bzl
pyproject.toml
pytest.ini
requirements.txt
setup.py
ubsan.supp
ufunc_defs.bzl
version.txt