├── .bazelignore ├── .bazelrc ├── .bazelversion ├── .ci ├── aarch64_linux │ ├── README.md │ ├── aarch64_ci_build.sh │ ├── aarch64_ci_setup.sh │ ├── aarch64_wheel_ci_build.py │ ├── build_aarch64_wheel.py │ └── embed_library.py ├── caffe2 │ ├── README.md │ ├── common.sh │ └── test.sh ├── docker │ ├── README.md │ ├── almalinux │ │ ├── Dockerfile │ │ └── build.sh │ ├── build.sh │ ├── centos-rocm │ │ └── Dockerfile │ ├── ci_commit_pins │ │ ├── executorch.txt │ │ ├── halide.txt │ │ ├── huggingface.txt │ │ ├── nccl-cu11.txt │ │ ├── nccl-cu12.txt │ │ ├── timm.txt │ │ ├── triton-cpu.txt │ │ ├── triton-xpu.txt │ │ └── triton.txt │ ├── common │ │ ├── cache_vision_models.sh │ │ ├── common_utils.sh │ │ ├── install_acl.sh │ │ ├── install_amdsmi.sh │ │ ├── install_base.sh │ │ ├── install_cache.sh │ │ ├── install_clang.sh │ │ ├── install_cmake.sh │ │ ├── install_conda.sh │ │ ├── install_conda_docker.sh │ │ ├── install_cpython.sh │ │ ├── install_cuda.sh │ │ ├── install_cuda_aarch64.sh │ │ ├── install_cudnn.sh │ │ ├── install_cudss.sh │ │ ├── install_cusparselt.sh │ │ ├── install_db.sh │ │ ├── install_devtoolset.sh │ │ ├── install_docs_reqs.sh │ │ ├── install_executorch.sh │ │ ├── install_gcc.sh │ │ ├── install_glibc.sh │ │ ├── install_halide.sh │ │ ├── install_inductor_benchmark_deps.sh │ │ ├── install_jni.sh │ │ ├── install_lcov.sh │ │ ├── install_libpng.sh │ │ ├── install_linter.sh │ │ ├── install_magma.sh │ │ ├── install_magma_conda.sh │ │ ├── install_miopen.sh │ │ ├── install_mkl.sh │ │ ├── install_mnist.sh │ │ ├── install_ninja.sh │ │ ├── install_nvpl.sh │ │ ├── install_onnx.sh │ │ ├── install_openblas.sh │ │ ├── install_openmpi.sh │ │ ├── install_openssl.sh │ │ ├── install_patchelf.sh │ │ ├── install_protobuf.sh │ │ ├── install_rocm.sh │ │ ├── install_rocm_drm.sh │ │ ├── install_rocm_magma.sh │ │ ├── install_swiftshader.sh │ │ ├── install_triton.sh │ │ ├── install_ucc.sh │ │ ├── install_user.sh │ │ ├── install_vision.sh │ │ ├── install_vulkan_sdk.sh │ │ └── install_xpu.sh │ ├── java │ │ └── jni.h │ ├── libtorch │ │ ├── Dockerfile │ │ └── build.sh │ ├── linter-cuda │ │ └── Dockerfile │ ├── linter │ │ └── Dockerfile │ ├── manywheel │ │ ├── Dockerfile │ │ ├── Dockerfile_2_28 │ │ ├── Dockerfile_2_28_aarch64 │ │ ├── Dockerfile_aarch64 │ │ ├── Dockerfile_cuda_aarch64 │ │ ├── Dockerfile_cxx11-abi │ │ ├── Dockerfile_s390x │ │ ├── build.sh │ │ └── build_scripts │ │ │ ├── build.sh │ │ │ ├── build_utils.sh │ │ │ ├── manylinux1-check.py │ │ │ └── ssl-check.py │ ├── requirements-ci.txt │ ├── requirements-docs.txt │ ├── triton_version.txt │ ├── ubuntu-cuda │ │ └── Dockerfile │ ├── ubuntu-rocm │ │ ├── .gitignore │ │ └── Dockerfile │ ├── ubuntu-xpu │ │ └── Dockerfile │ └── ubuntu │ │ └── Dockerfile ├── libtorch │ └── build.sh ├── magma │ ├── .gitignore │ ├── Makefile │ ├── README.md │ ├── build_magma.sh │ └── package_files │ │ ├── CMake.patch │ │ ├── build.sh │ │ ├── cmakelists.patch │ │ ├── getrf_nbparam.patch │ │ ├── getrf_shfl.patch │ │ ├── magma-2.6.1.sha256 │ │ └── thread_queue.patch ├── manywheel │ ├── LICENSE │ ├── build.sh │ ├── build_common.sh │ ├── build_cpu.sh │ ├── build_cuda.sh │ ├── build_libtorch.sh │ ├── build_rocm.sh │ ├── build_xpu.sh │ ├── set_desired_python.sh │ └── test_wheel.sh ├── onnx │ ├── README.md │ ├── common.sh │ └── test.sh ├── pytorch │ ├── .shellcheckrc │ ├── README.md │ ├── build-mobile.sh │ ├── build.sh │ ├── check_binary.sh │ ├── codegen-test.sh │ ├── common-build.sh │ ├── common.sh │ ├── common_utils.sh │ ├── cpp_doc_push_script.sh │ ├── create_test_cert.py │ ├── docker-build-test.sh │ ├── docs-test.sh │ ├── fake_numpy │ │ └── numpy.py │ ├── functorch_doc_push_script.sh │ ├── install_cache_xla.sh │ ├── macos-build-test.sh │ ├── macos-build.sh │ ├── macos-common.sh │ ├── macos-test.sh │ ├── multigpu-test.sh │ ├── perf_test │ │ ├── common.sh │ │ ├── compare_with_baseline.py │ │ ├── get_stats.py │ │ ├── test_cpu_speed_mini_sequence_labeler.sh │ │ ├── test_cpu_speed_mnist.sh │ │ ├── test_cpu_speed_torch.sh │ │ ├── test_cpu_speed_torch_tensor.sh │ │ ├── test_gpu_speed_cudnn_lstm.sh │ │ ├── test_gpu_speed_lstm.sh │ │ ├── test_gpu_speed_mlstm.sh │ │ ├── test_gpu_speed_mnist.sh │ │ ├── test_gpu_speed_word_language_model.sh │ │ └── update_commit_hash.py │ ├── print_sccache_log.py │ ├── python_doc_push_script.sh │ ├── run_glootls_test.sh │ ├── run_tests.sh │ ├── short-perf-test-cpu.sh │ ├── short-perf-test-gpu.sh │ ├── smoke_test │ │ ├── check_binary_symbols.py │ │ ├── max_autotune.py │ │ └── smoke_test.py │ ├── test.sh │ ├── test_example_code │ │ ├── CMakeLists.txt │ │ ├── check-torch-cuda.cpp │ │ ├── check-torch-mkl.cpp │ │ ├── check-torch-xnnpack.cpp │ │ ├── cnn_smoke.py │ │ ├── cnn_smoke_win_arm64.py │ │ ├── rnn_smoke.py │ │ ├── rnn_smoke_win_arm64.py │ │ └── simple-torch-test.cpp │ ├── win-build.sh │ ├── win-test-helpers │ │ ├── build_pytorch.bat │ │ ├── choose_runtime_cuda_version.bat │ │ ├── installation-helpers │ │ │ ├── activate_miniconda3.bat │ │ │ ├── install_magma.bat │ │ │ └── install_sccache.bat │ │ ├── run_python_nn_smoketests.py │ │ ├── setup_pytorch_env.bat │ │ ├── test_custom_backend.bat │ │ ├── test_custom_script_ops.bat │ │ ├── test_distributed.bat │ │ ├── test_libtorch.bat │ │ ├── test_python_jit_legacy.bat │ │ └── test_python_shard.bat │ ├── win-test.sh │ └── windows │ │ ├── arm64 │ │ ├── bootstrap_apl.bat │ │ ├── bootstrap_buildtools.bat │ │ ├── bootstrap_git.bat │ │ ├── bootstrap_libuv.bat │ │ ├── bootstrap_openblas.bat │ │ ├── bootstrap_python.bat │ │ ├── bootstrap_rust.bat │ │ ├── bootstrap_sccache.bat │ │ ├── bootstrap_tests.bat │ │ ├── build_libtorch.bat │ │ ├── build_pytorch.bat │ │ └── smoke_test.bat │ │ ├── build_pytorch.bat │ │ ├── condaenv.bat │ │ ├── cpu.bat │ │ ├── cuda118.bat │ │ ├── cuda124.bat │ │ ├── cuda126.bat │ │ ├── cuda128.bat │ │ ├── internal │ │ ├── 7z_install.bat │ │ ├── build_wheels.bat │ │ ├── check_deps.bat │ │ ├── check_opts.bat │ │ ├── clean.bat │ │ ├── clone.bat │ │ ├── copy.bat │ │ ├── copy_cpu.bat │ │ ├── cuda_install.bat │ │ ├── driver_update.bat │ │ ├── env_fix.bat │ │ ├── setup.bat │ │ ├── smoke_test.bat │ │ ├── static_lib_test.bat │ │ ├── vc_install_helper.bat │ │ ├── vs2022_install.ps1 │ │ ├── vs_install.bat │ │ └── xpu_install.bat │ │ └── xpu.bat └── wheel │ └── build_wheel.sh ├── .circleci ├── .gitignore ├── README.md ├── codegen_validation │ ├── compare_normalized_yaml.sh │ ├── normalize_yaml_fragment.py │ └── overwrite_with_normalized.sh ├── scripts │ ├── README.md │ ├── binary_ios_build.sh │ ├── binary_ios_test.sh │ ├── binary_ios_upload.sh │ ├── binary_linux_test.sh │ ├── binary_populate_env.sh │ ├── binary_upload.sh │ ├── binary_windows_build.sh │ ├── binary_windows_test.sh │ ├── driver_update.bat │ ├── functorch_doc_push_script.sh │ ├── publish_android_snapshot.sh │ └── trigger_azure_pipeline.py └── windows-jni │ └── include │ └── jni.h ├── .clang-format ├── .clang-tidy ├── .cmakelintrc ├── .coveragerc ├── .ctags.d └── pytorch.ctags ├── .devcontainer ├── Dockerfile ├── README.md ├── cpu │ ├── devcontainer.json │ └── environment.yml ├── cuda │ ├── devcontainer.json │ └── environment.yml ├── noop.txt └── scripts │ ├── install-dev-tools.sh │ └── update_alternatives_clang.sh ├── .dockerignore ├── .flake8 ├── .gdbinit ├── .git-blame-ignore-revs ├── .gitattributes ├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.yml │ ├── ci-sev.md │ ├── config.yml │ ├── disable-ci-jobs.md │ ├── documentation.yml │ ├── feature-request.yml │ └── pt2-bug-report.yml ├── PULL_REQUEST_TEMPLATE.md ├── actionlint.yaml ├── actions │ ├── build-android │ │ └── action.yml │ ├── checkout-pytorch │ │ └── action.yml │ ├── chown-workspace │ │ └── action.yml │ ├── diskspace-cleanup │ │ └── action.yml │ ├── download-build-artifacts │ │ └── action.yml │ ├── download-td-artifacts │ │ └── action.yml │ ├── filter-test-configs │ │ └── action.yml │ ├── get-workflow-job-id │ │ └── action.yml │ ├── linux-test │ │ └── action.yml │ ├── pytest-cache-download │ │ └── action.yml │ ├── pytest-cache-upload │ │ └── action.yml │ ├── setup-linux │ │ └── action.yml │ ├── setup-rocm │ │ └── action.yml │ ├── setup-win │ │ └── action.yml │ ├── setup-xpu │ │ └── action.yml │ ├── teardown-rocm │ │ └── action.yml │ ├── teardown-win │ │ └── action.yml │ ├── teardown-xpu │ │ └── action.yml │ ├── test-pytorch-binary │ │ └── action.yml │ ├── upload-sccache-stats │ │ └── action.yml │ ├── upload-test-artifacts │ │ └── action.yml │ └── upload-utilization-stats │ │ └── action.yml ├── auto_request_review.yml ├── ci_commit_pins │ ├── audio.txt │ ├── data.txt │ ├── fbgemm.txt │ ├── fbgemm_rocm.txt │ ├── multipy.txt │ ├── text.txt │ ├── torchao.txt │ ├── torchbench.txt │ ├── torchrec.txt │ ├── triton.txt │ ├── vision.txt │ └── xla.txt ├── label_to_label.yml ├── labeler.yml ├── merge_rules.yaml ├── nitpicks.yml ├── pytorch-circleci-labels.yml ├── pytorch-probot.yml ├── regenerate.sh ├── requirements-gha-cache.txt ├── requirements │ ├── README.md │ ├── conda-env-Linux-X64.txt │ ├── conda-env-iOS.txt │ ├── conda-env-macOS-ARM64 │ ├── pip-requirements-iOS.txt │ ├── pip-requirements-macOS.txt │ └── regenerate-requirements.txt ├── scripts │ ├── README.md │ ├── amd │ │ ├── package_triton_wheel.sh │ │ └── patch_triton_wheel.sh │ ├── build_triton_wheel.py │ ├── check_labels.py │ ├── cherry_pick.py │ ├── close_nonexistent_disable_issues.py │ ├── collect_ciflow_labels.py │ ├── comment_on_pr.py │ ├── convert_lintrunner_annotations_to_github.py │ ├── delete_old_branches.py │ ├── docathon-label-sync.py │ ├── drci_mocks.json.gz │ ├── ensure_actions_will_cancel.py │ ├── export_pytorch_labels.py │ ├── file_io_utils.py │ ├── filter_test_configs.py │ ├── generate_binary_build_matrix.py │ ├── generate_ci_workflows.py │ ├── generate_docker_release_matrix.py │ ├── generate_pytorch_version.py │ ├── get_aws_session_tokens.py │ ├── get_workflow_job_id.py │ ├── github_utils.py │ ├── gitutils.py │ ├── gql_mocks.json.gz │ ├── kill_active_ssh_sessions.ps1 │ ├── label_utils.py │ ├── lint_native_functions.py │ ├── lintrunner.sh │ ├── parse_ref.py │ ├── pr-sanity-check.sh │ ├── pytest_cache.py │ ├── pytest_caching_utils.py │ ├── report_git_status.sh │ ├── rockset_mocks.json.gz │ ├── runner_determinator.py │ ├── s390x-ci │ │ ├── README.md │ │ └── self-hosted-builder │ │ │ ├── actions-runner.Dockerfile │ │ │ ├── actions-runner@.service │ │ │ ├── fs │ │ │ └── usr │ │ │ │ └── bin │ │ │ │ ├── actions-runner │ │ │ │ └── entrypoint │ │ │ ├── helpers │ │ │ ├── app_token.sh │ │ │ ├── gh_cat_token.sh │ │ │ └── gh_token_generator.sh │ │ │ └── qemu-user-static.service │ ├── stop_runner_service.sh │ ├── tag_docker_images_for_release.py │ ├── td_llm_indexer.sh │ ├── test_check_labels.py │ ├── test_filter_test_configs.py │ ├── test_gitutils.py │ ├── test_label_utils.py │ ├── test_pytest_caching_utils.py │ ├── test_runner_determinator.py │ ├── test_trymerge.py │ ├── test_tryrebase.py │ ├── trymerge.py │ ├── trymerge_explainer.py │ ├── tryrebase.py │ ├── update_runner_determinator.py │ ├── upload_aws_ossci.sh │ ├── wait_for_ssh_to_drain.ps1 │ └── windows │ │ └── build_magma.bat ├── templates │ ├── common.yml.j2 │ ├── linux_binary_build_workflow.yml.j2 │ ├── macos_binary_build_workflow.yml.j2 │ ├── upload.yml.j2 │ └── windows_binary_build_workflow.yml.j2 └── workflows │ ├── _bazel-build-test.yml │ ├── _binary-build-linux.yml │ ├── _binary-test-linux.yml │ ├── _binary-upload.yml │ ├── _docs.yml │ ├── _linux-build.yml │ ├── _linux-test.yml │ ├── _mac-build.yml │ ├── _mac-test-mps.yml │ ├── _mac-test.yml │ ├── _rocm-test.yml │ ├── _runner-determinator.yml │ ├── _win-build.yml │ ├── _win-test.yml │ ├── _xpu-test.yml │ ├── assigntome-docathon.yml │ ├── auto_request_review.yml │ ├── build-almalinux-images.yml │ ├── build-libtorch-images.yml │ ├── build-magma-linux.yml │ ├── build-magma-windows.yml │ ├── build-manywheel-images-s390x.yml │ ├── build-manywheel-images.yml │ ├── build-triton-wheel.yml │ ├── check-labels.yml │ ├── check_mergeability_ghstack.yml │ ├── cherry-pick.yml │ ├── close-nonexistent-disable-issues.yml │ ├── create_release.yml │ ├── delete_old_branches.yml │ ├── docathon-sync-label.yml │ ├── docker-builds.yml │ ├── docker-release.yml │ ├── generated-linux-aarch64-binary-manywheel-nightly.yml │ ├── generated-linux-binary-libtorch-cxx11-abi-main.yml │ ├── generated-linux-binary-libtorch-cxx11-abi-nightly.yml │ ├── generated-linux-binary-manywheel-main.yml │ ├── generated-linux-binary-manywheel-nightly.yml │ ├── generated-linux-s390x-binary-manywheel-nightly.yml │ ├── generated-macos-arm64-binary-libtorch-cxx11-abi-nightly.yml │ ├── generated-macos-arm64-binary-wheel-nightly.yml │ ├── generated-windows-binary-libtorch-debug-main.yml │ ├── generated-windows-binary-libtorch-debug-nightly.yml │ ├── generated-windows-binary-libtorch-release-main.yml │ ├── generated-windows-binary-libtorch-release-nightly.yml │ ├── generated-windows-binary-wheel-nightly.yml │ ├── inductor-micro-benchmark-x86.yml │ ├── inductor-micro-benchmark.yml │ ├── inductor-perf-compare.yml │ ├── inductor-perf-test-nightly-aarch64.yml │ ├── inductor-perf-test-nightly-h100.yml │ ├── inductor-perf-test-nightly-macos.yml │ ├── inductor-perf-test-nightly-rocm.yml │ ├── inductor-perf-test-nightly-x86.yml │ ├── inductor-perf-test-nightly.yml │ ├── inductor-periodic.yml │ ├── inductor-rocm-mi300.yml │ ├── inductor-rocm.yml │ ├── inductor-unittest.yml │ ├── inductor.yml │ ├── lint-autoformat.yml │ ├── lint-bc.yml │ ├── lint.yml │ ├── linux-aarch64.yml │ ├── llm_td_retrieval.yml │ ├── mac-mps.yml │ ├── nightly-s3-uploads.yml │ ├── nightly.yml │ ├── nitpicker.yml │ ├── periodic.yml │ ├── pull.yml │ ├── revert.yml │ ├── rocm-mi300.yml │ ├── rocm.yml │ ├── runner-determinator-validator.yml │ ├── runner_determinator_script_sync.yaml │ ├── s390.yml │ ├── s390x-periodic.yml │ ├── scorecards.yml │ ├── slow.yml │ ├── stale.yml │ ├── target-determination-indexer.yml │ ├── target_determination.yml │ ├── test-check-binary.yml │ ├── torchbench.yml │ ├── trunk.yml │ ├── trymerge.yml │ ├── tryrebase.yml │ ├── unstable-periodic.yml │ ├── unstable.yml │ ├── update-viablestrict.yml │ ├── update_pytorch_labels.yml │ ├── upload-test-stats-while-running.yml │ ├── upload-test-stats.yml │ ├── upload-torch-dynamo-perf-stats.yml │ ├── upload_test_stats_intermediate.yml │ ├── weekly.yml │ └── xpu.yml ├── .gitignore ├── .gitmodules ├── .lintrunner.toml ├── .lldbinit ├── .vscode ├── extensions.json └── settings_recommended.json ├── 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 ├── android ├── .gitignore ├── README.md ├── build.gradle ├── build_test_app.sh ├── build_test_app_custom.sh ├── common.sh ├── gradle.properties ├── gradle │ ├── android_tasks.gradle │ ├── release.gradle │ └── wrapper │ │ ├── gradle-wrapper.jar │ │ └── gradle-wrapper.properties ├── gradlew ├── gradlew.bat ├── pytorch_android │ ├── CMakeLists.txt │ ├── build.gradle │ ├── generate_test_asset.cpp │ ├── generate_test_torchscripts.py │ ├── gradle.properties │ ├── host │ │ ├── build.gradle │ │ └── gradle.properties │ ├── src │ │ ├── androidTest │ │ │ ├── assets │ │ │ │ ├── activation_ops.ptl │ │ │ │ ├── android_api_module.ptl │ │ │ │ ├── blas_lapack_ops.ptl │ │ │ │ ├── comparison_ops.ptl │ │ │ │ ├── convolution_ops.ptl │ │ │ │ ├── distance_function_ops.ptl │ │ │ │ ├── dropout_ops.ptl │ │ │ │ ├── dynamic_quant_ops.ptl │ │ │ │ ├── fused_quant_ops.ptl │ │ │ │ ├── general_quant_ops.ptl │ │ │ │ ├── linear_ops.ptl │ │ │ │ ├── loss_function_ops.ptl │ │ │ │ ├── mobilenet_v2.ptl │ │ │ │ ├── nn_utils_ops.ptl │ │ │ │ ├── normalization_ops.ptl │ │ │ │ ├── other_math_ops.ptl │ │ │ │ ├── padding_ops.ptl │ │ │ │ ├── pointwise_ops.ptl │ │ │ │ ├── pooling_ops.ptl │ │ │ │ ├── recurrent_ops.ptl │ │ │ │ ├── reduction_ops.ptl │ │ │ │ ├── sampling_ops.ptl │ │ │ │ ├── shuffle_ops.ptl │ │ │ │ ├── sparse_ops.ptl │ │ │ │ ├── spectral_ops.ptl │ │ │ │ ├── static_quant_ops.ptl │ │ │ │ ├── tensor_creation_ops.ptl │ │ │ │ ├── tensor_general_ops.ptl │ │ │ │ ├── tensor_indexing_ops.ptl │ │ │ │ ├── tensor_typing_ops.ptl │ │ │ │ ├── tensor_view_ops.ptl │ │ │ │ ├── test.pt │ │ │ │ ├── torchscript_builtin_ops.ptl │ │ │ │ ├── torchscript_collection_ops.ptl │ │ │ │ ├── transformer_ops.ptl │ │ │ │ └── vision_function_ops.ptl │ │ │ ├── cpp │ │ │ │ └── pytorch_jni_common_test.cpp │ │ │ └── java │ │ │ │ └── org │ │ │ │ └── pytorch │ │ │ │ ├── PytorchHostTests.java │ │ │ │ ├── PytorchInstrumentedTests.java │ │ │ │ ├── PytorchLiteInstrumentedTests.java │ │ │ │ ├── PytorchTestBase.java │ │ │ │ └── suite │ │ │ │ ├── PytorchInstrumentedTestSuite.java │ │ │ │ └── PytorchLiteInstrumentedTestSuite.java │ │ └── main │ │ │ ├── AndroidManifest.xml │ │ │ ├── cpp │ │ │ ├── cmake_macros.h │ │ │ ├── cmake_macros.h.in │ │ │ ├── pytorch_jni_common.cpp │ │ │ ├── pytorch_jni_common.h │ │ │ ├── pytorch_jni_jit.cpp │ │ │ └── pytorch_jni_lite.cpp │ │ │ ├── java │ │ │ └── org │ │ │ │ └── pytorch │ │ │ │ ├── DType.java │ │ │ │ ├── Device.java │ │ │ │ ├── INativePeer.java │ │ │ │ ├── IValue.java │ │ │ │ ├── LiteModuleLoader.java │ │ │ │ ├── LiteNativePeer.java │ │ │ │ ├── MemoryFormat.java │ │ │ │ ├── Module.java │ │ │ │ ├── NativePeer.java │ │ │ │ ├── PyTorchAndroid.java │ │ │ │ ├── PyTorchCodegenLoader.java │ │ │ │ └── Tensor.java │ │ │ └── res │ │ │ └── values │ │ │ └── strings.xml │ └── test_asset.jit ├── pytorch_android_torchvision │ ├── CMakeLists.txt │ ├── build.gradle │ ├── gradle.properties │ └── src │ │ ├── androidTest │ │ └── java │ │ │ └── org │ │ │ └── pytorch │ │ │ └── torchvision │ │ │ ├── TorchVisionInstrumentedTests.java │ │ │ └── suite │ │ │ └── TorchVisionInstrumentedTestSuite.java │ │ └── main │ │ ├── AndroidManifest.xml │ │ ├── cpp │ │ └── pytorch_vision_jni.cpp │ │ ├── java │ │ └── org │ │ │ └── pytorch │ │ │ └── torchvision │ │ │ └── TensorImageUtils.java │ │ └── res │ │ └── values │ │ └── strings.xml ├── run_tests.sh ├── settings.gradle └── test_app │ ├── .gitignore │ ├── app │ ├── CMakeLists.txt │ ├── build.gradle │ └── src │ │ └── main │ │ ├── AndroidManifest.xml │ │ ├── assets │ │ └── .gitignore │ │ ├── cpp │ │ └── pytorch_testapp_jni.cpp │ │ ├── java │ │ └── org │ │ │ └── pytorch │ │ │ └── testapp │ │ │ ├── CameraActivity.java │ │ │ ├── Constants.java │ │ │ ├── LibtorchNativeClient.java │ │ │ ├── MainActivity.java │ │ │ ├── Result.java │ │ │ └── Utils.java │ │ └── res │ │ ├── layout │ │ ├── activity_camera.xml │ │ ├── activity_main.xml │ │ └── texture_view.xml │ │ ├── mipmap-mdpi │ │ ├── ic_launcher.png │ │ └── ic_launcher_round.png │ │ └── values │ │ ├── colors.xml │ │ ├── strings.xml │ │ └── styles.xml │ ├── make_assets.py │ └── make_assets_custom.py ├── aten.bzl ├── aten ├── CMakeLists.txt ├── conda │ ├── build.sh │ └── meta.yaml ├── src │ ├── ATen │ │ ├── .gitignore │ │ ├── ATen.h │ │ ├── ATenConfig.cmake.in │ │ ├── AccumulateType.cpp │ │ ├── AccumulateType.h │ │ ├── ArrayRef.h │ │ ├── Backend.h │ │ ├── Backtrace.h │ │ ├── BlasBackend.h │ │ ├── CMakeLists.txt │ │ ├── CPUApplyUtils.h │ │ ├── CPUFixedAllocator.h │ │ ├── CPUGeneratorImpl.cpp │ │ ├── CPUGeneratorImpl.h │ │ ├── CachedTensorUtils.cpp │ │ ├── CachedTensorUtils.h │ │ ├── CollapseDims.h │ │ ├── Config.h.in │ │ ├── ConjugateFallback.cpp │ │ ├── Context.cpp │ │ ├── Context.h │ │ ├── DLConvertor.cpp │ │ ├── DLConvertor.h │ │ ├── Device.h │ │ ├── DeviceAccelerator.cpp │ │ ├── DeviceAccelerator.h │ │ ├── DeviceGuard.h │ │ ├── DimVector.h │ │ ├── Dimname.h │ │ ├── Dispatch.cpp │ │ ├── Dispatch.h │ │ ├── Dispatch_v2.h │ │ ├── DynamicLibrary.cpp │ │ ├── DynamicLibrary.h │ │ ├── EmptyTensor.cpp │ │ ├── EmptyTensor.h │ │ ├── ExpandBase.h │ │ ├── ExpandUtils.cpp │ │ ├── ExpandUtils.h │ │ ├── Formatting.h │ │ ├── FuncTorchTLS.cpp │ │ ├── FuncTorchTLS.h │ │ ├── FunctionalInverses.cpp │ │ ├── FunctionalStorageImpl.cpp │ │ ├── FunctionalStorageImpl.h │ │ ├── FunctionalTensorWrapper.cpp │ │ ├── FunctionalTensorWrapper.h │ │ ├── FunctionalizeFallbackKernel.cpp │ │ ├── Generator.h │ │ ├── InferSize.h │ │ ├── InitialTensorOptions.h │ │ ├── Layout.h │ │ ├── LegacyBatchedFallback.cpp │ │ ├── LegacyBatchedFallback.h │ │ ├── LegacyBatchedTensorImpl.cpp │ │ ├── LegacyBatchedTensorImpl.h │ │ ├── LegacyBatchingRegistrations.cpp │ │ ├── LegacyVmapMode.cpp │ │ ├── LegacyVmapMode.h │ │ ├── LegacyVmapTransforms.cpp │ │ ├── LegacyVmapTransforms.h │ │ ├── LinalgBackend.h │ │ ├── MapAllocator.cpp │ │ ├── MapAllocator.h │ │ ├── MatrixRef.h │ │ ├── MemoryOverlap.cpp │ │ ├── MemoryOverlap.h │ │ ├── NamedTensor.h │ │ ├── NamedTensorUtils.cpp │ │ ├── NamedTensorUtils.h │ │ ├── NestedTensorImpl.cpp │ │ ├── NestedTensorImpl.h │ │ ├── NumericUtils.h │ │ ├── OpMathType.h │ │ ├── OpaqueTensorImpl.h │ │ ├── PTThreadPool.h │ │ ├── PadNd.h │ │ ├── Parallel-inl.h │ │ ├── Parallel.h │ │ ├── ParallelCommon.cpp │ │ ├── ParallelFuture.h │ │ ├── ParallelNative.cpp │ │ ├── ParallelNative.h │ │ ├── ParallelOpenMP.cpp │ │ ├── ParallelOpenMP.h │ │ ├── ParallelThreadPoolNative.cpp │ │ ├── PythonTorchFunctionTLS.cpp │ │ ├── PythonTorchFunctionTLS.h │ │ ├── ROCmFABackend.h │ │ ├── SDPBackend.h │ │ ├── SavedTensorHooks.cpp │ │ ├── SavedTensorHooks.h │ │ ├── Scalar.h │ │ ├── ScalarOps.cpp │ │ ├── ScalarOps.h │ │ ├── ScalarType.h │ │ ├── SequenceNumber.cpp │ │ ├── SequenceNumber.h │ │ ├── SmallVector.h │ │ ├── SparseCsrTensorImpl.cpp │ │ ├── SparseCsrTensorImpl.h │ │ ├── SparseCsrTensorUtils.h │ │ ├── SparseTensorImpl.cpp │ │ ├── SparseTensorImpl.h │ │ ├── Storage.h │ │ ├── StorageUtils.cpp │ │ ├── StorageUtils.h │ │ ├── Tensor.h │ │ ├── TensorAccessor.h │ │ ├── TensorGeometry.cpp │ │ ├── TensorGeometry.h │ │ ├── TensorIndexing.cpp │ │ ├── TensorIndexing.h │ │ ├── TensorIterator.cpp │ │ ├── TensorIterator.h │ │ ├── TensorIteratorInternal.h │ │ ├── TensorMeta.cpp │ │ ├── TensorMeta.h │ │ ├── TensorNames.cpp │ │ ├── TensorNames.h │ │ ├── TensorOperators.h │ │ ├── TensorOptions.h │ │ ├── TensorSubclassLikeUtils.h │ │ ├── TensorUtils.cpp │ │ ├── TensorUtils.h │ │ ├── ThreadLocalPythonObjects.cpp │ │ ├── ThreadLocalPythonObjects.h │ │ ├── ThreadLocalState.cpp │ │ ├── ThreadLocalState.h │ │ ├── TracerMode.h │ │ ├── TypeDefault.h │ │ ├── Utils.cpp │ │ ├── Utils.h │ │ ├── Version.cpp │ │ ├── Version.h │ │ ├── VmapModeRegistrations.cpp │ │ ├── WrapDimUtils.h │ │ ├── WrapDimUtilsMulti.h │ │ ├── ZeroTensorFallback.cpp │ │ ├── autocast_mode.cpp │ │ ├── autocast_mode.h │ │ ├── benchmarks │ │ │ ├── quantize_per_channel.cpp │ │ │ ├── stateful_conv1d.cpp │ │ │ └── tensor_add.cpp │ │ ├── ceil_div.h │ │ ├── code_template.h │ │ ├── core │ │ │ ├── ATenGeneral.cpp │ │ │ ├── ATenGeneral.h │ │ │ ├── ATenOpList.h │ │ │ ├── ATen_fwd.h │ │ │ ├── ATen_pch.h │ │ │ ├── Array.h │ │ │ ├── BackendSelectFallbackKernel.cpp │ │ │ ├── Backtrace.h │ │ │ ├── CachingHostAllocator.h │ │ │ ├── CheckMemoryFormat.h │ │ │ ├── DeprecatedTypeProperties.cpp │ │ │ ├── DeprecatedTypeProperties.h │ │ │ ├── DeprecatedTypePropertiesRegistry.cpp │ │ │ ├── DeprecatedTypePropertiesRegistry.h │ │ │ ├── Dict.cpp │ │ │ ├── Dict.h │ │ │ ├── Dict_inl.h │ │ │ ├── DimVector.h │ │ │ ├── Dimname.cpp │ │ │ ├── Dimname.h │ │ │ ├── DistributionsHelper.h │ │ │ ├── Formatting.cpp │ │ │ ├── Formatting.h │ │ │ ├── Generator.cpp │ │ │ ├── Generator.h │ │ │ ├── GeneratorForPrivateuseone.cpp │ │ │ ├── GeneratorForPrivateuseone.h │ │ │ ├── IListRef.h │ │ │ ├── IListRef_inl.h │ │ │ ├── IListRef_test.cpp │ │ │ ├── LegacyTypeDispatch.h │ │ │ ├── List.cpp │ │ │ ├── List.h │ │ │ ├── List_inl.h │ │ │ ├── List_test.cpp │ │ │ ├── MT19937RNGEngine.h │ │ │ ├── MetaFallbackKernel.cpp │ │ │ ├── NamedRegistrations.cpp │ │ │ ├── NamedTensor.cpp │ │ │ ├── NamedTensor.h │ │ │ ├── NestedIntSymNodeImpl.cpp │ │ │ ├── NestedIntSymNodeImpl.h │ │ │ ├── PhiloxRNGEngine.h │ │ │ ├── PythonFallbackKernel.cpp │ │ │ ├── PythonFallbackKernel.h │ │ │ ├── PythonOpRegistrationTrampoline.cpp │ │ │ ├── PythonOpRegistrationTrampoline.h │ │ │ ├── QuantizerBase.h │ │ │ ├── README.md │ │ │ ├── Range.cpp │ │ │ ├── Range.h │ │ │ ├── Reduction.h │ │ │ ├── Scalar.h │ │ │ ├── ScalarType.h │ │ │ ├── Tensor.cpp │ │ │ ├── Tensor.h │ │ │ ├── TensorAccessor.h │ │ │ ├── TensorBase.h │ │ │ ├── TorchDispatchUtils.cpp │ │ │ ├── TorchDispatchUtils.h │ │ │ ├── TransformationHelper.h │ │ │ ├── UndefinedTensorImpl.h │ │ │ ├── UnsafeFromTH.h │ │ │ ├── VariableFallbackKernel.cpp │ │ │ ├── VariableHooksInterface.cpp │ │ │ ├── VariableHooksInterface.h │ │ │ ├── Variadic.h │ │ │ ├── Vitals.cpp │ │ │ ├── Vitals.h │ │ │ ├── adaption.cpp │ │ │ ├── alias_info.h │ │ │ ├── blob.cpp │ │ │ ├── blob.h │ │ │ ├── boxing │ │ │ │ ├── BoxedKernel.h │ │ │ │ ├── BoxedKernel_impl.h │ │ │ │ ├── KernelFunction.cpp │ │ │ │ ├── KernelFunction.h │ │ │ │ ├── KernelFunction_impl.h │ │ │ │ ├── KernelFunction_test.cpp │ │ │ │ ├── OperatorKernel.h │ │ │ │ └── impl │ │ │ │ │ ├── WrapFunctionIntoFunctor.h │ │ │ │ │ ├── WrapFunctionIntoRuntimeFunctor.h │ │ │ │ │ ├── boxing.h │ │ │ │ │ ├── kernel_function_legacy_test.cpp │ │ │ │ │ ├── kernel_function_test.cpp │ │ │ │ │ ├── kernel_lambda_legacy_test.cpp │ │ │ │ │ ├── kernel_lambda_test.cpp │ │ │ │ │ ├── kernel_stackbased_test.cpp │ │ │ │ │ ├── make_boxed_from_unboxed_functor.h │ │ │ │ │ ├── make_boxed_from_unboxed_functor_test.cpp │ │ │ │ │ └── test_helpers.h │ │ │ ├── builtin_function.h │ │ │ ├── class_type.cpp │ │ │ ├── class_type.h │ │ │ ├── custom_class.cpp │ │ │ ├── custom_class.h │ │ │ ├── dispatch │ │ │ │ ├── CppSignature.h │ │ │ │ ├── CppSignature_test.cpp │ │ │ │ ├── DispatchKeyExtractor.cpp │ │ │ │ ├── DispatchKeyExtractor.h │ │ │ │ ├── Dispatcher.cpp │ │ │ │ ├── Dispatcher.h │ │ │ │ ├── ObservedOperators.cpp │ │ │ │ ├── ObservedOperators.h │ │ │ │ ├── OperatorEntry.cpp │ │ │ │ ├── OperatorEntry.h │ │ │ │ ├── OperatorOptions.h │ │ │ │ ├── README.md │ │ │ │ ├── RegistrationHandleRAII.h │ │ │ │ └── backend_fallback_test.cpp │ │ │ ├── dynamic_type.cpp │ │ │ ├── dynamic_type.h │ │ │ ├── enum_type.h │ │ │ ├── function.h │ │ │ ├── function_schema.cpp │ │ │ ├── function_schema.h │ │ │ ├── function_schema_inl.h │ │ │ ├── functional.h │ │ │ ├── grad_mode.h │ │ │ ├── interned_strings.cpp │ │ │ ├── interned_strings.h │ │ │ ├── interned_strings_class.h │ │ │ ├── ivalue.cpp │ │ │ ├── ivalue.h │ │ │ ├── ivalue_inl.h │ │ │ ├── ivalue_to.h │ │ │ ├── jit_type.h │ │ │ ├── jit_type_base.h │ │ │ ├── library.cpp │ │ │ ├── op_registration │ │ │ │ ├── README.md │ │ │ │ ├── adaption.h │ │ │ │ ├── infer_schema.cpp │ │ │ │ ├── infer_schema.h │ │ │ │ ├── op_allowlist.h │ │ │ │ ├── op_allowlist_test.cpp │ │ │ │ ├── op_registration.cpp │ │ │ │ ├── op_registration.h │ │ │ │ └── op_registration_test.cpp │ │ │ ├── operator_name.cpp │ │ │ ├── operator_name.h │ │ │ ├── qualified_name.h │ │ │ ├── register_symbols.cpp │ │ │ ├── rref_interface.h │ │ │ ├── stack.h │ │ │ ├── symbol.h │ │ │ ├── tensor_type.cpp │ │ │ ├── type.cpp │ │ │ ├── type_factory.cpp │ │ │ ├── type_factory.h │ │ │ ├── type_ptr.h │ │ │ ├── typeid.h │ │ │ └── union_type.cpp │ │ ├── cpp_custom_type_hack.h │ │ ├── cpu │ │ │ ├── FlushDenormal.cpp │ │ │ ├── FlushDenormal.h │ │ │ ├── Utils.cpp │ │ │ ├── Utils.h │ │ │ ├── vec │ │ │ │ ├── functional.h │ │ │ │ ├── functional_base.h │ │ │ │ ├── functional_bfloat16.h │ │ │ │ ├── intrinsics.h │ │ │ │ ├── sve │ │ │ │ │ ├── sve_helper.h │ │ │ │ │ ├── vec_common_sve.h │ │ │ │ │ ├── vec_double.h │ │ │ │ │ ├── vec_float.h │ │ │ │ │ ├── vec_int.h │ │ │ │ │ └── vec_qint.h │ │ │ │ ├── vec.h │ │ │ │ ├── vec128 │ │ │ │ │ ├── vec128.h │ │ │ │ │ ├── vec128_bfloat16_neon.h │ │ │ │ │ ├── vec128_convert.h │ │ │ │ │ ├── vec128_float_neon.h │ │ │ │ │ ├── vec128_half_neon.h │ │ │ │ │ └── vec128_reduced_precision_common_neon.h │ │ │ │ ├── vec256 │ │ │ │ │ ├── missing_vld1_neon.h │ │ │ │ │ ├── missing_vst1_neon.h │ │ │ │ │ ├── vec256.h │ │ │ │ │ ├── vec256_16bit_float.h │ │ │ │ │ ├── vec256_bfloat16.h │ │ │ │ │ ├── vec256_complex_double.h │ │ │ │ │ ├── vec256_complex_float.h │ │ │ │ │ ├── vec256_convert.h │ │ │ │ │ ├── vec256_double.h │ │ │ │ │ ├── vec256_float.h │ │ │ │ │ ├── vec256_half.h │ │ │ │ │ ├── vec256_int.h │ │ │ │ │ ├── vec256_mask.h │ │ │ │ │ ├── vec256_qint.h │ │ │ │ │ ├── vsx │ │ │ │ │ │ ├── vec256_bfloat16_vsx.h │ │ │ │ │ │ ├── vec256_common_vsx.h │ │ │ │ │ │ ├── vec256_complex_double_vsx.h │ │ │ │ │ │ ├── vec256_complex_float_vsx.h │ │ │ │ │ │ ├── vec256_double_vsx.h │ │ │ │ │ │ ├── vec256_float_vsx.h │ │ │ │ │ │ ├── vec256_int16_vsx.h │ │ │ │ │ │ ├── vec256_int32_vsx.h │ │ │ │ │ │ ├── vec256_int64_vsx.h │ │ │ │ │ │ ├── vec256_qint32_vsx.h │ │ │ │ │ │ ├── vec256_qint8_vsx.h │ │ │ │ │ │ ├── vec256_quint8_vsx.h │ │ │ │ │ │ └── vsx_helpers.h │ │ │ │ │ └── zarch │ │ │ │ │ │ └── vec256_zarch.h │ │ │ │ ├── vec512 │ │ │ │ │ ├── vec512.h │ │ │ │ │ ├── vec512_bfloat16.h │ │ │ │ │ ├── vec512_complex_double.h │ │ │ │ │ ├── vec512_complex_float.h │ │ │ │ │ ├── vec512_convert.h │ │ │ │ │ ├── vec512_double.h │ │ │ │ │ ├── vec512_float.h │ │ │ │ │ ├── vec512_int.h │ │ │ │ │ ├── vec512_mask.h │ │ │ │ │ └── vec512_qint.h │ │ │ │ ├── vec_base.h │ │ │ │ ├── vec_convert.h │ │ │ │ ├── vec_half.h │ │ │ │ ├── vec_mask.h │ │ │ │ └── vec_n.h │ │ │ └── vml.h │ │ ├── cuda │ │ │ ├── ATenCUDAGeneral.h │ │ │ ├── ApplyGridUtils.cuh │ │ │ ├── AsmUtils.cuh │ │ │ ├── Atomic.cuh │ │ │ ├── CUDAApplyUtils.cuh │ │ │ ├── CUDABlas.cpp │ │ │ ├── CUDABlas.h │ │ │ ├── CUDAConfig.h.in │ │ │ ├── CUDAContext.cpp │ │ │ ├── CUDAContext.h │ │ │ ├── CUDAContextLight.h │ │ │ ├── CUDADataType.h │ │ │ ├── CUDADevice.h │ │ │ ├── CUDAEvent.h │ │ │ ├── CUDAGeneratorImpl.cpp │ │ │ ├── CUDAGeneratorImpl.h │ │ │ ├── CUDAGraph.cpp │ │ │ ├── CUDAGraph.h │ │ │ ├── CUDAGraphsUtils.cuh │ │ │ ├── CUDASparse.h │ │ │ ├── CUDASparseBlas.cpp │ │ │ ├── CUDASparseBlas.h │ │ │ ├── CUDASparseDescriptors.cpp │ │ │ ├── CUDASparseDescriptors.h │ │ │ ├── CUDATensorMethods.cuh │ │ │ ├── CUDAUtils.h │ │ │ ├── CachingHostAllocator.cpp │ │ │ ├── CachingHostAllocator.h │ │ │ ├── CuSparseHandlePool.cpp │ │ │ ├── CublasHandlePool.cpp │ │ │ ├── DeviceUtils.cuh │ │ │ ├── EmptyTensor.cpp │ │ │ ├── EmptyTensor.h │ │ │ ├── Exceptions.cpp │ │ │ ├── Exceptions.h │ │ │ ├── NumericLimits.cuh │ │ │ ├── PeerToPeerAccess.cpp │ │ │ ├── PeerToPeerAccess.h │ │ │ ├── PhiloxCudaState.h │ │ │ ├── PhiloxUtils.cuh │ │ │ ├── PinnedMemoryAllocator.h │ │ │ ├── ScanUtils.cuh │ │ │ ├── Sleep.cu │ │ │ ├── Sleep.h │ │ │ ├── ThrustAllocator.h │ │ │ ├── cub-RadixSortKeys.cu │ │ │ ├── cub-RadixSortPairs.cu │ │ │ ├── cub.cu │ │ │ ├── cub.cuh │ │ │ ├── cub.h │ │ │ ├── cub_definitions.cuh │ │ │ ├── detail │ │ │ │ ├── CUDAHooks.cpp │ │ │ │ ├── CUDAHooks.h │ │ │ │ ├── DeviceThreadHandles.h │ │ │ │ ├── IndexUtils.cu │ │ │ │ ├── IndexUtils.cuh │ │ │ │ ├── IntegerDivider.cuh │ │ │ │ ├── KernelUtils.h │ │ │ │ ├── LazyNVRTC.cpp │ │ │ │ ├── LazyNVRTC.h │ │ │ │ ├── OffsetCalculator.cuh │ │ │ │ ├── PhiloxCudaStateRaw.cuh │ │ │ │ ├── TensorInfo.cuh │ │ │ │ └── UnpackRaw.cuh │ │ │ ├── jiterator.cu │ │ │ ├── jiterator.h │ │ │ ├── jiterator_impl.h │ │ │ ├── llvm_basic.cpp │ │ │ ├── llvm_complex.cpp │ │ │ ├── llvm_jit_strings.h │ │ │ ├── nvrtc_stub │ │ │ │ ├── ATenNVRTC.cpp │ │ │ │ └── ATenNVRTC.h │ │ │ └── tunable │ │ │ │ ├── GemmCommon.h │ │ │ │ ├── GemmHipblaslt.h │ │ │ │ ├── GemmRocblas.h │ │ │ │ ├── README.md │ │ │ │ ├── StreamTimer.cpp │ │ │ │ ├── StreamTimer.h │ │ │ │ ├── Tunable.cpp │ │ │ │ ├── Tunable.h │ │ │ │ ├── TunableGemm.h │ │ │ │ └── TunableOp.h │ │ ├── cudnn │ │ │ ├── AutocastRNN.cpp │ │ │ ├── Descriptors.cpp │ │ │ ├── Descriptors.h │ │ │ ├── Handle.cpp │ │ │ ├── Handle.h │ │ │ ├── Handles.h │ │ │ ├── README.md │ │ │ ├── Types.cpp │ │ │ ├── Types.h │ │ │ ├── Utils.h │ │ │ └── cudnn-wrapper.h │ │ ├── detail │ │ │ ├── AcceleratorHooksInterface.h │ │ │ ├── CPUGuardImpl.cpp │ │ │ ├── CUDAHooksInterface.cpp │ │ │ ├── CUDAHooksInterface.h │ │ │ ├── FunctionTraits.h │ │ │ ├── HIPHooksInterface.cpp │ │ │ ├── HIPHooksInterface.h │ │ │ ├── HPUHooksInterface.cpp │ │ │ ├── HPUHooksInterface.h │ │ │ ├── IPUHooksInterface.cpp │ │ │ ├── IPUHooksInterface.h │ │ │ ├── MAIAHooksInterface.cpp │ │ │ ├── MAIAHooksInterface.h │ │ │ ├── MPSHooksInterface.cpp │ │ │ ├── MPSHooksInterface.h │ │ │ ├── MTIAHooksInterface.cpp │ │ │ ├── MTIAHooksInterface.h │ │ │ ├── MetaGuardImpl.cpp │ │ │ ├── PrivateUse1HooksInterface.cpp │ │ │ ├── PrivateUse1HooksInterface.h │ │ │ ├── XPUHooksInterface.cpp │ │ │ └── XPUHooksInterface.h │ │ ├── div_rtn.h │ │ ├── dlpack.h │ │ ├── functorch │ │ │ ├── ADInterpreters.cpp │ │ │ ├── ADInterpreters.h │ │ │ ├── BatchRulesActivation.cpp │ │ │ ├── BatchRulesBinaryOps.cpp │ │ │ ├── BatchRulesConvolution.cpp │ │ │ ├── BatchRulesDecompositions.cpp │ │ │ ├── BatchRulesDynamic.cpp │ │ │ ├── BatchRulesFactory.cpp │ │ │ ├── BatchRulesHelper.cpp │ │ │ ├── BatchRulesHelper.h │ │ │ ├── BatchRulesIndexing.cpp │ │ │ ├── BatchRulesLinearAlgebra.cpp │ │ │ ├── BatchRulesLoss.cpp │ │ │ ├── BatchRulesModules.cpp │ │ │ ├── BatchRulesNorm.cpp │ │ │ ├── BatchRulesPooling.cpp │ │ │ ├── BatchRulesRandomness.cpp │ │ │ ├── BatchRulesReduceOps.cpp │ │ │ ├── BatchRulesScatterOps.cpp │ │ │ ├── BatchRulesUnaryOps.cpp │ │ │ ├── BatchRulesViews.cpp │ │ │ ├── BatchedFallback.cpp │ │ │ ├── BatchedFallback.h │ │ │ ├── BatchedTensorImpl.cpp │ │ │ ├── BatchedTensorImpl.h │ │ │ ├── BatchingMetaprogramming.h │ │ │ ├── DynamicLayer.cpp │ │ │ ├── DynamicLayer.h │ │ │ ├── FunctionalizeInterpreter.cpp │ │ │ ├── FunctionalizeInterpreter.h │ │ │ ├── Interpreter.cpp │ │ │ ├── Interpreter.h │ │ │ ├── LegacyBatchingRegistrations.cpp │ │ │ ├── LegacyVmapTransforms.cpp │ │ │ ├── LegacyVmapTransforms.h │ │ │ ├── Macros.h │ │ │ ├── PlumbingHelper.cpp │ │ │ ├── PlumbingHelper.h │ │ │ ├── PyTorchOperatorHacks.cpp │ │ │ ├── TensorWrapper.cpp │ │ │ ├── TensorWrapper.h │ │ │ ├── VmapInterpreter.cpp │ │ │ ├── VmapInterpreter.h │ │ │ └── VmapModeRegistrations.cpp │ │ ├── hip │ │ │ └── impl │ │ │ │ ├── HIPAllocatorMasqueradingAsCUDA.h │ │ │ │ ├── HIPCachingAllocatorMasqueradingAsCUDA.cpp │ │ │ │ ├── HIPCachingAllocatorMasqueradingAsCUDA.h │ │ │ │ ├── HIPGuardImplMasqueradingAsCUDA.cpp │ │ │ │ ├── HIPGuardImplMasqueradingAsCUDA.h │ │ │ │ └── HIPStreamMasqueradingAsCUDA.h │ │ ├── jit_macros.h │ │ ├── jiterator_macros.h │ │ ├── metal │ │ │ ├── Context.cpp │ │ │ └── Context.h │ │ ├── miopen │ │ │ ├── AutocastRNN.cpp │ │ │ ├── Descriptors.cpp │ │ │ ├── Descriptors.h │ │ │ ├── Exceptions.h │ │ │ ├── Handle.cpp │ │ │ ├── Handle.h │ │ │ ├── Types.cpp │ │ │ ├── Types.h │ │ │ ├── Utils.h │ │ │ └── miopen-wrapper.h │ │ ├── mkl │ │ │ ├── Descriptors.h │ │ │ ├── Exceptions.h │ │ │ ├── Limits.h │ │ │ ├── README.md │ │ │ ├── Sparse.h │ │ │ ├── SparseBlas.cpp │ │ │ ├── SparseBlas.h │ │ │ ├── SparseDescriptors.h │ │ │ └── Utils.h │ │ ├── mps │ │ │ ├── EmptyTensor.cpp │ │ │ ├── EmptyTensor.h │ │ │ ├── IndexKernels.h │ │ │ ├── MPSAllocator.h │ │ │ ├── MPSAllocator.mm │ │ │ ├── MPSAllocatorInterface.h │ │ │ ├── MPSDevice.h │ │ │ ├── MPSDevice.mm │ │ │ ├── MPSEvent.h │ │ │ ├── MPSEvent.mm │ │ │ ├── MPSFallback.mm │ │ │ ├── MPSGeneratorImpl.h │ │ │ ├── MPSGeneratorImpl.mm │ │ │ ├── MPSGuardImpl.h │ │ │ ├── MPSGuardImpl.mm │ │ │ ├── MPSHooks.h │ │ │ ├── MPSHooks.mm │ │ │ ├── MPSProfiler.h │ │ │ ├── MPSProfiler.mm │ │ │ ├── MPSStream.h │ │ │ └── MPSStream.mm │ │ ├── native │ │ │ ├── Activation.cpp │ │ │ ├── Activation.h │ │ │ ├── AdaptiveAveragePooling.cpp │ │ │ ├── AdaptiveAveragePooling3d.cpp │ │ │ ├── AdaptiveMaxPooling2d.cpp │ │ │ ├── AdaptiveMaxPooling3d.cpp │ │ │ ├── AdaptivePooling.h │ │ │ ├── AffineGridGenerator.cpp │ │ │ ├── AmpKernels.cpp │ │ │ ├── AmpKernels.h │ │ │ ├── AutogradComposite.cpp │ │ │ ├── AveragePool2d.cpp │ │ │ ├── AveragePool3d.cpp │ │ │ ├── BatchLinearAlgebra.cpp │ │ │ ├── BatchLinearAlgebra.h │ │ │ ├── BatchLinearAlgebraKernel.cpp │ │ │ ├── BinaryOps.cpp │ │ │ ├── BinaryOps.h │ │ │ ├── Blas.cpp │ │ │ ├── BlasKernel.cpp │ │ │ ├── Bucketization.cpp │ │ │ ├── BucketizationUtils.h │ │ │ ├── CPUBlas.cpp │ │ │ ├── CPUBlas.h │ │ │ ├── CPUFallback.cpp │ │ │ ├── CPUFallback.h │ │ │ ├── CanUse32BitIndexMath.h │ │ │ ├── ChanelShuffle.cpp │ │ │ ├── Col2Im.cpp │ │ │ ├── ComparisonUtils.cpp │ │ │ ├── ComplexHelper.h │ │ │ ├── CompositeRandomAccessor.h │ │ │ ├── CompositeRandomAccessorCommon.h │ │ │ ├── Constraints.cpp │ │ │ ├── ConvUtils.h │ │ │ ├── Convolution.cpp │ │ │ ├── ConvolutionMM2d.cpp │ │ │ ├── ConvolutionMM3d.cpp │ │ │ ├── ConvolutionMM3d.h │ │ │ ├── ConvolutionTBC.cpp │ │ │ ├── Copy.cpp │ │ │ ├── Copy.h │ │ │ ├── Correlation.cpp │ │ │ ├── Cross.cpp │ │ │ ├── Cross.h │ │ │ ├── DilatedConvolutionUtils.h │ │ │ ├── DilatedMaxPool2d.cpp │ │ │ ├── DilatedMaxPool3d.cpp │ │ │ ├── DispatchStub.cpp │ │ │ ├── DispatchStub.h │ │ │ ├── Distance.cpp │ │ │ ├── Distance.h │ │ │ ├── DistributionTemplates.h │ │ │ ├── Distributions.cpp │ │ │ ├── Distributions.h │ │ │ ├── Dropout.cpp │ │ │ ├── Embedding.cpp │ │ │ ├── EmbeddingBag.cpp │ │ │ ├── EmbeddingBag.h │ │ │ ├── Fill.cpp │ │ │ ├── Fill.h │ │ │ ├── ForeachOpsKernels.cpp │ │ │ ├── ForeachUtils.h │ │ │ ├── FractionalMaxPool2d.cpp │ │ │ ├── FractionalMaxPool3d.cpp │ │ │ ├── FractionalMaxPooling.h │ │ │ ├── FunctionOfAMatrixUtils.cpp │ │ │ ├── FunctionOfAMatrixUtils.h │ │ │ ├── FusedAdagrad.cpp │ │ │ ├── FusedAdagrad.h │ │ │ ├── FusedAdam.cpp │ │ │ ├── FusedAdam.h │ │ │ ├── FusedSGD.cpp │ │ │ ├── FusedSGD.h │ │ │ ├── GatedLinearUnit.cpp │ │ │ ├── Gelu.h │ │ │ ├── GridSampler.cpp │ │ │ ├── GridSampler.h │ │ │ ├── GridSamplerUtils.h │ │ │ ├── Histogram.cpp │ │ │ ├── Histogram.h │ │ │ ├── Im2Col.cpp │ │ │ ├── IndexKernel.h │ │ │ ├── IndexingUtils.cpp │ │ │ ├── IndexingUtils.h │ │ │ ├── Integration.cpp │ │ │ ├── Itertools.cpp │ │ │ ├── LegacyBatching.cpp │ │ │ ├── Lerp.cpp │ │ │ ├── Lerp.h │ │ │ ├── Linear.cpp │ │ │ ├── LinearAlgebra.cpp │ │ │ ├── LinearAlgebra.h │ │ │ ├── LinearAlgebraUtils.h │ │ │ ├── Loss.cpp │ │ │ ├── LossCTC.cpp │ │ │ ├── LossMulti.h │ │ │ ├── LossMultiLabelMargin.cpp │ │ │ ├── LossMultiMargin.cpp │ │ │ ├── LossNLL.cpp │ │ │ ├── LossNLL2d.cpp │ │ │ ├── Math.h │ │ │ ├── MathBitFallThroughLists.h │ │ │ ├── MathBitsFallback.h │ │ │ ├── MaxPooling.cpp │ │ │ ├── MaxPooling.h │ │ │ ├── MaxUnpooling.cpp │ │ │ ├── Memory.cpp │ │ │ ├── MetaTensor.cpp │ │ │ ├── NNPACK.cpp │ │ │ ├── NaiveConvolutionTranspose2d.cpp │ │ │ ├── NaiveConvolutionTranspose3d.cpp │ │ │ ├── NaiveDilatedConvolution.cpp │ │ │ ├── NamedTensor.cpp │ │ │ ├── NegateFallback.cpp │ │ │ ├── NonEmptyUtils.h │ │ │ ├── NonSymbolicBC.h │ │ │ ├── Normalization.cpp │ │ │ ├── Normalization.h │ │ │ ├── Onehot.cpp │ │ │ ├── PackedSequence.cpp │ │ │ ├── PadNd.cpp │ │ │ ├── Padding.h │ │ │ ├── PixelShuffle.cpp │ │ │ ├── PixelShuffle.h │ │ │ ├── PointwiseOps.cpp │ │ │ ├── PointwiseOps.h │ │ │ ├── Pool.h │ │ │ ├── Pooling.cpp │ │ │ ├── Pow.cpp │ │ │ ├── Pow.h │ │ │ ├── QuantizedLinear.cpp │ │ │ ├── README.md │ │ │ ├── RNN.cpp │ │ │ ├── RNN.h │ │ │ ├── RangeFactories.cpp │ │ │ ├── RangeFactories.h │ │ │ ├── RangeUtils.h │ │ │ ├── ReduceAllOps.cpp │ │ │ ├── ReduceAllOps.h │ │ │ ├── ReduceOps.cpp │ │ │ ├── ReduceOps.h │ │ │ ├── ReduceOpsUtils.h │ │ │ ├── ReductionType.h │ │ │ ├── ReflectionPad.cpp │ │ │ ├── Repeat.cpp │ │ │ ├── Repeat.h │ │ │ ├── ReplicationPadding.cpp │ │ │ ├── Resize.cpp │ │ │ ├── Resize.h │ │ │ ├── ResizeCommon.h │ │ │ ├── RowwisePrune.cpp │ │ │ ├── Scalar.cpp │ │ │ ├── ScatterGatherChecks.h │ │ │ ├── SegmentReduce.cpp │ │ │ ├── SegmentReduce.h │ │ │ ├── SharedReduceOps.h │ │ │ ├── SobolEngineOps.cpp │ │ │ ├── SobolEngineOpsUtils.cpp │ │ │ ├── SobolEngineOpsUtils.h │ │ │ ├── SoftMax.cpp │ │ │ ├── Sorting.cpp │ │ │ ├── Sorting.h │ │ │ ├── SortingUtils.h │ │ │ ├── SparseTensorUtils.cpp │ │ │ ├── SparseTensorUtils.h │ │ │ ├── SpectralOps.cpp │ │ │ ├── SpectralOpsUtils.h │ │ │ ├── StridedRandomAccessor.h │ │ │ ├── SummaryOps.cpp │ │ │ ├── TensorAdvancedIndexing.cpp │ │ │ ├── TensorAdvancedIndexing.h │ │ │ ├── TensorAdvancedIndexingUtils.h │ │ │ ├── TensorCompare.cpp │ │ │ ├── TensorCompare.h │ │ │ ├── TensorConversions.cpp │ │ │ ├── TensorConversions.h │ │ │ ├── TensorDimApply.h │ │ │ ├── TensorFactories.cpp │ │ │ ├── TensorFactories.h │ │ │ ├── TensorIterator.h │ │ │ ├── TensorIteratorDynamicCasting.h │ │ │ ├── TensorIteratorReduce.cpp │ │ │ ├── TensorProperties.cpp │ │ │ ├── TensorProperties.h │ │ │ ├── TensorShape.cpp │ │ │ ├── TensorShape.h │ │ │ ├── TensorTransformations.cpp │ │ │ ├── TensorTransformations.h │ │ │ ├── TestOps.cpp │ │ │ ├── TopKImpl.h │ │ │ ├── TransposeType.h │ │ │ ├── TriangularOps.cpp │ │ │ ├── TriangularOpsUtils.h │ │ │ ├── TypeProperties.cpp │ │ │ ├── TypeProperties.h │ │ │ ├── UnaryOps.cpp │ │ │ ├── UnaryOps.h │ │ │ ├── Unfold2d.cpp │ │ │ ├── Unfold2d.h │ │ │ ├── Unfold3d.cpp │ │ │ ├── Unfold3d.h │ │ │ ├── UnfoldBackward.cpp │ │ │ ├── UnfoldBackward.h │ │ │ ├── Unique.cpp │ │ │ ├── UpSample.cpp │ │ │ ├── UpSample.h │ │ │ ├── UpSampleBicubic2d.cpp │ │ │ ├── UpSampleBilinear2d.cpp │ │ │ ├── UpSampleLinear1d.cpp │ │ │ ├── UpSampleNearest1d.cpp │ │ │ ├── UpSampleNearest2d.cpp │ │ │ ├── UpSampleNearest3d.cpp │ │ │ ├── UpSampleTrilinear3d.cpp │ │ │ ├── VariableMethodStubs.cpp │ │ │ ├── WeightNorm.cpp │ │ │ ├── ao_sparse │ │ │ │ ├── README │ │ │ │ ├── library.cpp │ │ │ │ └── quantized │ │ │ │ │ └── cpu │ │ │ │ │ ├── fbgemm_utils.cpp │ │ │ │ │ ├── fbgemm_utils.h │ │ │ │ │ ├── packed_params.h │ │ │ │ │ ├── qlinear.cpp │ │ │ │ │ ├── qlinear_deserialize.cpp │ │ │ │ │ ├── qlinear_dynamic.cpp │ │ │ │ │ ├── qlinear_prepack.cpp │ │ │ │ │ ├── qlinear_serialize.cpp │ │ │ │ │ ├── qlinear_unpack.cpp │ │ │ │ │ └── qnnpack_utils.h │ │ │ ├── batch_norm.h │ │ │ ├── cpu │ │ │ │ ├── Activation.cpp │ │ │ │ ├── AdaptiveAvgPoolKernel.cpp │ │ │ │ ├── AdaptiveMaxPoolKernel.cpp │ │ │ │ ├── AmpGradScalerKernels.cpp │ │ │ │ ├── AtomicAddFloat.h │ │ │ │ ├── AvgPoolKernel.cpp │ │ │ │ ├── BinaryOpsKernel.cpp │ │ │ │ ├── BlasKernel.cpp │ │ │ │ ├── CatKernel.cpp │ │ │ │ ├── CatKernel.h │ │ │ │ ├── ChannelShuffleKernel.cpp │ │ │ │ ├── ChannelShuffleKernel.h │ │ │ │ ├── ComplexKernel.cpp │ │ │ │ ├── CopyKernel.cpp │ │ │ │ ├── CopyKernel.h │ │ │ │ ├── CrossKernel.cpp │ │ │ │ ├── DepthwiseConvKernel.cpp │ │ │ │ ├── DepthwiseConvKernel.h │ │ │ │ ├── DistanceOpsKernel.cpp │ │ │ │ ├── DistributionKernels.cpp │ │ │ │ ├── DistributionTemplates.h │ │ │ │ ├── FillKernel.cpp │ │ │ │ ├── FlashAttentionKernel.cpp │ │ │ │ ├── FunctionOfAMatrixUtilsKernel.cpp │ │ │ │ ├── FusedAdagradKernel.cpp │ │ │ │ ├── FusedAdamKernel.cpp │ │ │ │ ├── FusedSGDKernel.cpp │ │ │ │ ├── Gelu.h │ │ │ │ ├── GridSamplerKernel.cpp │ │ │ │ ├── GridSamplerKernel.h │ │ │ │ ├── HistogramKernel.cpp │ │ │ │ ├── IndexKernel.cpp │ │ │ │ ├── IndexKernelUtils.h │ │ │ │ ├── Intrinsics.h │ │ │ │ ├── IsContiguous.h │ │ │ │ ├── LerpKernel.cpp │ │ │ │ ├── LinearAlgebraKernel.cpp │ │ │ │ ├── LogAddExp.h │ │ │ │ ├── Loops.h │ │ │ │ ├── MaxPoolKernel.cpp │ │ │ │ ├── MaxPooling.cpp │ │ │ │ ├── MaxUnpoolKernel.cpp │ │ │ │ ├── MaxUnpoolKernel.h │ │ │ │ ├── MultinomialKernel.cpp │ │ │ │ ├── NativeMultiheadAttnKernel.cpp │ │ │ │ ├── PaddingKernel.cpp │ │ │ │ ├── PixelShuffleKernel.cpp │ │ │ │ ├── PixelShuffleKernel.h │ │ │ │ ├── PointwiseOpsKernel.cpp │ │ │ │ ├── PowKernel.cpp │ │ │ │ ├── README.md │ │ │ │ ├── RangeFactoriesKernel.cpp │ │ │ │ ├── Reduce.h │ │ │ │ ├── ReduceAllOpsKernel.cpp │ │ │ │ ├── ReduceOpsKernel.cpp │ │ │ │ ├── ReduceUtils.h │ │ │ │ ├── ReducedPrecisionFloatGemvFastPathKernel.cpp │ │ │ │ ├── ReducedPrecisionFloatGemvFastPathKernel.h │ │ │ │ ├── RenormKernel.cpp │ │ │ │ ├── SampledAddmmKernel.cpp │ │ │ │ ├── SampledAddmmKernel.h │ │ │ │ ├── ScatterGatherKernel.cpp │ │ │ │ ├── SerialStackImpl.h │ │ │ │ ├── SoftMaxKernel.cpp │ │ │ │ ├── SoftmaxKernel.h │ │ │ │ ├── SortingKernel.cpp │ │ │ │ ├── SparseFactories.cpp │ │ │ │ ├── SpmmReduceKernel.cpp │ │ │ │ ├── SpmmReduceKernel.h │ │ │ │ ├── StackKernel.cpp │ │ │ │ ├── StackKernel.h │ │ │ │ ├── SumKernel.cpp │ │ │ │ ├── TensorCompareKernel.cpp │ │ │ │ ├── UnaryOpsKernel.cpp │ │ │ │ ├── Unfold2d.cpp │ │ │ │ ├── UnfoldBackwardKernel.cpp │ │ │ │ ├── UpSampleKernel.cpp │ │ │ │ ├── UpSampleKernelAVXAntialias.h │ │ │ │ ├── UpSampleMoreKernel.cpp │ │ │ │ ├── WeightNormKernel.cpp │ │ │ │ ├── WeightNormKernel.h │ │ │ │ ├── airy_ai.cpp │ │ │ │ ├── avx_mathfun.h │ │ │ │ ├── batch_norm_kernel.cpp │ │ │ │ ├── group_norm_kernel.cpp │ │ │ │ ├── int4mm_kernel.cpp │ │ │ │ ├── int8mm_kernel.cpp │ │ │ │ ├── int_mm_kernel.h │ │ │ │ ├── layer_norm_kernel.cpp │ │ │ │ ├── mixed_data_type.h │ │ │ │ ├── moments_utils.h │ │ │ │ ├── scaled_modified_bessel_k0.cpp │ │ │ │ ├── scaled_modified_bessel_k1.cpp │ │ │ │ ├── spherical_bessel_j0.cpp │ │ │ │ ├── utils.h │ │ │ │ └── zmath.h │ │ │ ├── cuda │ │ │ │ ├── AbsKernel.cu │ │ │ │ ├── Activation.cpp │ │ │ │ ├── Activation.h │ │ │ │ ├── ActivationEluKernel.cu │ │ │ │ ├── ActivationGeluKernel.cu │ │ │ │ ├── ActivationGluKernel.cu │ │ │ │ ├── ActivationHardshrinkKernel.cu │ │ │ │ ├── ActivationHardsigmoidKernel.cu │ │ │ │ ├── ActivationHardswishKernel.cu │ │ │ │ ├── ActivationHardtanhKernel.cu │ │ │ │ ├── ActivationLeakyReluKernel.cu │ │ │ │ ├── ActivationLogSigmoidKernel.cu │ │ │ │ ├── ActivationMishKernel.cu │ │ │ │ ├── ActivationPreluKernel.cu │ │ │ │ ├── ActivationSiluKernel.cu │ │ │ │ ├── ActivationSoftplusKernel.cu │ │ │ │ ├── ActivationSoftshrinkKernel.cu │ │ │ │ ├── ActivationThresholdKernel.cu │ │ │ │ ├── AdaptiveAveragePooling.cu │ │ │ │ ├── AdaptiveAveragePooling3d.cu │ │ │ │ ├── AdaptiveMaxPooling2d.cu │ │ │ │ ├── AdaptiveMaxPooling3d.cu │ │ │ │ ├── AmpKernels.cu │ │ │ │ ├── AveragePool2d.cu │ │ │ │ ├── AveragePool3d.cu │ │ │ │ ├── BinaryBitwiseOpsKernels.cu │ │ │ │ ├── BinaryDivFloorKernel.cu │ │ │ │ ├── BinaryDivTrueKernel.cu │ │ │ │ ├── BinaryDivTruncKernel.cu │ │ │ │ ├── BinaryGeometricKernels.cu │ │ │ │ ├── BinaryInternal.h │ │ │ │ ├── BinaryLogicalOpsKernels.cu │ │ │ │ ├── BinaryMiscBackwardOpsKernels.cu │ │ │ │ ├── BinaryMiscOpsKernels.cu │ │ │ │ ├── BinaryMulKernel.cu │ │ │ │ ├── BinaryRemainderKernel.cu │ │ │ │ ├── BinaryShiftOpsKernels.cu │ │ │ │ ├── Blas.cpp │ │ │ │ ├── Bucketization.cu │ │ │ │ ├── CUDAJitLoops.cuh │ │ │ │ ├── CUDALoops.cuh │ │ │ │ ├── CUDAScalar.cu │ │ │ │ ├── Col2Im.cu │ │ │ │ ├── CompareEQKernel.cu │ │ │ │ ├── CompareKernels.cu │ │ │ │ ├── ComplexKernel.cu │ │ │ │ ├── CompositeRandomAccessor.h │ │ │ │ ├── ConvolutionMM2d.cu │ │ │ │ ├── Copy.cu │ │ │ │ ├── Copy.h │ │ │ │ ├── CopysignKernel.cu │ │ │ │ ├── CrossKernel.cu │ │ │ │ ├── CuFFTPlanCache.h │ │ │ │ ├── CuFFTUtils.h │ │ │ │ ├── CumminmaxKernel.cu │ │ │ │ ├── CumprodKernel.cu │ │ │ │ ├── CumsumKernel.cu │ │ │ │ ├── DepthwiseConv2d.cu │ │ │ │ ├── DepthwiseConv3d.cu │ │ │ │ ├── DeviceSqrt.cuh │ │ │ │ ├── DilatedMaxPool2d.cu │ │ │ │ ├── DilatedMaxPool3d.cu │ │ │ │ ├── DistanceKernel.cu │ │ │ │ ├── DistributionBernoulli.cu │ │ │ │ ├── DistributionCauchyKernel.cu │ │ │ │ ├── DistributionExponentialKernel.cu │ │ │ │ ├── DistributionGeometricKernel.cu │ │ │ │ ├── DistributionLogNormalKernel.cu │ │ │ │ ├── DistributionNormal.cu │ │ │ │ ├── DistributionRandomKernel.cu │ │ │ │ ├── DistributionTemplates.h │ │ │ │ ├── DistributionUniform.cu │ │ │ │ ├── Distributions.cpp │ │ │ │ ├── Distributions.cu │ │ │ │ ├── Distributions.h │ │ │ │ ├── Dropout.cu │ │ │ │ ├── Embedding.cu │ │ │ │ ├── EmbeddingBackwardKernel.cu │ │ │ │ ├── EmbeddingBackwardKernel.cuh │ │ │ │ ├── EmbeddingBag.cu │ │ │ │ ├── Equal.cpp │ │ │ │ ├── FillKernel.cu │ │ │ │ ├── FlattenIndicesKernel.cu │ │ │ │ ├── ForeachBinaryOpList.cu │ │ │ │ ├── ForeachBinaryOpScalar.cu │ │ │ │ ├── ForeachBinaryOpScalarList.cu │ │ │ │ ├── ForeachBinaryOpScalarTensor.cu │ │ │ │ ├── ForeachFunctors.cuh │ │ │ │ ├── ForeachMinMaxFunctors.cuh │ │ │ │ ├── ForeachPointwiseOp.cu │ │ │ │ ├── ForeachReduceOp.cu │ │ │ │ ├── ForeachTernaryOp.cu │ │ │ │ ├── ForeachUnaryOp.cu │ │ │ │ ├── FractionalMaxPool2d.cu │ │ │ │ ├── FractionalMaxPool3d.cu │ │ │ │ ├── FunctionOfAMatrixUtilsKernel.cu │ │ │ │ ├── FusedAdamKernel.cu │ │ │ │ ├── FusedAdamWKernel.cu │ │ │ │ ├── FusedSgdKernel.cu │ │ │ │ ├── GcdLcmKernel.cu │ │ │ │ ├── GridSampler.cpp │ │ │ │ ├── GridSampler.cu │ │ │ │ ├── GridSampler.cuh │ │ │ │ ├── GridSampler.h │ │ │ │ ├── IGammaKernel.cu │ │ │ │ ├── Im2Col.cu │ │ │ │ ├── IndexKernel.cpp │ │ │ │ ├── IndexKernel.cu │ │ │ │ ├── IndexKernel.h │ │ │ │ ├── Indexing.cu │ │ │ │ ├── JitLoops.cuh │ │ │ │ ├── KernelUtils.cuh │ │ │ │ ├── LaunchUtils.h │ │ │ │ ├── LegacyThrustHelpers.cu │ │ │ │ ├── Lerp.cu │ │ │ │ ├── LinearAlgebra.cu │ │ │ │ ├── LinearAlgebraStubs.cpp │ │ │ │ ├── LogAddExpKernel.cu │ │ │ │ ├── LogcumsumexpKernel.cu │ │ │ │ ├── Loops.cuh │ │ │ │ ├── Loss.cu │ │ │ │ ├── LossCTC.cu │ │ │ │ ├── Math.cuh │ │ │ │ ├── MaxMinElementwiseKernel.cu │ │ │ │ ├── MaxUnpooling.cu │ │ │ │ ├── MemoryAccess.cuh │ │ │ │ ├── MiscUtils.h │ │ │ │ ├── MixedDtypesLinear.cu │ │ │ │ ├── MultiLabelMarginCriterion.cu │ │ │ │ ├── MultiMarginLoss.cu │ │ │ │ ├── MultiTensorApply.cuh │ │ │ │ ├── MultinomialKernel.cu │ │ │ │ ├── NLLLoss2d.cu │ │ │ │ ├── NaiveConvolutionTranspose2d.cu │ │ │ │ ├── NaiveConvolutionTranspose3d.cu │ │ │ │ ├── NaiveDilatedConvolution.cu │ │ │ │ ├── Nonzero.cu │ │ │ │ ├── Normalization.cu │ │ │ │ ├── Normalization.cuh │ │ │ │ ├── PersistentSoftmax.cuh │ │ │ │ ├── PointwiseOpsKernel.cu │ │ │ │ ├── Pow.cuh │ │ │ │ ├── PowKernel.cu │ │ │ │ ├── RNN.cu │ │ │ │ ├── Randperm.cu │ │ │ │ ├── Randperm.cuh │ │ │ │ ├── RangeFactories.cu │ │ │ │ ├── RecordStream.cu │ │ │ │ ├── Reduce.cu │ │ │ │ ├── Reduce.cuh │ │ │ │ ├── ReduceAMinMaxKernel.cu │ │ │ │ ├── ReduceArgMaxKernel.cu │ │ │ │ ├── ReduceArgMinKernel.cu │ │ │ │ ├── ReduceLogicKernel.cu │ │ │ │ ├── ReduceMaxValuesKernel.cu │ │ │ │ ├── ReduceMinValuesKernel.cu │ │ │ │ ├── ReduceMomentKernel.cu │ │ │ │ ├── ReduceNormKernel.cu │ │ │ │ ├── ReduceOps.cpp │ │ │ │ ├── ReduceOps.h │ │ │ │ ├── ReduceSumProdKernel.cu │ │ │ │ ├── ReflectionPad.cu │ │ │ │ ├── RenormKernel.cu │ │ │ │ ├── Repeat.cu │ │ │ │ ├── ReplicationPadding.cu │ │ │ │ ├── Resize.cpp │ │ │ │ ├── Resize.h │ │ │ │ ├── RowwiseScaledMM.cu │ │ │ │ ├── RowwiseScaledMM.h │ │ │ │ ├── RreluWithNoise.cu │ │ │ │ ├── ScanKernels.cpp │ │ │ │ ├── ScanKernels.h │ │ │ │ ├── ScanUtils.cuh │ │ │ │ ├── ScatterGatherKernel.cu │ │ │ │ ├── SegmentReduce.cu │ │ │ │ ├── Shape.cu │ │ │ │ ├── SoftMax.cu │ │ │ │ ├── Sort.cpp │ │ │ │ ├── Sort.cu │ │ │ │ ├── Sort.h │ │ │ │ ├── SortImpl.cu │ │ │ │ ├── SortStable.cu │ │ │ │ ├── SortStable.h │ │ │ │ ├── SortUtils.cuh │ │ │ │ ├── Sorting.cpp │ │ │ │ ├── Sorting.cu │ │ │ │ ├── Sorting.h │ │ │ │ ├── SortingCommon.cuh │ │ │ │ ├── SortingRadixSelect.cuh │ │ │ │ ├── SparseBinaryOpIntersectionKernel.cu │ │ │ │ ├── SparseMM.cu │ │ │ │ ├── SpectralOps.cpp │ │ │ │ ├── SpectralOps.cu │ │ │ │ ├── StepKernel.cu │ │ │ │ ├── SummaryOps.cu │ │ │ │ ├── TensorCompare.cpp │ │ │ │ ├── TensorCompare.cu │ │ │ │ ├── TensorFactories.cu │ │ │ │ ├── TensorModeKernel.cpp │ │ │ │ ├── TensorModeKernel.cu │ │ │ │ ├── TensorModeKernel.cuh │ │ │ │ ├── TensorModeKernel.h │ │ │ │ ├── TensorShape.cu │ │ │ │ ├── TensorShapeCUDA.cpp │ │ │ │ ├── TensorTopK.cpp │ │ │ │ ├── TensorTopK.cu │ │ │ │ ├── TensorTopK.h │ │ │ │ ├── TensorTransformations.cu │ │ │ │ ├── TriangularOps.cu │ │ │ │ ├── UnaryComplexKernels.cu │ │ │ │ ├── UnaryFractionKernels.cu │ │ │ │ ├── UnaryGammaKernels.cu │ │ │ │ ├── UnaryGeometricAcosKernel.cu │ │ │ │ ├── UnaryGeometricAcoshKernel.cu │ │ │ │ ├── UnaryGeometricAsinKernel.cu │ │ │ │ ├── UnaryGeometricAsinhKernel.cu │ │ │ │ ├── UnaryGeometricAtanKernel.cu │ │ │ │ ├── UnaryGeometricAtanhKernel.cu │ │ │ │ ├── UnaryGeometricCosKernel.cu │ │ │ │ ├── UnaryGeometricCoshKernel.cu │ │ │ │ ├── UnaryGeometricSinKernel.cu │ │ │ │ ├── UnaryGeometricSinhKernel.cu │ │ │ │ ├── UnaryGeometricTanKernel.cu │ │ │ │ ├── UnaryGeometricTanhKernel.cu │ │ │ │ ├── UnaryLogKernels.cu │ │ │ │ ├── UnaryOpsKernel.cu │ │ │ │ ├── UnarySignKernels.cu │ │ │ │ ├── UnarySpecialOpsKernel.cu │ │ │ │ ├── UnfoldBackwardKernel.cu │ │ │ │ ├── Unique.cu │ │ │ │ ├── UniqueCub.cu │ │ │ │ ├── UniqueCub.cuh │ │ │ │ ├── UpSample.cuh │ │ │ │ ├── UpSampleBicubic2d.cu │ │ │ │ ├── UpSampleBilinear2d.cu │ │ │ │ ├── UpSampleLinear1d.cu │ │ │ │ ├── UpSampleNearest1d.cu │ │ │ │ ├── UpSampleNearest2d.cu │ │ │ │ ├── UpSampleNearest3d.cu │ │ │ │ ├── UpSampleTrilinear3d.cu │ │ │ │ ├── ValidateCompressedIndicesKernel.cu │ │ │ │ ├── WeightNorm.cu │ │ │ │ ├── ZetaKernel.cu │ │ │ │ ├── airy_ai.cu │ │ │ │ ├── bessel_j0.cu │ │ │ │ ├── bessel_j1.cu │ │ │ │ ├── bessel_y0.cu │ │ │ │ ├── bessel_y1.cu │ │ │ │ ├── block_reduce.cuh │ │ │ │ ├── chebyshev_polynomial_t.cu │ │ │ │ ├── chebyshev_polynomial_u.cu │ │ │ │ ├── chebyshev_polynomial_v.cu │ │ │ │ ├── chebyshev_polynomial_w.cu │ │ │ │ ├── cutlass_extensions │ │ │ │ │ ├── README.md │ │ │ │ │ ├── arch │ │ │ │ │ │ └── mma.h │ │ │ │ │ ├── epilogue │ │ │ │ │ │ └── thread │ │ │ │ │ │ │ └── ft_fused_activations.h │ │ │ │ │ ├── epilogue_helpers.h │ │ │ │ │ ├── ft_gemm_configs.h │ │ │ │ │ ├── gemm │ │ │ │ │ │ ├── kernel │ │ │ │ │ │ │ ├── default_fpA_intB_traits.h │ │ │ │ │ │ │ ├── fpA_intB_gemm.h │ │ │ │ │ │ │ └── mixed_gemm_B_layout.h │ │ │ │ │ │ ├── threadblock │ │ │ │ │ │ │ ├── default_dq_mma.h │ │ │ │ │ │ │ ├── default_dq_mma_multistage.h │ │ │ │ │ │ │ ├── default_dq_mma_pipelined.h │ │ │ │ │ │ │ ├── default_mma.h │ │ │ │ │ │ │ ├── default_mma_bf16.h │ │ │ │ │ │ │ ├── dq_mma_base.h │ │ │ │ │ │ │ ├── dq_mma_multistage.h │ │ │ │ │ │ │ └── dq_mma_pipelined.h │ │ │ │ │ │ └── warp │ │ │ │ │ │ │ ├── default_mma_tensor_op.h │ │ │ │ │ │ │ ├── mma_tensorop_compute_B_with_f16.h │ │ │ │ │ │ │ └── mma_tensorop_dequantizer.h │ │ │ │ │ ├── interleaved_numeric_conversion.h │ │ │ │ │ └── tile_interleaved_layout.h │ │ │ │ ├── fused_adam_amsgrad_impl.cu │ │ │ │ ├── fused_adam_amsgrad_impl.cuh │ │ │ │ ├── fused_adam_impl.cu │ │ │ │ ├── fused_adam_impl.cuh │ │ │ │ ├── fused_adam_utils.cuh │ │ │ │ ├── fused_adamw_amsgrad_impl.cu │ │ │ │ ├── fused_adamw_amsgrad_impl.cuh │ │ │ │ ├── fused_adamw_impl.cu │ │ │ │ ├── fused_adamw_impl.cuh │ │ │ │ ├── group_norm_kernel.cu │ │ │ │ ├── hermite_polynomial_h.cu │ │ │ │ ├── hermite_polynomial_he.cu │ │ │ │ ├── im2col.cuh │ │ │ │ ├── int4mm.cu │ │ │ │ ├── jit_utils.cpp │ │ │ │ ├── jit_utils.h │ │ │ │ ├── laguerre_polynomial_l.cu │ │ │ │ ├── layer_norm_kernel.cu │ │ │ │ ├── legendre_polynomial_p.cu │ │ │ │ ├── linalg │ │ │ │ │ ├── BatchLinearAlgebra.cpp │ │ │ │ │ ├── BatchLinearAlgebraLib.cpp │ │ │ │ │ ├── BatchLinearAlgebraLib.h │ │ │ │ │ ├── BatchLinearAlgebraLibBlas.cpp │ │ │ │ │ ├── CUDASolver.cpp │ │ │ │ │ ├── CUDASolver.h │ │ │ │ │ ├── CudssHandlePool.cpp │ │ │ │ │ ├── CusolverDnHandlePool.cpp │ │ │ │ │ └── MagmaUtils.h │ │ │ │ ├── modified_bessel_i0.cu │ │ │ │ ├── modified_bessel_i1.cu │ │ │ │ ├── modified_bessel_k0.cu │ │ │ │ ├── modified_bessel_k1.cu │ │ │ │ ├── reduction_template.cuh │ │ │ │ ├── scaled_modified_bessel_k0.cu │ │ │ │ ├── scaled_modified_bessel_k1.cu │ │ │ │ ├── shifted_chebyshev_polynomial_t.cu │ │ │ │ ├── shifted_chebyshev_polynomial_u.cu │ │ │ │ ├── shifted_chebyshev_polynomial_v.cu │ │ │ │ ├── shifted_chebyshev_polynomial_w.cu │ │ │ │ ├── spherical_bessel_j0.cu │ │ │ │ ├── thread_constants.h │ │ │ │ └── vol2col.cuh │ │ │ ├── cudnn │ │ │ │ ├── AffineGridGenerator.cpp │ │ │ │ ├── BatchNorm.cpp │ │ │ │ ├── BatchNorm.h │ │ │ │ ├── ConvPlaceholders.cpp │ │ │ │ ├── ConvShared.cpp │ │ │ │ ├── ConvShared.h │ │ │ │ ├── Conv_v7.cpp │ │ │ │ ├── Conv_v8.cpp │ │ │ │ ├── GridSampler.cpp │ │ │ │ ├── LossCTC.cpp │ │ │ │ ├── MHA.cpp │ │ │ │ ├── MHA.h │ │ │ │ ├── RNN.cpp │ │ │ │ └── RNNUtils.h │ │ │ ├── fhe │ │ │ │ └── cuda │ │ │ │ │ ├── AutomorphismTransform.cu │ │ │ │ │ ├── DropLastElementScale.cu │ │ │ │ │ ├── InnerProduct.cu │ │ │ │ │ ├── ModRaise.cu │ │ │ │ │ ├── Moddown.cu │ │ │ │ │ ├── Modup.cu │ │ │ │ │ ├── MulByMonomial.cu │ │ │ │ │ └── arithmetic.cu │ │ │ ├── group_norm.cpp │ │ │ ├── group_norm.h │ │ │ ├── hip │ │ │ │ ├── bgemm_kernels │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_128x16x64_16x16_4x1_8x16x1_8x16x1_1x16x1x8_2_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_16x128x64_16x16_1x4_8x16x1_8x16x1_1x16x1x8_4_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_16x32x64_16x16_1x1_16x8x1_16x8x1_1x16x1x8_4_Intrawave_v1.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_16x32x64_16x16_1x1_32x4x1_32x4x1_1x16x1x8_4_Intrawave_v1.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_16x32x64_16x16_1x1_8x16x1_8x16x1_1x16x1x8_4_Intrawave_v1.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_16x32x64_16x16_1x1_8x16x1_8x16x1_1x16x1x8_4_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_16x64x64_16x16_1x2_8x16x1_8x16x1_1x16x1x8_4_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_32x16x64_16x16_1x1_8x16x1_8x16x1_1x16x1x8_2_Intrawave_v1.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_32x16x64_16x16_1x1_8x16x1_8x16x1_1x16x1x8_2_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_128_64x16x64_16x16_2x1_8x16x1_8x16x1_1x16x1x8_2_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_128x128x64_32x32_2x2_8x32x1_8x32x1_1x16x1x16_4_Intrawave_v1.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_128x128x64_32x32_2x2_8x32x1_8x32x1_1x16x1x16_4_Intrawave_v3.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_128x128x64_32x32_2x2_8x32x1_8x32x1_1x16x1x16_4_Intrawave_v5.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_16x256x64_16x16_1x4_8x16x1_8x16x1_1x16x1x16_4_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_224x256x64_16x16_7x8_8x32x1_8x32x1_1x16x1x16_4_Intrawave_v3.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x16x64_16x16_4x1_16x16x1_16x8x1_1x32x1x8_2_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x16x64_16x16_4x1_32x8x1_32x4x1_1x32x1x8_2_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x16x64_16x16_4x1_8x32x1_8x16x1_1x32x1x8_2_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x224x64_16x16_8x7_8x32x1_8x32x1_1x32x1x8_4_Intrawave_v3.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x256x32_32x32_4x4_16x16x1_16x16x1_1x16x1x16_4_Intrawave_v4.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x256x32_32x32_4x4_4x64x1_4x64x1_1x16x1x16_4_Intrawave_v3.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x256x32_32x32_4x4_4x64x1_4x64x1_1x16x1x16_4_Intrawave_v5.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_256_256x256x32_32x32_4x4_8x32x1_8x32x1_1x16x1x16_4_Intrawave_v4.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_64_16x16x64_16x16_1x1_8x8x1_8x8x1_1x16x1x4_4_Intrawave_v1.hip │ │ │ │ │ ├── bgemm_kernel_bf16bf16bf16_64_16x16x64_16x16_1x1_8x8x1_8x8x1_1x16x1x4_4_Intrawave_v2.hip │ │ │ │ │ ├── bgemm_kernel_collection.h │ │ │ │ │ └── bgemm_kernel_template.h │ │ │ │ ├── ck_bgemm.h │ │ │ │ ├── ck_bgemm_bfloat16.hip │ │ │ │ ├── ck_gemm.h │ │ │ │ ├── ck_gemm_bfloat16.hip │ │ │ │ ├── ck_gemm_float.hip │ │ │ │ ├── ck_gemm_half.hip │ │ │ │ ├── ck_gemm_template.h │ │ │ │ └── ck_types.h │ │ │ ├── im2col.h │ │ │ ├── im2col_shape_check.h │ │ │ ├── kleidiai │ │ │ │ ├── kai_kernels.cpp │ │ │ │ ├── kai_kernels.h │ │ │ │ ├── kai_pack.h │ │ │ │ ├── kai_ukernel_interface.cpp │ │ │ │ └── kai_ukernel_interface.h │ │ │ ├── layer_norm.cpp │ │ │ ├── layer_norm.h │ │ │ ├── metal │ │ │ │ ├── MetalAten.mm │ │ │ │ ├── MetalCommandBuffer.h │ │ │ │ ├── MetalCommandBuffer.mm │ │ │ │ ├── MetalContext.h │ │ │ │ ├── MetalContext.mm │ │ │ │ ├── MetalConvParams.h │ │ │ │ ├── MetalConvParams.mm │ │ │ │ ├── MetalDevice.h │ │ │ │ ├── MetalGuardImpl.cpp │ │ │ │ ├── MetalNeuronType.h │ │ │ │ ├── MetalPrepackOpContext.h │ │ │ │ ├── MetalPrepackOpRegister.cpp │ │ │ │ ├── MetalShaders.h │ │ │ │ ├── MetalTensorImpl.h │ │ │ │ ├── MetalTensorImplStorage.h │ │ │ │ ├── MetalTensorImplStorage.mm │ │ │ │ ├── MetalTensorUtils.h │ │ │ │ ├── MetalTensorUtils.mm │ │ │ │ ├── mpscnn │ │ │ │ │ ├── MPSCNNClampOp.h │ │ │ │ │ ├── MPSCNNClampOp.mm │ │ │ │ │ ├── MPSCNNConvOp.h │ │ │ │ │ ├── MPSCNNConvOp.mm │ │ │ │ │ ├── MPSCNNFullyConnectedOp.h │ │ │ │ │ ├── MPSCNNFullyConnectedOp.mm │ │ │ │ │ ├── MPSCNNNeuronOp.h │ │ │ │ │ ├── MPSCNNNeuronOp.mm │ │ │ │ │ ├── MPSCNNOp.h │ │ │ │ │ ├── MPSCNNUtils.h │ │ │ │ │ ├── MPSCNNUtils.mm │ │ │ │ │ ├── MPSImage+Tensor.h │ │ │ │ │ ├── MPSImage+Tensor.mm │ │ │ │ │ ├── MPSImageUtils.h │ │ │ │ │ ├── MPSImageUtils.mm │ │ │ │ │ ├── MPSImageWrapper.h │ │ │ │ │ ├── MPSImageWrapper.mm │ │ │ │ │ └── tests │ │ │ │ │ │ ├── MPSCNNTests.h │ │ │ │ │ │ ├── MPSCNNTests.mm │ │ │ │ │ │ ├── MetalOpTestRunner.h │ │ │ │ │ │ └── MetalOpTestRunner.mm │ │ │ │ └── ops │ │ │ │ │ ├── MetalAddmm.mm │ │ │ │ │ ├── MetalBinaryElementwise.mm │ │ │ │ │ ├── MetalChunk.mm │ │ │ │ │ ├── MetalClamp.mm │ │ │ │ │ ├── MetalConcat.mm │ │ │ │ │ ├── MetalConvolution.h │ │ │ │ │ ├── MetalConvolution.mm │ │ │ │ │ ├── MetalCopy.h │ │ │ │ │ ├── MetalCopy.mm │ │ │ │ │ ├── MetalHardshrink.mm │ │ │ │ │ ├── MetalHardswish.mm │ │ │ │ │ ├── MetalLeakyReLU.mm │ │ │ │ │ ├── MetalNeurons.mm │ │ │ │ │ ├── MetalPadding.mm │ │ │ │ │ ├── MetalPooling.mm │ │ │ │ │ ├── MetalReduce.mm │ │ │ │ │ ├── MetalReshape.mm │ │ │ │ │ ├── MetalSoftmax.mm │ │ │ │ │ ├── MetalTranspose.mm │ │ │ │ │ └── MetalUpsamplingNearest.mm │ │ │ ├── miopen │ │ │ │ ├── BatchNorm_miopen.cpp │ │ │ │ ├── Conv_miopen.cpp │ │ │ │ └── RNN_miopen.cpp │ │ │ ├── mkl │ │ │ │ ├── LinearAlgebra.cpp │ │ │ │ ├── LinearAlgebra.h │ │ │ │ ├── MklAllocationHelper.cpp │ │ │ │ ├── SparseBlasImpl.cpp │ │ │ │ ├── SparseBlasImpl.h │ │ │ │ ├── SparseCsrLinearAlgebra.cpp │ │ │ │ ├── SparseCsrLinearAlgebra.h │ │ │ │ └── SpectralOps.cpp │ │ │ ├── mkldnn │ │ │ │ ├── BinaryOps.cpp │ │ │ │ ├── Common.h │ │ │ │ ├── Conv.cpp │ │ │ │ ├── Conv.h │ │ │ │ ├── ConvPrepack.cpp │ │ │ │ ├── ConvPrepack.h │ │ │ │ ├── Copy.cpp │ │ │ │ ├── Gelu.cpp │ │ │ │ ├── IDeepRegistration.cpp │ │ │ │ ├── Linear.cpp │ │ │ │ ├── Linear.h │ │ │ │ ├── MKLDNNCommon.cpp │ │ │ │ ├── MKLDNNCommon.h │ │ │ │ ├── MKLDNNConversions.cpp │ │ │ │ ├── Matmul.cpp │ │ │ │ ├── Matmul.h │ │ │ │ ├── MkldnnTensorMath.cpp │ │ │ │ ├── Normalization.cpp │ │ │ │ ├── OpContext.cpp │ │ │ │ ├── OpContext.h │ │ │ │ ├── Pooling.cpp │ │ │ │ ├── Prelu.cpp │ │ │ │ ├── RNN.cpp │ │ │ │ ├── RegisterMkldnnOpContextClass.cpp │ │ │ │ ├── Relu.cpp │ │ │ │ ├── SoftMax.cpp │ │ │ │ ├── TensorFactories.cpp │ │ │ │ ├── TensorShape.cpp │ │ │ │ ├── TensorShape.h │ │ │ │ ├── UnaryOps.cpp │ │ │ │ ├── Utils.cpp │ │ │ │ ├── Utils.h │ │ │ │ └── xpu │ │ │ │ │ ├── Attention.cpp │ │ │ │ │ ├── Blas.cpp │ │ │ │ │ ├── Conv.cpp │ │ │ │ │ ├── detail │ │ │ │ │ ├── Attention.cpp │ │ │ │ │ ├── Attr.h │ │ │ │ │ ├── Conv.cpp │ │ │ │ │ ├── Deconv.cpp │ │ │ │ │ ├── Matmul.cpp │ │ │ │ │ ├── QConv.cpp │ │ │ │ │ ├── QMatmul.cpp │ │ │ │ │ ├── Utils.cpp │ │ │ │ │ ├── Utils.h │ │ │ │ │ ├── oneDNN.h │ │ │ │ │ ├── oneDNNContext.cpp │ │ │ │ │ └── oneDNNContext.h │ │ │ │ │ ├── qconv.cpp │ │ │ │ │ └── qlinear.cpp │ │ │ ├── mps │ │ │ │ ├── Copy.h │ │ │ │ ├── MPSGraphSequoiaOps.h │ │ │ │ ├── MPSGraphSonomaOps.h │ │ │ │ ├── MPSGraphVenturaOps.h │ │ │ │ ├── MetalShaderLibrary.h │ │ │ │ ├── OperationUtils.h │ │ │ │ ├── OperationUtils.mm │ │ │ │ ├── TensorFactory.cpp │ │ │ │ ├── TensorFactory.h │ │ │ │ ├── kernels │ │ │ │ │ ├── BinaryKernel.metal │ │ │ │ │ ├── Bucketization.metal │ │ │ │ │ ├── CrossKernel.metal │ │ │ │ │ ├── FusedOptimizerOps.metal │ │ │ │ │ ├── Gamma.metal │ │ │ │ │ ├── HistogramKernel.metal │ │ │ │ │ ├── Im2Col.metal │ │ │ │ │ ├── Indexing.metal │ │ │ │ │ ├── LinearAlgebra.metal │ │ │ │ │ ├── Quantized.metal │ │ │ │ │ ├── RMSNorm.metal │ │ │ │ │ ├── RenormKernel.metal │ │ │ │ │ ├── Repeat.metal │ │ │ │ │ ├── SpecialOps.metal │ │ │ │ │ ├── TriangularOps.metal │ │ │ │ │ ├── UnaryKernel.metal │ │ │ │ │ ├── UnfoldBackward.metal │ │ │ │ │ └── UpSample.metal │ │ │ │ └── operations │ │ │ │ │ ├── Activation.mm │ │ │ │ │ ├── AdaptivePooling.mm │ │ │ │ │ ├── Attention.mm │ │ │ │ │ ├── BinaryKernel.h │ │ │ │ │ ├── BinaryKernel.mm │ │ │ │ │ ├── BinaryOps.mm │ │ │ │ │ ├── BitwiseOps.mm │ │ │ │ │ ├── Blas.mm │ │ │ │ │ ├── Bucketization.mm │ │ │ │ │ ├── ConstantOps.mm │ │ │ │ │ ├── Convolution.mm │ │ │ │ │ ├── Copy.mm │ │ │ │ │ ├── CrossKernel.mm │ │ │ │ │ ├── Distributions.mm │ │ │ │ │ ├── Equal.cpp │ │ │ │ │ ├── Eye.mm │ │ │ │ │ ├── FastFourierTransform.mm │ │ │ │ │ ├── FusedAdamAmsgradKernelImpl.h │ │ │ │ │ ├── FusedAdamAmsgradKernelImpl.mm │ │ │ │ │ ├── FusedAdamKernel.mm │ │ │ │ │ ├── FusedAdamKernelImpl.h │ │ │ │ │ ├── FusedAdamKernelImpl.mm │ │ │ │ │ ├── FusedAdamWAmsgradKernelImpl.h │ │ │ │ │ ├── FusedAdamWAmsgradKernelImpl.mm │ │ │ │ │ ├── FusedAdamWKernel.mm │ │ │ │ │ ├── FusedAdamWKernelImpl.h │ │ │ │ │ ├── FusedAdamWKernelImpl.mm │ │ │ │ │ ├── FusedSgdKernel.mm │ │ │ │ │ ├── Gamma.mm │ │ │ │ │ ├── GridSampler.mm │ │ │ │ │ ├── HistogramKernel.mm │ │ │ │ │ ├── Im2Col.mm │ │ │ │ │ ├── Indexing.h │ │ │ │ │ ├── Indexing.mm │ │ │ │ │ ├── Inverse.mm │ │ │ │ │ ├── Lerp.mm │ │ │ │ │ ├── Linear.mm │ │ │ │ │ ├── LinearAlgebra.mm │ │ │ │ │ ├── LossOps.mm │ │ │ │ │ ├── MultiTensorApply.h │ │ │ │ │ ├── Normalization.mm │ │ │ │ │ ├── Pad.mm │ │ │ │ │ ├── PixelShuffle.mm │ │ │ │ │ ├── PointwiseOps.mm │ │ │ │ │ ├── Pooling.mm │ │ │ │ │ ├── Quantized.mm │ │ │ │ │ ├── RMSNorm.h │ │ │ │ │ ├── RMSNorm.mm │ │ │ │ │ ├── RangeFactories.mm │ │ │ │ │ ├── ReduceOps.mm │ │ │ │ │ ├── RenormKernel.mm │ │ │ │ │ ├── Repeat.mm │ │ │ │ │ ├── RnnOps.mm │ │ │ │ │ ├── Scalar.mm │ │ │ │ │ ├── ScatterGather.mm │ │ │ │ │ ├── Shape.mm │ │ │ │ │ ├── SoftMax.mm │ │ │ │ │ ├── Sort.mm │ │ │ │ │ ├── SpecialOps.mm │ │ │ │ │ ├── SummaryOps.mm │ │ │ │ │ ├── TensorCompare.mm │ │ │ │ │ ├── TriangularOps.mm │ │ │ │ │ ├── UnaryKernel.mm │ │ │ │ │ ├── UnaryOps.mm │ │ │ │ │ ├── UnfoldBackward.mm │ │ │ │ │ ├── Unique.mm │ │ │ │ │ ├── UpSample.mm │ │ │ │ │ ├── View.mm │ │ │ │ │ └── WeightNorm.mm │ │ │ ├── native_functions.yaml │ │ │ ├── nested │ │ │ │ ├── NestedTensorAliases.cpp │ │ │ │ ├── NestedTensorBackward.cpp │ │ │ │ ├── NestedTensorBinaryOps.cpp │ │ │ │ ├── NestedTensorBinaryOps.h │ │ │ │ ├── NestedTensorFactories.cpp │ │ │ │ ├── NestedTensorMath.cpp │ │ │ │ ├── NestedTensorMath.h │ │ │ │ ├── NestedTensorMatmul.cpp │ │ │ │ ├── NestedTensorTransformerFunctions.cpp │ │ │ │ ├── NestedTensorTransformerFunctions.h │ │ │ │ ├── NestedTensorTransformerUtils.h │ │ │ │ ├── NestedTensorUnaryOps.cpp │ │ │ │ ├── NestedTensorUtils.cpp │ │ │ │ ├── NestedTensorUtils.h │ │ │ │ ├── README.md │ │ │ │ └── cuda │ │ │ │ │ ├── NestedTensorBinaryOps.cu │ │ │ │ │ ├── NestedTensorMatmul.cu │ │ │ │ │ ├── NestedTensorTransformerFunctions.cpp │ │ │ │ │ ├── NestedTensorTransformerFunctions.cu │ │ │ │ │ └── NestedTensorTransformerUtils.cpp │ │ │ ├── prim_native_functions.cpp │ │ │ ├── quantized │ │ │ │ ├── AffineQuantizer.cpp │ │ │ │ ├── AffineQuantizer.h │ │ │ │ ├── AffineQuantizerBase.cpp │ │ │ │ ├── AffineQuantizerBase.h │ │ │ │ ├── ConvUtils.h │ │ │ │ ├── Copy.cpp │ │ │ │ ├── Copy.h │ │ │ │ ├── FakeQuantAffine.h │ │ │ │ ├── FakeQuantPerChannelAffine.cpp │ │ │ │ ├── FakeQuantPerTensorAffine.cpp │ │ │ │ ├── IndexKernel.h │ │ │ │ ├── PackedParams.h │ │ │ │ ├── QTensor.cpp │ │ │ │ ├── README.md │ │ │ │ ├── TensorAdvancedIndexing.cpp │ │ │ │ ├── TensorCompare.cpp │ │ │ │ ├── TensorFactories.cpp │ │ │ │ ├── cpu │ │ │ │ │ ├── AdaptiveAveragePooling.cpp │ │ │ │ │ ├── AveragePool2d.cpp │ │ │ │ │ ├── AveragePool3d.cpp │ │ │ │ │ ├── BinaryOps.cpp │ │ │ │ │ ├── BinaryOps.h │ │ │ │ │ ├── ChannelShuffle.cpp │ │ │ │ │ ├── EmbeddingPackedParams.h │ │ │ │ │ ├── IntReprQuant.cpp │ │ │ │ │ ├── LinearUnpackImpl.cpp │ │ │ │ │ ├── MakePerTensorQuantizedTensor.cpp │ │ │ │ │ ├── Normalization.cpp │ │ │ │ │ ├── OnednnUtils.h │ │ │ │ │ ├── Pooling.cpp │ │ │ │ │ ├── QnnpackUtils.h │ │ │ │ │ ├── QuantUtils.h │ │ │ │ │ ├── QuantizedOps.h │ │ │ │ │ ├── ReduceOps.cpp │ │ │ │ │ ├── RuyUtils.cpp │ │ │ │ │ ├── RuyUtils.h │ │ │ │ │ ├── Sorting.cpp │ │ │ │ │ ├── TensorOperators.cpp │ │ │ │ │ ├── TensorShape.cpp │ │ │ │ │ ├── UpSampleBilinear2d.cpp │ │ │ │ │ ├── UpSampleNearest2d.cpp │ │ │ │ │ ├── UpSampleNearest3d.cpp │ │ │ │ │ ├── XnnpackUtils.cpp │ │ │ │ │ ├── XnnpackUtils.h │ │ │ │ │ ├── conv_serialization.h │ │ │ │ │ ├── fbgemm_utils.cpp │ │ │ │ │ ├── fbgemm_utils.h │ │ │ │ │ ├── fused_obs_fake_quant.cpp │ │ │ │ │ ├── init_qnnpack.cpp │ │ │ │ │ ├── init_qnnpack.h │ │ │ │ │ ├── kernels │ │ │ │ │ │ ├── QuantizedOpKernels.cpp │ │ │ │ │ │ └── README.md │ │ │ │ │ ├── qclamp.cpp │ │ │ │ │ ├── qconv.cpp │ │ │ │ │ ├── qconv.h │ │ │ │ │ ├── qconv_dynamic.cpp │ │ │ │ │ ├── qconv_prepack.cpp │ │ │ │ │ ├── qconv_unpack_impl.cpp │ │ │ │ │ ├── qdropout.cpp │ │ │ │ │ ├── qelu.cpp │ │ │ │ │ ├── qembeddingbag.cpp │ │ │ │ │ ├── qembeddingbag.h │ │ │ │ │ ├── qembeddingbag_prepack.cpp │ │ │ │ │ ├── qembeddingbag_prepack.h │ │ │ │ │ ├── qembeddingbag_unpack.cpp │ │ │ │ │ ├── qgelu.cpp │ │ │ │ │ ├── qhardsigmoid.cpp │ │ │ │ │ ├── qhardswish.cpp │ │ │ │ │ ├── qlinear.cpp │ │ │ │ │ ├── qlinear.h │ │ │ │ │ ├── qlinear_dynamic.cpp │ │ │ │ │ ├── qlinear_prepack.cpp │ │ │ │ │ ├── qmatmul.cpp │ │ │ │ │ ├── qmul.cpp │ │ │ │ │ ├── qnnpack │ │ │ │ │ │ ├── .gitignore │ │ │ │ │ │ ├── BUCK.oss │ │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ │ ├── CODE_OF_CONDUCT.md │ │ │ │ │ │ ├── CONTRIBUTING.md │ │ │ │ │ │ ├── LICENSE │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── bench │ │ │ │ │ │ │ ├── add.cc │ │ │ │ │ │ │ ├── average-pooling.cc │ │ │ │ │ │ │ ├── channel-shuffle.cc │ │ │ │ │ │ │ ├── convolution.cc │ │ │ │ │ │ │ ├── global-average-pooling.cc │ │ │ │ │ │ │ ├── hardsigmoid.cc │ │ │ │ │ │ │ ├── hardswish.cc │ │ │ │ │ │ │ ├── hgemm.cc │ │ │ │ │ │ │ ├── max-pooling.cc │ │ │ │ │ │ │ ├── q8gemm.cc │ │ │ │ │ │ │ ├── q8gemm_sparse.cc │ │ │ │ │ │ │ ├── requantization.cc │ │ │ │ │ │ │ ├── sgemm.cc │ │ │ │ │ │ │ ├── sigmoid.cc │ │ │ │ │ │ │ ├── softargmax.cc │ │ │ │ │ │ │ └── tanh.cc │ │ │ │ │ │ ├── buckbuild.bzl │ │ │ │ │ │ ├── cmake │ │ │ │ │ │ │ ├── DownloadCpuinfo.cmake │ │ │ │ │ │ │ ├── DownloadFP16.cmake │ │ │ │ │ │ │ ├── DownloadFXdiv.cmake │ │ │ │ │ │ │ ├── DownloadGoogleBenchmark.cmake │ │ │ │ │ │ │ ├── DownloadGoogleTest.cmake │ │ │ │ │ │ │ ├── DownloadPSimd.cmake │ │ │ │ │ │ │ └── DownloadPThreadPool.cmake │ │ │ │ │ │ ├── configure.py │ │ │ │ │ │ ├── confu.yaml │ │ │ │ │ │ ├── deps │ │ │ │ │ │ │ └── clog │ │ │ │ │ │ │ │ ├── .gitignore │ │ │ │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ │ │ │ ├── LICENSE │ │ │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ │ │ ├── cmake │ │ │ │ │ │ │ │ └── DownloadGoogleTest.cmake │ │ │ │ │ │ │ │ ├── configure.py │ │ │ │ │ │ │ │ ├── confu.yaml │ │ │ │ │ │ │ │ ├── include │ │ │ │ │ │ │ │ └── clog.h │ │ │ │ │ │ │ │ ├── src │ │ │ │ │ │ │ │ └── clog.c │ │ │ │ │ │ │ │ └── test │ │ │ │ │ │ │ │ └── clog.cc │ │ │ │ │ │ ├── generate-wrapper.py │ │ │ │ │ │ ├── include │ │ │ │ │ │ │ ├── pack_block_sparse.h │ │ │ │ │ │ │ ├── pytorch_qnnpack.h │ │ │ │ │ │ │ └── qnnpack_func.h │ │ │ │ │ │ ├── scripts │ │ │ │ │ │ │ ├── build-android-arm64.sh │ │ │ │ │ │ │ ├── build-android-armv7.sh │ │ │ │ │ │ │ ├── build-android-x86.sh │ │ │ │ │ │ │ ├── build-ios-arm64.sh │ │ │ │ │ │ │ ├── build-ios-arm64e.sh │ │ │ │ │ │ │ ├── build-ios-armv7.sh │ │ │ │ │ │ │ ├── build-ios-armv7s.sh │ │ │ │ │ │ │ ├── build-ios-i386.sh │ │ │ │ │ │ │ ├── build-ios-x86_64.sh │ │ │ │ │ │ │ ├── build-local.sh │ │ │ │ │ │ │ ├── test-android-arm64.sh │ │ │ │ │ │ │ ├── test-android-armv7.sh │ │ │ │ │ │ │ └── test-android-x86.sh │ │ │ │ │ │ ├── src │ │ │ │ │ │ │ ├── add.c │ │ │ │ │ │ │ ├── average-pooling.c │ │ │ │ │ │ │ ├── channel-shuffle.c │ │ │ │ │ │ │ ├── clamp.c │ │ │ │ │ │ │ ├── conv-prepack.cc │ │ │ │ │ │ │ ├── conv-run.cc │ │ │ │ │ │ │ ├── convolution.c │ │ │ │ │ │ │ ├── deconv-run.cc │ │ │ │ │ │ │ ├── deconvolution.c │ │ │ │ │ │ │ ├── fc-dynamic-run.cc │ │ │ │ │ │ │ ├── fc-prepack.cc │ │ │ │ │ │ │ ├── fc-run.cc │ │ │ │ │ │ │ ├── fc-unpack.cc │ │ │ │ │ │ │ ├── fully-connected-sparse.c │ │ │ │ │ │ │ ├── fully-connected.c │ │ │ │ │ │ │ ├── global-average-pooling.c │ │ │ │ │ │ │ ├── hardsigmoid.c │ │ │ │ │ │ │ ├── hardswish.c │ │ │ │ │ │ │ ├── hgemm │ │ │ │ │ │ │ │ ├── 8x8-aarch32-neonfp16arith.S │ │ │ │ │ │ │ │ └── 8x8-neonfp16arith.c │ │ │ │ │ │ │ ├── indirection.c │ │ │ │ │ │ │ ├── init.c │ │ │ │ │ │ │ ├── leaky-relu.c │ │ │ │ │ │ │ ├── max-pooling.c │ │ │ │ │ │ │ ├── operator-delete.c │ │ │ │ │ │ │ ├── operator-run.c │ │ │ │ │ │ │ ├── q8avgpool │ │ │ │ │ │ │ │ ├── mp8x9p8q-neon.c │ │ │ │ │ │ │ │ ├── mp8x9p8q-sse2.c │ │ │ │ │ │ │ │ ├── up8x9-neon.c │ │ │ │ │ │ │ │ ├── up8x9-sse2.c │ │ │ │ │ │ │ │ ├── up8xm-neon.c │ │ │ │ │ │ │ │ └── up8xm-sse2.c │ │ │ │ │ │ │ ├── q8conv │ │ │ │ │ │ │ │ ├── 4x4c2-sse2.c │ │ │ │ │ │ │ │ ├── 4x8-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 4x8-neon.c │ │ │ │ │ │ │ │ ├── 8x8-aarch64-neon.S │ │ │ │ │ │ │ │ └── 8x8-neon.c │ │ │ │ │ │ │ ├── q8dwconv │ │ │ │ │ │ │ │ ├── mp8x25-neon-per-channel.c │ │ │ │ │ │ │ │ ├── mp8x25-neon.c │ │ │ │ │ │ │ │ ├── mp8x25-sse2-per-channel.c │ │ │ │ │ │ │ │ ├── mp8x25-sse2.c │ │ │ │ │ │ │ │ ├── mp8x27-neon.c │ │ │ │ │ │ │ │ ├── mp8x27-sse2.c │ │ │ │ │ │ │ │ ├── up8x9-aarch32-neon-per-channel.S │ │ │ │ │ │ │ │ ├── up8x9-aarch32-neon.S │ │ │ │ │ │ │ │ ├── up8x9-neon-per-channel.c │ │ │ │ │ │ │ │ ├── up8x9-neon.c │ │ │ │ │ │ │ │ ├── up8x9-sse2-per-channel.c │ │ │ │ │ │ │ │ └── up8x9-sse2.c │ │ │ │ │ │ │ ├── q8gavgpool │ │ │ │ │ │ │ │ ├── mp8x7p7q-neon.c │ │ │ │ │ │ │ │ ├── mp8x7p7q-sse2.c │ │ │ │ │ │ │ │ ├── up8x7-neon.c │ │ │ │ │ │ │ │ ├── up8x7-sse2.c │ │ │ │ │ │ │ │ ├── up8xm-neon.c │ │ │ │ │ │ │ │ └── up8xm-sse2.c │ │ │ │ │ │ │ ├── q8gemm │ │ │ │ │ │ │ │ ├── 2x4c8-sse2.c │ │ │ │ │ │ │ │ ├── 4x-sumrows-neon.c │ │ │ │ │ │ │ │ ├── 4x4c2-dq-sse2.c │ │ │ │ │ │ │ │ ├── 4x4c2-sse2.c │ │ │ │ │ │ │ │ ├── 4x8-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 4x8-dq-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 4x8-dq-neon.c │ │ │ │ │ │ │ │ ├── 4x8-neon.c │ │ │ │ │ │ │ │ ├── 4x8c2-xzp-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 4x8c2-xzp-neon.c │ │ │ │ │ │ │ │ ├── 6x4-neon.c │ │ │ │ │ │ │ │ ├── 8x8-aarch64-neon.S │ │ │ │ │ │ │ │ ├── 8x8-dq-aarch64-neon.S │ │ │ │ │ │ │ │ └── 8x8-neon.c │ │ │ │ │ │ │ ├── q8gemm_sparse │ │ │ │ │ │ │ │ ├── 4x4-packA-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 4x8c1x4-dq-packedA-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 4x8c8x1-dq-packedA-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 8x4-packA-aarch32-neon.S │ │ │ │ │ │ │ │ ├── 8x4-packA-aarch64-neon.S │ │ │ │ │ │ │ │ ├── 8x4-packA-sse2.c │ │ │ │ │ │ │ │ ├── 8x4c1x4-dq-packedA-sse2.c │ │ │ │ │ │ │ │ ├── 8x4c1x4-dq-packedA-sse2.h │ │ │ │ │ │ │ │ ├── 8x4c1x4-packed-sse2.h │ │ │ │ │ │ │ │ ├── 8x8c1x4-dq-packedA-aarch64-neon.S │ │ │ │ │ │ │ │ └── 8x8c8x1-dq-packedA-aarch64-neon.S │ │ │ │ │ │ │ ├── q8vadd │ │ │ │ │ │ │ │ ├── neon.c │ │ │ │ │ │ │ │ └── sse2.c │ │ │ │ │ │ │ ├── qnnpack │ │ │ │ │ │ │ │ ├── AlignedAllocator.h │ │ │ │ │ │ │ │ ├── assembly.h │ │ │ │ │ │ │ │ ├── common.h │ │ │ │ │ │ │ │ ├── hgemm.h │ │ │ │ │ │ │ │ ├── indirection.h │ │ │ │ │ │ │ │ ├── isa-checks.h │ │ │ │ │ │ │ │ ├── log.h │ │ │ │ │ │ │ │ ├── math.h │ │ │ │ │ │ │ │ ├── operator.h │ │ │ │ │ │ │ │ ├── pack.h │ │ │ │ │ │ │ │ ├── params.h │ │ │ │ │ │ │ │ ├── q8avgpool.h │ │ │ │ │ │ │ │ ├── q8conv.h │ │ │ │ │ │ │ │ ├── q8dwconv.h │ │ │ │ │ │ │ │ ├── q8gavgpool.h │ │ │ │ │ │ │ │ ├── q8gemm.h │ │ │ │ │ │ │ │ ├── q8gemm_sparse.h │ │ │ │ │ │ │ │ ├── q8vadd.h │ │ │ │ │ │ │ │ ├── requantization-stubs.h │ │ │ │ │ │ │ │ ├── requantization.h │ │ │ │ │ │ │ │ ├── scalar-utils.h │ │ │ │ │ │ │ │ ├── sconv.h │ │ │ │ │ │ │ │ ├── sdwconv.h │ │ │ │ │ │ │ │ ├── sgemm.h │ │ │ │ │ │ │ │ ├── u8clamp.h │ │ │ │ │ │ │ │ ├── u8lut32norm.h │ │ │ │ │ │ │ │ ├── u8maxpool.h │ │ │ │ │ │ │ │ ├── u8rmax.h │ │ │ │ │ │ │ │ ├── x8lut.h │ │ │ │ │ │ │ │ └── x8zip.h │ │ │ │ │ │ │ ├── requantization │ │ │ │ │ │ │ │ ├── fp32-neon.c │ │ │ │ │ │ │ │ ├── fp32-psimd.c │ │ │ │ │ │ │ │ ├── fp32-scalar.c │ │ │ │ │ │ │ │ ├── fp32-sse2.c │ │ │ │ │ │ │ │ ├── gemmlowp-neon.c │ │ │ │ │ │ │ │ ├── gemmlowp-scalar.c │ │ │ │ │ │ │ │ ├── gemmlowp-scalar.h │ │ │ │ │ │ │ │ ├── gemmlowp-sse.h │ │ │ │ │ │ │ │ ├── gemmlowp-sse2.c │ │ │ │ │ │ │ │ ├── gemmlowp-sse4.c │ │ │ │ │ │ │ │ ├── gemmlowp-ssse3.c │ │ │ │ │ │ │ │ ├── precise-neon.c │ │ │ │ │ │ │ │ ├── precise-psimd.c │ │ │ │ │ │ │ │ ├── precise-scalar.c │ │ │ │ │ │ │ │ ├── precise-sse2.c │ │ │ │ │ │ │ │ ├── precise-sse4.c │ │ │ │ │ │ │ │ ├── precise-ssse3.c │ │ │ │ │ │ │ │ ├── q31-neon.c │ │ │ │ │ │ │ │ ├── q31-scalar.c │ │ │ │ │ │ │ │ ├── q31-sse2.c │ │ │ │ │ │ │ │ ├── q31-sse4.c │ │ │ │ │ │ │ │ ├── q31-ssse3.c │ │ │ │ │ │ │ │ ├── runtime-assembly.h │ │ │ │ │ │ │ │ ├── runtime-neon.h │ │ │ │ │ │ │ │ └── runtime-sse2.h │ │ │ │ │ │ │ ├── sconv │ │ │ │ │ │ │ │ └── 6x8-psimd.c │ │ │ │ │ │ │ ├── sdwconv │ │ │ │ │ │ │ │ └── up4x9-psimd.c │ │ │ │ │ │ │ ├── sgemm │ │ │ │ │ │ │ │ ├── 5x8-neon.c │ │ │ │ │ │ │ │ ├── 6x8-neon.c │ │ │ │ │ │ │ │ └── 6x8-psimd.c │ │ │ │ │ │ │ ├── sigmoid.c │ │ │ │ │ │ │ ├── softargmax.c │ │ │ │ │ │ │ ├── tanh.c │ │ │ │ │ │ │ ├── u8clamp │ │ │ │ │ │ │ │ ├── neon.c │ │ │ │ │ │ │ │ └── sse2.c │ │ │ │ │ │ │ ├── u8lut32norm │ │ │ │ │ │ │ │ └── scalar.c │ │ │ │ │ │ │ ├── u8maxpool │ │ │ │ │ │ │ │ ├── 16x9p8q-neon.c │ │ │ │ │ │ │ │ ├── 16x9p8q-sse2.c │ │ │ │ │ │ │ │ ├── sub16-neon.c │ │ │ │ │ │ │ │ └── sub16-sse2.c │ │ │ │ │ │ │ ├── u8rmax │ │ │ │ │ │ │ │ ├── neon.c │ │ │ │ │ │ │ │ └── sse2.c │ │ │ │ │ │ │ ├── x8lut │ │ │ │ │ │ │ │ └── scalar.c │ │ │ │ │ │ │ └── x8zip │ │ │ │ │ │ │ │ ├── x2-neon.c │ │ │ │ │ │ │ │ ├── x2-sse2.c │ │ │ │ │ │ │ │ ├── x3-neon.c │ │ │ │ │ │ │ │ ├── x3-sse2.c │ │ │ │ │ │ │ │ ├── x4-neon.c │ │ │ │ │ │ │ │ ├── x4-sse2.c │ │ │ │ │ │ │ │ ├── xm-neon.c │ │ │ │ │ │ │ │ └── xm-sse2.c │ │ │ │ │ │ ├── test │ │ │ │ │ │ │ ├── add-operator-tester.h │ │ │ │ │ │ │ ├── add.cc │ │ │ │ │ │ │ ├── average-pooling-operator-tester.h │ │ │ │ │ │ │ ├── average-pooling.cc │ │ │ │ │ │ │ ├── avgpool-microkernel-tester.h │ │ │ │ │ │ │ ├── channel-shuffle-operator-tester.h │ │ │ │ │ │ │ ├── channel-shuffle.cc │ │ │ │ │ │ │ ├── clamp-microkernel-tester.h │ │ │ │ │ │ │ ├── clamp-operator-tester.h │ │ │ │ │ │ │ ├── clamp.cc │ │ │ │ │ │ │ ├── convolution-operator-tester.h │ │ │ │ │ │ │ ├── convolution.cc │ │ │ │ │ │ │ ├── deconvolution-operator-tester.h │ │ │ │ │ │ │ ├── deconvolution.cc │ │ │ │ │ │ │ ├── dwconv-microkernel-tester.h │ │ │ │ │ │ │ ├── fully-connected-operator-tester.h │ │ │ │ │ │ │ ├── fully-connected-sparse-operator-tester.h │ │ │ │ │ │ │ ├── fully-connected-sparse.cc │ │ │ │ │ │ │ ├── fully-connected.cc │ │ │ │ │ │ │ ├── gavgpool-microkernel-tester.h │ │ │ │ │ │ │ ├── gemm-block-sparse-microkernel-tester.h │ │ │ │ │ │ │ ├── gemm-microkernel-tester.h │ │ │ │ │ │ │ ├── global-average-pooling-operator-tester.h │ │ │ │ │ │ │ ├── global-average-pooling.cc │ │ │ │ │ │ │ ├── hardsigmoid-operator-tester.h │ │ │ │ │ │ │ ├── hardsigmoid.cc │ │ │ │ │ │ │ ├── hardswish-operator-tester.h │ │ │ │ │ │ │ ├── hardswish.cc │ │ │ │ │ │ │ ├── hgemm.cc │ │ │ │ │ │ │ ├── leaky-relu-operator-tester.h │ │ │ │ │ │ │ ├── leaky-relu.cc │ │ │ │ │ │ │ ├── lut-microkernel-tester.h │ │ │ │ │ │ │ ├── lut-norm-microkernel-tester.h │ │ │ │ │ │ │ ├── max-pooling-operator-tester.h │ │ │ │ │ │ │ ├── max-pooling.cc │ │ │ │ │ │ │ ├── maxpool-microkernel-tester.h │ │ │ │ │ │ │ ├── q8avgpool.cc │ │ │ │ │ │ │ ├── q8conv.cc │ │ │ │ │ │ │ ├── q8dwconv.cc │ │ │ │ │ │ │ ├── q8gavgpool.cc │ │ │ │ │ │ │ ├── q8gemm.cc │ │ │ │ │ │ │ ├── q8gemm_sparse.cc │ │ │ │ │ │ │ ├── q8vadd.cc │ │ │ │ │ │ │ ├── requantization-tester.h │ │ │ │ │ │ │ ├── requantization.cc │ │ │ │ │ │ │ ├── rmax-microkernel-tester.h │ │ │ │ │ │ │ ├── sconv.cc │ │ │ │ │ │ │ ├── sgemm.cc │ │ │ │ │ │ │ ├── sigmoid-operator-tester.h │ │ │ │ │ │ │ ├── sigmoid.cc │ │ │ │ │ │ │ ├── softargmax-operator-tester.h │ │ │ │ │ │ │ ├── softargmax.cc │ │ │ │ │ │ │ ├── tanh-operator-tester.h │ │ │ │ │ │ │ ├── tanh.cc │ │ │ │ │ │ │ ├── test_utils.h │ │ │ │ │ │ │ ├── u8clamp.cc │ │ │ │ │ │ │ ├── u8lut32norm.cc │ │ │ │ │ │ │ ├── u8maxpool.cc │ │ │ │ │ │ │ ├── u8rmax.cc │ │ │ │ │ │ │ ├── vadd-microkernel-tester.h │ │ │ │ │ │ │ ├── x8lut.cc │ │ │ │ │ │ │ ├── x8zip.cc │ │ │ │ │ │ │ └── zip-microkernel-tester.h │ │ │ │ │ │ └── wrappers │ │ │ │ │ │ │ ├── hgemm │ │ │ │ │ │ │ └── 8x8-aarch32-neonfp16arith.S │ │ │ │ │ │ │ ├── q8avgpool │ │ │ │ │ │ │ ├── mp8x9p8q-neon.c │ │ │ │ │ │ │ ├── mp8x9p8q-sse2.c │ │ │ │ │ │ │ ├── up8x9-neon.c │ │ │ │ │ │ │ ├── up8x9-sse2.c │ │ │ │ │ │ │ ├── up8xm-neon.c │ │ │ │ │ │ │ └── up8xm-sse2.c │ │ │ │ │ │ │ ├── q8conv │ │ │ │ │ │ │ ├── 4x4c2-sse2.c │ │ │ │ │ │ │ ├── 4x8-aarch32-neon.S │ │ │ │ │ │ │ ├── 4x8-neon.c │ │ │ │ │ │ │ ├── 8x8-aarch64-neon.S │ │ │ │ │ │ │ └── 8x8-neon.c │ │ │ │ │ │ │ ├── q8dwconv │ │ │ │ │ │ │ ├── mp8x25-neon-per-channel.c │ │ │ │ │ │ │ ├── mp8x25-neon.c │ │ │ │ │ │ │ ├── mp8x25-sse2-per-channel.c │ │ │ │ │ │ │ ├── mp8x25-sse2.c │ │ │ │ │ │ │ ├── mp8x27-neon.c │ │ │ │ │ │ │ ├── mp8x27-sse2.c │ │ │ │ │ │ │ ├── up8x9-aarch32-neon-per-channel.S │ │ │ │ │ │ │ ├── up8x9-aarch32-neon.S │ │ │ │ │ │ │ ├── up8x9-neon-per-channel.c │ │ │ │ │ │ │ ├── up8x9-neon.c │ │ │ │ │ │ │ ├── up8x9-sse2-per-channel.c │ │ │ │ │ │ │ └── up8x9-sse2.c │ │ │ │ │ │ │ ├── q8gavgpool │ │ │ │ │ │ │ ├── mp8x7p7q-neon.c │ │ │ │ │ │ │ ├── mp8x7p7q-sse2.c │ │ │ │ │ │ │ ├── up8x7-neon.c │ │ │ │ │ │ │ ├── up8x7-sse2.c │ │ │ │ │ │ │ ├── up8xm-neon.c │ │ │ │ │ │ │ └── up8xm-sse2.c │ │ │ │ │ │ │ ├── q8gemm │ │ │ │ │ │ │ ├── 2x4c8-sse2.c │ │ │ │ │ │ │ ├── 4x-sumrows-neon.c │ │ │ │ │ │ │ ├── 4x4c2-dq-sse2.c │ │ │ │ │ │ │ ├── 4x4c2-sse2.c │ │ │ │ │ │ │ ├── 4x8-aarch32-neon.S │ │ │ │ │ │ │ ├── 4x8-dq-aarch32-neon.S │ │ │ │ │ │ │ ├── 4x8-dq-neon.c │ │ │ │ │ │ │ ├── 4x8-neon.c │ │ │ │ │ │ │ ├── 4x8c2-xzp-aarch32-neon.S │ │ │ │ │ │ │ ├── 4x8c2-xzp-neon.c │ │ │ │ │ │ │ ├── 6x4-neon.c │ │ │ │ │ │ │ ├── 8x8-aarch64-neon.S │ │ │ │ │ │ │ ├── 8x8-dq-aarch64-neon.S │ │ │ │ │ │ │ └── 8x8-neon.c │ │ │ │ │ │ │ ├── q8gemm_sparse │ │ │ │ │ │ │ ├── 4x4-packA-aarch32-neon.S │ │ │ │ │ │ │ ├── 4x8c1x4-dq-packedA-aarch32-neon.S │ │ │ │ │ │ │ ├── 4x8c8x1-dq-packedA-aarch32-neon.S │ │ │ │ │ │ │ ├── 8x4-packA-aarch32-neon.S │ │ │ │ │ │ │ ├── 8x4-packA-aarch64-neon.S │ │ │ │ │ │ │ ├── 8x4c1x4-packed-sse2.c │ │ │ │ │ │ │ ├── 8x8c1x4-dq-packedA-aarch64-neon.S │ │ │ │ │ │ │ └── 8x8c8x1-dq-packedA-aarch64-neon.S │ │ │ │ │ │ │ ├── q8vadd │ │ │ │ │ │ │ ├── neon.c │ │ │ │ │ │ │ └── sse2.c │ │ │ │ │ │ │ ├── requantization │ │ │ │ │ │ │ ├── fp32-neon.c │ │ │ │ │ │ │ ├── fp32-psimd.c │ │ │ │ │ │ │ ├── fp32-scalar.c │ │ │ │ │ │ │ ├── fp32-sse2.c │ │ │ │ │ │ │ ├── gemmlowp-neon.c │ │ │ │ │ │ │ ├── gemmlowp-scalar.c │ │ │ │ │ │ │ ├── gemmlowp-sse2.c │ │ │ │ │ │ │ ├── gemmlowp-sse4.c │ │ │ │ │ │ │ ├── gemmlowp-ssse3.c │ │ │ │ │ │ │ ├── precise-neon.c │ │ │ │ │ │ │ ├── precise-psimd.c │ │ │ │ │ │ │ ├── precise-scalar.c │ │ │ │ │ │ │ ├── precise-sse2.c │ │ │ │ │ │ │ ├── precise-sse4.c │ │ │ │ │ │ │ ├── precise-ssse3.c │ │ │ │ │ │ │ ├── q31-neon.c │ │ │ │ │ │ │ ├── q31-scalar.c │ │ │ │ │ │ │ ├── q31-sse2.c │ │ │ │ │ │ │ ├── q31-sse4.c │ │ │ │ │ │ │ └── q31-ssse3.c │ │ │ │ │ │ │ ├── sgemm │ │ │ │ │ │ │ ├── 5x8-neon.c │ │ │ │ │ │ │ ├── 6x8-neon.c │ │ │ │ │ │ │ └── 6x8-psimd.c │ │ │ │ │ │ │ ├── u8clamp │ │ │ │ │ │ │ ├── neon.c │ │ │ │ │ │ │ └── sse2.c │ │ │ │ │ │ │ ├── u8lut32norm │ │ │ │ │ │ │ └── scalar.c │ │ │ │ │ │ │ ├── u8maxpool │ │ │ │ │ │ │ ├── 16x9p8q-neon.c │ │ │ │ │ │ │ ├── 16x9p8q-sse2.c │ │ │ │ │ │ │ ├── sub16-neon.c │ │ │ │ │ │ │ └── sub16-sse2.c │ │ │ │ │ │ │ ├── u8rmax │ │ │ │ │ │ │ ├── neon.c │ │ │ │ │ │ │ └── sse2.c │ │ │ │ │ │ │ ├── x8lut │ │ │ │ │ │ │ └── scalar.c │ │ │ │ │ │ │ └── x8zip │ │ │ │ │ │ │ ├── x2-neon.c │ │ │ │ │ │ │ ├── x2-sse2.c │ │ │ │ │ │ │ ├── x3-neon.c │ │ │ │ │ │ │ ├── x3-sse2.c │ │ │ │ │ │ │ ├── x4-neon.c │ │ │ │ │ │ │ ├── x4-sse2.c │ │ │ │ │ │ │ ├── xm-neon.c │ │ │ │ │ │ │ └── xm-sse2.c │ │ │ │ │ ├── qnormalization.cpp │ │ │ │ │ ├── qrelu.cpp │ │ │ │ │ ├── qsigmoid.cpp │ │ │ │ │ ├── qsoftmax.cpp │ │ │ │ │ ├── qtanh.cpp │ │ │ │ │ └── qthreshold.cpp │ │ │ │ ├── cuda │ │ │ │ │ ├── Activation.cpp │ │ │ │ │ ├── Activation.cu │ │ │ │ │ ├── AffineQuantizer.cu │ │ │ │ │ ├── EmbeddingBag.cu │ │ │ │ │ ├── FakeQuantizeCore.cu │ │ │ │ │ ├── FusedObsFakeQuant.cu │ │ │ │ │ ├── IntReprQuant.cu │ │ │ │ │ └── MakePerTensorQuantizedTensor.cu │ │ │ │ ├── cudnn │ │ │ │ │ ├── BinaryOps.cpp │ │ │ │ │ ├── Conv.cpp │ │ │ │ │ ├── ConvPrepack.cpp │ │ │ │ │ ├── ConvUnpackImpl.cpp │ │ │ │ │ ├── Linear.cpp │ │ │ │ │ ├── LinearPrepack.cpp │ │ │ │ │ ├── LinearUnpackImpl.cpp │ │ │ │ │ ├── Pooling.cpp │ │ │ │ │ └── utils.h │ │ │ │ ├── library.cpp │ │ │ │ ├── library.h │ │ │ │ ├── qconv_unpack.cpp │ │ │ │ └── qlinear_unpack.cpp │ │ │ ├── sparse │ │ │ │ ├── FlattenIndicesCommon.h │ │ │ │ ├── FlattenIndicesKernel.cpp │ │ │ │ ├── Macros.h │ │ │ │ ├── ParamUtils.cpp │ │ │ │ ├── ParamUtils.h │ │ │ │ ├── SoftMax.cpp │ │ │ │ ├── SparseBinaryOpIntersectionCommon.h │ │ │ │ ├── SparseBinaryOpIntersectionKernel.cpp │ │ │ │ ├── SparseBlas.cpp │ │ │ │ ├── SparseBlas.h │ │ │ │ ├── SparseBlasImpl.cpp │ │ │ │ ├── SparseBlasImpl.h │ │ │ │ ├── SparseCsrTensor.cpp │ │ │ │ ├── SparseCsrTensorMath.cpp │ │ │ │ ├── SparseCsrTensorMath.h │ │ │ │ ├── SparseFactories.cpp │ │ │ │ ├── SparseFactories.h │ │ │ │ ├── SparseMatMul.cpp │ │ │ │ ├── SparseStubs.h │ │ │ │ ├── SparseTensor.cpp │ │ │ │ ├── SparseTensorMath.cpp │ │ │ │ ├── SparseTensorMath.h │ │ │ │ ├── SparseUnaryOps.cpp │ │ │ │ ├── ValidateCompressedIndicesCommon.h │ │ │ │ ├── ValidateCompressedIndicesKernel.cpp │ │ │ │ └── cuda │ │ │ │ │ ├── ComputeSparseTile.h │ │ │ │ │ ├── SoftMax.cu │ │ │ │ │ ├── SparseBlas.cpp │ │ │ │ │ ├── SparseBlasImpl.cpp │ │ │ │ │ ├── SparseBlasImpl.h │ │ │ │ │ ├── SparseBlasLegacy.cpp │ │ │ │ │ ├── SparseBlasLegacy.h │ │ │ │ │ ├── SparseCUDAApplyUtils.cuh │ │ │ │ │ ├── SparseCUDABlas.cpp │ │ │ │ │ ├── SparseCUDABlas.h │ │ │ │ │ ├── SparseCUDATensor.cu │ │ │ │ │ ├── SparseCUDATensorMath.cu │ │ │ │ │ ├── SparseCUDATensorMath.cuh │ │ │ │ │ ├── SparseCsrTensorMath.cu │ │ │ │ │ ├── SparseMatMul.cu │ │ │ │ │ ├── SparseSemiStructuredApplyDense.cu │ │ │ │ │ ├── SparseSemiStructuredLinear.cu │ │ │ │ │ ├── SparseSemiStructuredOps.cu │ │ │ │ │ ├── SparseSemiStructuredPack.h │ │ │ │ │ ├── SparseSemiStructuredTile.cu │ │ │ │ │ ├── SparseSemiSturcturedApply.cu │ │ │ │ │ ├── StaticSort.h │ │ │ │ │ ├── cuSPARSELtOps.cpp │ │ │ │ │ └── cuSPARSELtOps.h │ │ │ ├── tags.yaml │ │ │ ├── transformers │ │ │ │ ├── attention.cpp │ │ │ │ ├── attention.h │ │ │ │ ├── cuda │ │ │ │ │ ├── attention.cu │ │ │ │ │ ├── attention_backward.cu │ │ │ │ │ ├── flash_attn │ │ │ │ │ │ ├── flash_api.cpp │ │ │ │ │ │ ├── flash_api.h │ │ │ │ │ │ ├── philox.cuh │ │ │ │ │ │ └── static_switch.h │ │ │ │ │ ├── mem_eff_attention │ │ │ │ │ │ ├── debug_utils.h │ │ │ │ │ │ ├── epilogue │ │ │ │ │ │ │ ├── epilogue_pipelined.h │ │ │ │ │ │ │ ├── epilogue_rescale_output.h │ │ │ │ │ │ │ └── epilogue_thread_apply_logsumexp.h │ │ │ │ │ │ ├── gemm │ │ │ │ │ │ │ ├── custom_mma.h │ │ │ │ │ │ │ ├── custom_mma_base.h │ │ │ │ │ │ │ ├── custom_mma_multistage.h │ │ │ │ │ │ │ ├── custom_mma_pipelined.h │ │ │ │ │ │ │ ├── find_default_mma.h │ │ │ │ │ │ │ ├── mma_accum_lambda_iterator.h │ │ │ │ │ │ │ └── mma_from_smem.h │ │ │ │ │ │ ├── gemm_kernel_utils.h │ │ │ │ │ │ ├── iterators │ │ │ │ │ │ │ ├── default_warp_iterator_from_smem.h │ │ │ │ │ │ │ ├── epilogue_predicated_tile_iterator.h │ │ │ │ │ │ │ ├── make_residual_last.h │ │ │ │ │ │ │ ├── predicated_tile_access_iterator_residual_last.h │ │ │ │ │ │ │ ├── predicated_tile_iterator_residual_last.h │ │ │ │ │ │ │ ├── transpose_warp_iterator.h │ │ │ │ │ │ │ └── warp_iterator_from_smem.h │ │ │ │ │ │ ├── kernel_backward.h │ │ │ │ │ │ ├── kernel_forward.h │ │ │ │ │ │ ├── kernels │ │ │ │ │ │ │ ├── cutlassB.h │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k128.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k128_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k32.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k32_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k64.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k64_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k65536.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k65536_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_bf16_aligned_k96.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k128.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k128_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k32.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k32_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k64.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k64_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k65536.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k65536_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f16_aligned_k96.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k128.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k128_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k32.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k32_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k64.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k64_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k65536.cu │ │ │ │ │ │ │ ├── cutlassB_f16_notaligned_k65536_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k128.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k128_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k32.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k32_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k64.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k64_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k65536.cu │ │ │ │ │ │ │ ├── cutlassB_f32_aligned_k65536_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k128.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k128_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k32.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k32_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k64.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k64_dropout.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k65536.cu │ │ │ │ │ │ │ ├── cutlassB_f32_notaligned_k65536_dropout.cu │ │ │ │ │ │ │ ├── cutlassF.h │ │ │ │ │ │ │ ├── cutlassF_bf16_aligned.cu │ │ │ │ │ │ │ ├── cutlassF_f16_aligned.cu │ │ │ │ │ │ │ ├── cutlassF_f16_notaligned.cu │ │ │ │ │ │ │ ├── cutlassF_f32_aligned.cu │ │ │ │ │ │ │ ├── cutlassF_f32_notaligned.cu │ │ │ │ │ │ │ └── generate_kernels.py │ │ │ │ │ │ ├── pytorch_utils.h │ │ │ │ │ │ └── transform │ │ │ │ │ │ │ └── tile_smem_loader.h │ │ │ │ │ ├── sdp_utils.cpp │ │ │ │ │ └── sdp_utils.h │ │ │ │ ├── hip │ │ │ │ │ ├── aotriton_adapter.h │ │ │ │ │ └── flash_attn │ │ │ │ │ │ ├── aot │ │ │ │ │ │ └── mha_all_aot.hip │ │ │ │ │ │ ├── ck │ │ │ │ │ │ ├── CMakeLists.txt │ │ │ │ │ │ ├── add_make_kernel_pt.sh │ │ │ │ │ │ ├── bias.hpp │ │ │ │ │ │ ├── fmha_bwd.hpp │ │ │ │ │ │ ├── fmha_fwd.hpp │ │ │ │ │ │ ├── launch_kernel_pt.hpp │ │ │ │ │ │ ├── mask.hpp │ │ │ │ │ │ ├── mha_bwd_ck.hip │ │ │ │ │ │ ├── mha_fwd_ck.hip │ │ │ │ │ │ ├── mha_varlen_bwd_ck.hip │ │ │ │ │ │ ├── mha_varlen_fwd_ck.hip │ │ │ │ │ │ ├── rename_ck_autogen_files.output.txt │ │ │ │ │ │ ├── rename_ck_autogen_files.sh │ │ │ │ │ │ └── rotary.hpp │ │ │ │ │ │ ├── flash_api.h │ │ │ │ │ │ └── flash_common_hip.hpp │ │ │ │ ├── sdp_utils_cpp.cpp │ │ │ │ ├── sdp_utils_cpp.h │ │ │ │ └── transformer.cpp │ │ │ ├── ts_native_functions.yaml │ │ │ ├── ufunc │ │ │ │ └── add.h │ │ │ ├── utils │ │ │ │ ├── Factory.cpp │ │ │ │ ├── Factory.h │ │ │ │ ├── ParamUtils.h │ │ │ │ └── ParamsHash.h │ │ │ ├── verbose_wrapper.cpp │ │ │ ├── verbose_wrapper.h │ │ │ ├── vol2col.h │ │ │ ├── vulkan │ │ │ │ ├── VulkanGuardImpl.cpp │ │ │ │ ├── VulkanOpaqueTensorImpl.h │ │ │ │ ├── api │ │ │ │ │ ├── Adapter.cpp │ │ │ │ │ ├── Adapter.h │ │ │ │ │ ├── Allocator.cpp │ │ │ │ │ ├── Allocator.h │ │ │ │ │ ├── Command.cpp │ │ │ │ │ ├── Command.h │ │ │ │ │ ├── Context.cpp │ │ │ │ │ ├── Context.h │ │ │ │ │ ├── Descriptor.cpp │ │ │ │ │ ├── Descriptor.h │ │ │ │ │ ├── Exception.cpp │ │ │ │ │ ├── Exception.h │ │ │ │ │ ├── Pipeline.cpp │ │ │ │ │ ├── Pipeline.h │ │ │ │ │ ├── QueryPool.cpp │ │ │ │ │ ├── QueryPool.h │ │ │ │ │ ├── Resource.cpp │ │ │ │ │ ├── Resource.h │ │ │ │ │ ├── Runtime.cpp │ │ │ │ │ ├── Runtime.h │ │ │ │ │ ├── Shader.cpp │ │ │ │ │ ├── Shader.h │ │ │ │ │ ├── ShaderRegistry.cpp │ │ │ │ │ ├── ShaderRegistry.h │ │ │ │ │ ├── StringUtil.h │ │ │ │ │ ├── Tensor.cpp │ │ │ │ │ ├── Tensor.h │ │ │ │ │ ├── Types.h │ │ │ │ │ ├── Utils.h │ │ │ │ │ ├── api.h │ │ │ │ │ └── vk_api.h │ │ │ │ ├── glsl │ │ │ │ │ ├── abs.glsl │ │ │ │ │ ├── abs_.glsl │ │ │ │ │ ├── adaptive_avg_pool2d.glsl │ │ │ │ │ ├── avg_pool2d.glsl │ │ │ │ │ ├── batchnorm.glsl │ │ │ │ │ ├── binary_op_scalar.glsl │ │ │ │ │ ├── binary_op_tensor.glsl │ │ │ │ │ ├── buffer_to_buffer.glsl │ │ │ │ │ ├── cat_feature.glsl │ │ │ │ │ ├── clamp.glsl │ │ │ │ │ ├── clamp_.glsl │ │ │ │ │ ├── conv1d.glsl │ │ │ │ │ ├── conv2d.glsl │ │ │ │ │ ├── conv2d_dw.glsl │ │ │ │ │ ├── conv2d_dw_output_tile.glsl │ │ │ │ │ ├── conv2d_pw.glsl │ │ │ │ │ ├── conv_transpose2d.glsl │ │ │ │ │ ├── convert_channels_to_height_packed.glsl │ │ │ │ │ ├── convert_channels_to_width_packed.glsl │ │ │ │ │ ├── cumsum_batch_height_width.glsl │ │ │ │ │ ├── cumsum_channel.glsl │ │ │ │ │ ├── dequantize.glsl │ │ │ │ │ ├── extract_texel.glsl │ │ │ │ │ ├── flip.glsl │ │ │ │ │ ├── gelu_tanh.glsl │ │ │ │ │ ├── gelu_tanh_.glsl │ │ │ │ │ ├── glu_channel.glsl │ │ │ │ │ ├── glu_channel_mul4.glsl │ │ │ │ │ ├── hardshrink.glsl │ │ │ │ │ ├── hardshrink_.glsl │ │ │ │ │ ├── hardsigmoid.glsl │ │ │ │ │ ├── hardsigmoid_.glsl │ │ │ │ │ ├── hardswish.glsl │ │ │ │ │ ├── hardswish_.glsl │ │ │ │ │ ├── image2d_to_nchw.glsl │ │ │ │ │ ├── image_to_nchw.glsl │ │ │ │ │ ├── image_to_nchw_int32.glsl │ │ │ │ │ ├── image_to_nchw_quantized_mul4.glsl │ │ │ │ │ ├── image_to_nchw_uint.glsl │ │ │ │ │ ├── indexing.h │ │ │ │ │ ├── leaky_relu.glsl │ │ │ │ │ ├── leaky_relu_.glsl │ │ │ │ │ ├── lerp.glsl │ │ │ │ │ ├── lerp_.glsl │ │ │ │ │ ├── lerp_scalar.glsl │ │ │ │ │ ├── lerp_scalar_.glsl │ │ │ │ │ ├── log_softmax.glsl │ │ │ │ │ ├── masked_fill.glsl │ │ │ │ │ ├── max_pool2d.glsl │ │ │ │ │ ├── mean_dim.glsl │ │ │ │ │ ├── mean_dim_keepdim.glsl │ │ │ │ │ ├── mm.glsl │ │ │ │ │ ├── nchw_to_image.glsl │ │ │ │ │ ├── nchw_to_image2d.glsl │ │ │ │ │ ├── nchw_to_image2d_int32.glsl │ │ │ │ │ ├── nchw_to_image2d_int8.glsl │ │ │ │ │ ├── nchw_to_image2d_uint8.glsl │ │ │ │ │ ├── nchw_to_image_bool.glsl │ │ │ │ │ ├── nchw_to_image_int32.glsl │ │ │ │ │ ├── nchw_to_image_int8.glsl │ │ │ │ │ ├── nchw_to_image_uint8.glsl │ │ │ │ │ ├── normal_.glsl │ │ │ │ │ ├── permute_4d.glsl │ │ │ │ │ ├── pow_scalar_tensor.glsl │ │ │ │ │ ├── quantize_per_tensor_qint32.glsl │ │ │ │ │ ├── quantize_per_tensor_qint8.glsl │ │ │ │ │ ├── quantize_per_tensor_quint8.glsl │ │ │ │ │ ├── quantized_add.glsl │ │ │ │ │ ├── quantized_addmm_qint8.glsl │ │ │ │ │ ├── quantized_addmm_quint8.glsl │ │ │ │ │ ├── quantized_clamp_qint8.glsl │ │ │ │ │ ├── quantized_clamp_qint8_.glsl │ │ │ │ │ ├── quantized_clamp_quint8.glsl │ │ │ │ │ ├── quantized_clamp_quint8_.glsl │ │ │ │ │ ├── quantized_conv2d.glsl │ │ │ │ │ ├── quantized_conv2d_dw.glsl │ │ │ │ │ ├── quantized_conv2d_pw_2x2.glsl │ │ │ │ │ ├── quantized_conv_transpose2d.glsl │ │ │ │ │ ├── quantized_div.glsl │ │ │ │ │ ├── quantized_gelu_tanh_qint8.glsl │ │ │ │ │ ├── quantized_gelu_tanh_qint8_.glsl │ │ │ │ │ ├── quantized_gelu_tanh_quint8.glsl │ │ │ │ │ ├── quantized_gelu_tanh_quint8_.glsl │ │ │ │ │ ├── quantized_max_pool2d_qint8.glsl │ │ │ │ │ ├── quantized_max_pool2d_quint8.glsl │ │ │ │ │ ├── quantized_mm_qint8.glsl │ │ │ │ │ ├── quantized_mm_quint8.glsl │ │ │ │ │ ├── quantized_mul.glsl │ │ │ │ │ ├── quantized_sub.glsl │ │ │ │ │ ├── quantized_upsample_nearest2d.glsl │ │ │ │ │ ├── random.h │ │ │ │ │ ├── reflection_pad2d.glsl │ │ │ │ │ ├── replication_pad2d.glsl │ │ │ │ │ ├── select_batch_4d.glsl │ │ │ │ │ ├── select_depth_3d.glsl │ │ │ │ │ ├── select_depth_4d.glsl │ │ │ │ │ ├── select_height_3d.glsl │ │ │ │ │ ├── select_height_4d.glsl │ │ │ │ │ ├── select_width_3d.glsl │ │ │ │ │ ├── select_width_4d.glsl │ │ │ │ │ ├── shader_params.yaml │ │ │ │ │ ├── sigmoid.glsl │ │ │ │ │ ├── sigmoid_.glsl │ │ │ │ │ ├── slice_4d.glsl │ │ │ │ │ ├── softmax_batch_height_width.glsl │ │ │ │ │ ├── softmax_channel.glsl │ │ │ │ │ ├── stack_feature.glsl │ │ │ │ │ ├── sum_dim.glsl │ │ │ │ │ ├── sum_dim_keepdim.glsl │ │ │ │ │ ├── tanh.glsl │ │ │ │ │ ├── tanh_.glsl │ │ │ │ │ ├── texel_access.h │ │ │ │ │ ├── threshold.glsl │ │ │ │ │ ├── unary_op.glsl │ │ │ │ │ ├── uniform_.glsl │ │ │ │ │ ├── unsqueeze.glsl │ │ │ │ │ ├── upsample_bilinear2d_align_false.glsl │ │ │ │ │ ├── upsample_bilinear2d_align_true.glsl │ │ │ │ │ ├── upsample_nearest2d.glsl │ │ │ │ │ └── zero.glsl │ │ │ │ ├── impl │ │ │ │ │ ├── Arithmetic.cpp │ │ │ │ │ ├── Arithmetic.h │ │ │ │ │ ├── Common.cpp │ │ │ │ │ ├── Common.h │ │ │ │ │ ├── Packing.cpp │ │ │ │ │ └── Packing.h │ │ │ │ └── ops │ │ │ │ │ ├── Batchnorm.cpp │ │ │ │ │ ├── Batchnorm.h │ │ │ │ │ ├── BinaryOp.cpp │ │ │ │ │ ├── Clamp.cpp │ │ │ │ │ ├── Clone.cpp │ │ │ │ │ ├── Common.h │ │ │ │ │ ├── Concat.cpp │ │ │ │ │ ├── Convert.h │ │ │ │ │ ├── Convolution.cpp │ │ │ │ │ ├── Convolution.h │ │ │ │ │ ├── Copy.cpp │ │ │ │ │ ├── Copy.h │ │ │ │ │ ├── Expand.cpp │ │ │ │ │ ├── Factory.cpp │ │ │ │ │ ├── Factory.h │ │ │ │ │ ├── Flip.cpp │ │ │ │ │ ├── Glu.cpp │ │ │ │ │ ├── Gru.cpp │ │ │ │ │ ├── Gru.h │ │ │ │ │ ├── Layernorm.cpp │ │ │ │ │ ├── Layernorm.h │ │ │ │ │ ├── Lerp.cpp │ │ │ │ │ ├── Lstm.cpp │ │ │ │ │ ├── Lstm.h │ │ │ │ │ ├── MaskedFill.cpp │ │ │ │ │ ├── Mean.cpp │ │ │ │ │ ├── Mm.cpp │ │ │ │ │ ├── Mm.h │ │ │ │ │ ├── NativeLayerNorm.cpp │ │ │ │ │ ├── Padding.cpp │ │ │ │ │ ├── Permute.cpp │ │ │ │ │ ├── Pool.cpp │ │ │ │ │ ├── QuantizedFunctions.h │ │ │ │ │ ├── QuantizedTensor.cpp │ │ │ │ │ ├── Random.cpp │ │ │ │ │ ├── Register.cpp │ │ │ │ │ ├── Register.h │ │ │ │ │ ├── Repeat.cpp │ │ │ │ │ ├── Scalar.cpp │ │ │ │ │ ├── Select.cpp │ │ │ │ │ ├── Shape.cpp │ │ │ │ │ ├── Slice.cpp │ │ │ │ │ ├── Softmax.cpp │ │ │ │ │ ├── Stack.cpp │ │ │ │ │ ├── Sum.cpp │ │ │ │ │ ├── Tile.cpp │ │ │ │ │ ├── Transpose.cpp │ │ │ │ │ ├── UnaryOp.cpp │ │ │ │ │ ├── Unsqueeze.cpp │ │ │ │ │ ├── Upsample.cpp │ │ │ │ │ ├── Utils.cpp │ │ │ │ │ ├── Utils.h │ │ │ │ │ ├── Var.cpp │ │ │ │ │ ├── VulkanPackedContext.h │ │ │ │ │ ├── Zero.cpp │ │ │ │ │ └── cumsum.cpp │ │ │ └── xnnpack │ │ │ │ ├── Activation.cpp │ │ │ │ ├── AveragePooling.cpp │ │ │ │ ├── ChannelShuffle.cpp │ │ │ │ ├── Common.h │ │ │ │ ├── Convolution.cpp │ │ │ │ ├── Convolution.h │ │ │ │ ├── Engine.h │ │ │ │ ├── Init.cpp │ │ │ │ ├── Linear.cpp │ │ │ │ ├── Linear.h │ │ │ │ ├── MaxPooling.cpp │ │ │ │ ├── OpContext.cpp │ │ │ │ ├── OpContext.h │ │ │ │ ├── Pooling.h │ │ │ │ ├── RegisterOpContextClass.cpp │ │ │ │ └── Shim.cpp │ │ ├── nnapi │ │ │ ├── CMakeLists.txt │ │ │ ├── NeuralNetworks.h │ │ │ ├── codegen.py │ │ │ ├── nnapi_bind.cpp │ │ │ ├── nnapi_bind.h │ │ │ ├── nnapi_model_loader.cpp │ │ │ ├── nnapi_model_loader.h │ │ │ ├── nnapi_register.cpp │ │ │ ├── nnapi_wrapper.cpp │ │ │ └── nnapi_wrapper.h │ │ ├── ops │ │ │ ├── from_blob.h │ │ │ └── tensor.h │ │ ├── quantized │ │ │ ├── CMakeLists.txt │ │ │ ├── QTensorImpl.cpp │ │ │ ├── QTensorImpl.h │ │ │ ├── Quantizer.cpp │ │ │ └── Quantizer.h │ │ ├── record_function.cpp │ │ ├── record_function.h │ │ ├── templates │ │ │ ├── ATenOpList.cpp │ │ │ ├── CompositeViewCopyKernels.cpp │ │ │ ├── DispatchKeyFunction.h │ │ │ ├── DispatchKeyFunctions.h │ │ │ ├── DispatchKeyFunctions_inl.h │ │ │ ├── DispatchKeyNativeFunctions.cpp │ │ │ ├── DispatchKeyNativeFunctions.h │ │ │ ├── Function.h │ │ │ ├── FunctionalInverses.h │ │ │ ├── Functions.cpp │ │ │ ├── Functions.h │ │ │ ├── LazyIr.h │ │ │ ├── LazyNonNativeIr.h │ │ │ ├── MethodOperators.h │ │ │ ├── NativeFunction.h │ │ │ ├── NativeFunctions.h │ │ │ ├── NativeMetaFunction.h │ │ │ ├── NativeMetaFunctions.h │ │ │ ├── Operator.h │ │ │ ├── Operators.cpp │ │ │ ├── Operators.h │ │ │ ├── RedispatchFunctions.cpp │ │ │ ├── RedispatchFunctions.h │ │ │ ├── RegisterBackendSelect.cpp │ │ │ ├── RegisterCodegenUnboxedKernels.cpp │ │ │ ├── RegisterDispatchDefinitions.ini │ │ │ ├── RegisterDispatchKey.cpp │ │ │ ├── RegisterFunctionalization.cpp │ │ │ ├── RegisterSchema.cpp │ │ │ ├── RegistrationDeclarations.h │ │ │ ├── TensorBody.h │ │ │ ├── TensorMethods.cpp │ │ │ ├── UfuncCPU.cpp │ │ │ ├── UfuncCPUKernel.cpp │ │ │ ├── UfuncCUDA.cu │ │ │ ├── UnboxingFunctions.cpp │ │ │ ├── UnboxingFunctions.h │ │ │ ├── aten_interned_strings.h │ │ │ └── enum_tag.h │ │ ├── test │ │ │ ├── CMakeLists.txt │ │ │ ├── Dict_test.cpp │ │ │ ├── Dimname_test.cpp │ │ │ ├── ExclusivelyOwned_test.cpp │ │ │ ├── MaybeOwned_test.cpp │ │ │ ├── NamedTensor_test.cpp │ │ │ ├── StorageUtils_test.cpp │ │ │ ├── allocator_clone_test.h │ │ │ ├── apply_utils_test.cpp │ │ │ ├── atest.cpp │ │ │ ├── basic.cpp │ │ │ ├── broadcast_test.cpp │ │ │ ├── cpu_allocator_test.cpp │ │ │ ├── cpu_caching_allocator_test.cpp │ │ │ ├── cpu_generator_test.cpp │ │ │ ├── cpu_profiling_allocator_test.cpp │ │ │ ├── cpu_rng_test.cpp │ │ │ ├── cuda_allocatorTraceTracker_test.cpp │ │ │ ├── cuda_allocator_test.cpp │ │ │ ├── cuda_apply_test.cpp │ │ │ ├── cuda_atomic_ops_test.cu │ │ │ ├── cuda_caching_host_allocator_test.cpp │ │ │ ├── cuda_complex_math_test.cu │ │ │ ├── cuda_complex_test.cu │ │ │ ├── cuda_cub_test.cu │ │ │ ├── cuda_cudnn_test.cpp │ │ │ ├── cuda_device_test.cpp │ │ │ ├── cuda_distributions_test.cu │ │ │ ├── cuda_dlconvertor_test.cpp │ │ │ ├── cuda_generator_test.cu │ │ │ ├── cuda_half_test.cu │ │ │ ├── cuda_integer_divider_test.cu │ │ │ ├── cuda_optional_test.cu │ │ │ ├── cuda_packedtensoraccessor_test.cu │ │ │ ├── cuda_reportMemoryUsage_test.cpp │ │ │ ├── cuda_stream_test.cpp │ │ │ ├── cuda_tensor_interop_test.cpp │ │ │ ├── cuda_vectorized_test.cu │ │ │ ├── dispatch_key_set_test.cpp │ │ │ ├── dlconvertor_test.cpp │ │ │ ├── extension_backend_test.cpp │ │ │ ├── half_test.cpp │ │ │ ├── ivalue_test.cpp │ │ │ ├── lazy_tensor_test.cpp │ │ │ ├── legacy_vmap_test.cpp │ │ │ ├── math_kernel_test.cpp │ │ │ ├── memory_format_test.cpp │ │ │ ├── memory_overlapping_test.cpp │ │ │ ├── mobile_memory_cleanup.cpp │ │ │ ├── mps_test_allocator.cpp │ │ │ ├── mps_test_metal_library.cpp │ │ │ ├── mps_test_objc_interface.mm │ │ │ ├── mps_test_print.cpp │ │ │ ├── native_test.cpp │ │ │ ├── operator_name_test.cpp │ │ │ ├── operators_test.cpp │ │ │ ├── packedtensoraccessor_test.cpp │ │ │ ├── pow_test.cpp │ │ │ ├── quantized_test.cpp │ │ │ ├── reduce_ops_test.cpp │ │ │ ├── reportMemoryUsage.h │ │ │ ├── reportMemoryUsage_test.cpp │ │ │ ├── rng_test.h │ │ │ ├── scalar_tensor_test.cpp │ │ │ ├── scalar_test.cpp │ │ │ ├── stride_properties_test.cpp │ │ │ ├── tensor_interop_test.cpp │ │ │ ├── tensor_iterator_test.cpp │ │ │ ├── test_assert.h │ │ │ ├── test_install │ │ │ │ ├── CMakeLists.txt │ │ │ │ └── main.cpp │ │ │ ├── test_parallel.cpp │ │ │ ├── test_thread_pool_guard.cpp │ │ │ ├── thread_init_test.cpp │ │ │ ├── type_ptr_test.cpp │ │ │ ├── type_test.cpp │ │ │ ├── undefined_tensor_test.cpp │ │ │ ├── vec_test_all_types.cpp │ │ │ ├── vec_test_all_types.h │ │ │ ├── verify_api_visibility.cpp │ │ │ ├── vitals.cpp │ │ │ ├── vulkan_api_test.cpp │ │ │ ├── vulkan_quantized_api_test.cpp │ │ │ ├── weakref_test.cpp │ │ │ ├── wrapdim_test.cpp │ │ │ ├── xla_tensor_test.cpp │ │ │ ├── xnnpack_test.cpp │ │ │ ├── xpu_caching_host_allocator_test.cpp │ │ │ ├── xpu_device_test.cpp │ │ │ ├── xpu_event_test.cpp │ │ │ └── xpu_generator_test.cpp │ │ ├── vulkan │ │ │ ├── Context.cpp │ │ │ └── Context.h │ │ └── xpu │ │ │ ├── CachingHostAllocator.cpp │ │ │ ├── CachingHostAllocator.h │ │ │ ├── PinnedMemoryAllocator.h │ │ │ ├── XPUContext.cpp │ │ │ ├── XPUContext.h │ │ │ ├── XPUDevice.h │ │ │ ├── XPUEvent.h │ │ │ ├── XPUGeneratorImpl.cpp │ │ │ ├── XPUGeneratorImpl.h │ │ │ └── detail │ │ │ ├── XPUHooks.cpp │ │ │ └── XPUHooks.h │ ├── README.md │ └── THC │ │ ├── CMakeLists.txt │ │ ├── THCAtomics.cuh │ │ └── THCDeviceUtils.cuh └── tools │ ├── run_tests.sh │ ├── test_install.sh │ └── valgrind.sup ├── benchmarks ├── README.md ├── compare-fastrnn-results.py ├── compare.sh ├── distributed │ ├── ddp │ │ ├── README.md │ │ ├── benchmark.py │ │ └── diff.py │ └── rpc │ │ ├── parameter_server │ │ ├── README.md │ │ ├── configurations │ │ │ ├── data_configurations.json │ │ │ └── model_configurations.json │ │ ├── data │ │ │ ├── DummyData.py │ │ │ └── __init__.py │ │ ├── launcher.py │ │ ├── metrics │ │ │ ├── CPUMetric.py │ │ │ ├── CUDAMetric.py │ │ │ ├── MetricBase.py │ │ │ ├── MetricsLogger.py │ │ │ └── ProcessedMetricsPrinter.py │ │ ├── models │ │ │ ├── DummyModel.py │ │ │ └── __init__.py │ │ ├── server │ │ │ ├── __init__.py │ │ │ └── server.py │ │ ├── trainer │ │ │ ├── __init__.py │ │ │ ├── criterions.py │ │ │ ├── ddp_models.py │ │ │ ├── hook_states.py │ │ │ ├── hooks.py │ │ │ ├── iteration_steps.py │ │ │ ├── preprocess_data.py │ │ │ └── trainer.py │ │ └── utils.py │ │ └── rl │ │ ├── README.md │ │ ├── agent.py │ │ ├── coordinator.py │ │ ├── launcher.py │ │ └── observer.py ├── dynamo │ ├── Makefile │ ├── README.md │ ├── __init__.py │ ├── all_torchbench_models_list.txt │ ├── benchmarks.py │ ├── cachebench.py │ ├── check_accuracy.py │ ├── check_csv.py │ ├── check_graph_breaks.py │ ├── check_memory_compression_ratio.py │ ├── check_perf_csv.py │ ├── ci_expected_accuracy │ │ ├── aot_eager_huggingface_inference.csv │ │ ├── aot_eager_huggingface_training.csv │ │ ├── aot_eager_timm_inference.csv │ │ ├── aot_eager_timm_training.csv │ │ ├── aot_eager_torchbench_inference.csv │ │ ├── aot_eager_torchbench_training.csv │ │ ├── aot_inductor_huggingface_inference.csv │ │ ├── aot_inductor_timm_inference.csv │ │ ├── aot_inductor_torchbench_inference.csv │ │ ├── cpu_aot_inductor_amp_freezing_torchbench_inference.csv │ │ ├── cpu_aot_inductor_freezing_huggingface_inference.csv │ │ ├── cpu_aot_inductor_freezing_timm_inference.csv │ │ ├── cpu_aot_inductor_freezing_torchbench_inference.csv │ │ ├── cpu_inductor_amp_freezing_huggingface_inference.csv │ │ ├── cpu_inductor_amp_freezing_timm_inference.csv │ │ ├── cpu_inductor_amp_freezing_torchbench_inference.csv │ │ ├── cpu_inductor_freezing_huggingface_inference.csv │ │ ├── cpu_inductor_freezing_timm_inference.csv │ │ ├── cpu_inductor_freezing_torchbench_inference.csv │ │ ├── cpu_inductor_huggingface_inference.csv │ │ ├── cpu_inductor_timm_inference.csv │ │ ├── cpu_inductor_torchbench_inference.csv │ │ ├── cu124 │ │ │ ├── aot_inductor_torchbench_inference.csv │ │ │ ├── dynamic_inductor_huggingface_training.csv │ │ │ └── inductor_huggingface_training.csv │ │ ├── dynamic_aot_eager_huggingface_inference.csv │ │ ├── dynamic_aot_eager_huggingface_training.csv │ │ ├── dynamic_aot_eager_timm_inference.csv │ │ ├── dynamic_aot_eager_timm_training.csv │ │ ├── dynamic_aot_eager_torchbench_inference.csv │ │ ├── dynamic_aot_eager_torchbench_training.csv │ │ ├── dynamic_cpu_aot_inductor_amp_freezing_torchbench_inference.csv │ │ ├── dynamic_cpu_aot_inductor_freezing_torchbench_inference.csv │ │ ├── dynamic_cpu_inductor_huggingface_inference.csv │ │ ├── dynamic_cpu_inductor_timm_inference.csv │ │ ├── dynamic_cpu_inductor_torchbench_inference.csv │ │ ├── dynamic_inductor_huggingface_inference.csv │ │ ├── dynamic_inductor_huggingface_training.csv │ │ ├── dynamic_inductor_timm_inference.csv │ │ ├── dynamic_inductor_timm_training.csv │ │ ├── dynamic_inductor_torchbench_inference.csv │ │ ├── dynamic_inductor_torchbench_training.csv │ │ ├── dynamo_eager_huggingface_inference.csv │ │ ├── dynamo_eager_huggingface_training.csv │ │ ├── dynamo_eager_timm_inference.csv │ │ ├── dynamo_eager_timm_training.csv │ │ ├── dynamo_eager_torchbench_inference.csv │ │ ├── dynamo_eager_torchbench_training.csv │ │ ├── inductor_huggingface_inference.csv │ │ ├── inductor_huggingface_training.csv │ │ ├── inductor_timm_inference.csv │ │ ├── inductor_timm_training.csv │ │ ├── inductor_torchbench_inference.csv │ │ ├── inductor_torchbench_training.csv │ │ ├── rocm │ │ │ ├── aot_eager_huggingface_inference.csv │ │ │ ├── aot_eager_huggingface_training.csv │ │ │ ├── aot_eager_timm_inference.csv │ │ │ ├── aot_eager_timm_training.csv │ │ │ ├── aot_eager_torchbench_inference.csv │ │ │ ├── aot_eager_torchbench_training.csv │ │ │ ├── aot_inductor_huggingface_inference.csv │ │ │ ├── aot_inductor_timm_inference.csv │ │ │ ├── aot_inductor_torchbench_inference.csv │ │ │ ├── dynamic_aot_eager_huggingface_inference.csv │ │ │ ├── dynamic_aot_eager_huggingface_training.csv │ │ │ ├── dynamic_aot_eager_timm_inference.csv │ │ │ ├── dynamic_aot_eager_timm_training.csv │ │ │ ├── dynamic_aot_eager_torchbench_inference.csv │ │ │ ├── dynamic_aot_eager_torchbench_training.csv │ │ │ ├── dynamic_inductor_huggingface_inference.csv │ │ │ ├── dynamic_inductor_huggingface_training.csv │ │ │ ├── dynamic_inductor_timm_inference.csv │ │ │ ├── dynamic_inductor_timm_training.csv │ │ │ ├── dynamic_inductor_torchbench_inference.csv │ │ │ ├── dynamic_inductor_torchbench_training.csv │ │ │ ├── dynamo_eager_huggingface_inference.csv │ │ │ ├── dynamo_eager_huggingface_training.csv │ │ │ ├── dynamo_eager_timm_inference.csv │ │ │ ├── dynamo_eager_timm_training.csv │ │ │ ├── dynamo_eager_torchbench_inference.csv │ │ │ ├── dynamo_eager_torchbench_training.csv │ │ │ ├── inductor_huggingface_inference.csv │ │ │ ├── inductor_huggingface_training.csv │ │ │ ├── inductor_timm_inference.csv │ │ │ ├── inductor_timm_training.csv │ │ │ ├── inductor_torchbench_inference.csv │ │ │ └── inductor_torchbench_training.csv │ │ └── update_expected.py │ ├── combine_csv.py │ ├── common.py │ ├── dist_util.py │ ├── distributed.py │ ├── expected_ci_perf_inductor_torchbench.csv │ ├── expected_ci_speedup_inductor_torchbench_cpu.csv │ ├── huggingface.py │ ├── huggingface.yaml │ ├── huggingface_models_list.txt │ ├── huggingface_models_list_cpu.txt │ ├── join_results.py │ ├── microbenchmarks │ │ ├── __init__.py │ │ ├── analyze_templates.py │ │ ├── bench_mm_fusion.py │ │ ├── benchmark_helper.py │ │ ├── cache_debug_microbenchmarks.py │ │ ├── cache_hit_microbenchmarks.py │ │ ├── dynamo_guard_eval.py │ │ ├── dynamo_microbenchmarks.py │ │ ├── fx_microbenchmarks.py │ │ ├── inductor_bmm.py │ │ ├── inductor_cpu_atomic.py │ │ ├── inductor_mm.py │ │ ├── matmul_relu.py │ │ ├── microbench.py │ │ ├── model.py │ │ ├── operator_inp_logs │ │ │ ├── hf_train │ │ │ │ ├── AlbertForMaskedLM_training.txt │ │ │ │ ├── AlbertForQuestionAnswering_training.txt │ │ │ │ ├── AllenaiLongformerBase_training.txt │ │ │ │ ├── BartForCausalLM_training.txt │ │ │ │ ├── BartForConditionalGeneration_training.txt │ │ │ │ ├── BertForMaskedLM_training.txt │ │ │ │ ├── BertForQuestionAnswering_training.txt │ │ │ │ ├── BigBird_training.txt │ │ │ │ ├── BlenderbotSmallForCausalLM_training.txt │ │ │ │ ├── BlenderbotSmallForConditionalGeneration_training.txt │ │ │ │ ├── CamemBert_training.txt │ │ │ │ ├── DebertaForMaskedLM_training.txt │ │ │ │ ├── DebertaForQuestionAnswering_training.txt │ │ │ │ ├── DebertaV2ForMaskedLM_training.txt │ │ │ │ ├── DebertaV2ForQuestionAnswering_training.txt │ │ │ │ ├── DistilBertForMaskedLM_training.txt │ │ │ │ ├── DistilBertForQuestionAnswering_training.txt │ │ │ │ ├── DistillGPT2_training.txt │ │ │ │ ├── ElectraForCausalLM_training.txt │ │ │ │ ├── ElectraForQuestionAnswering_training.txt │ │ │ │ ├── GPT2ForSequenceClassification_training.txt │ │ │ │ ├── GPTNeoForCausalLM_training.txt │ │ │ │ ├── GPTNeoForSequenceClassification_training.txt │ │ │ │ ├── GoogleFnet_training.txt │ │ │ │ ├── LayoutLMForMaskedLM_training.txt │ │ │ │ ├── LayoutLMForSequenceClassification_training.txt │ │ │ │ ├── M2M100ForConditionalGeneration_training.txt │ │ │ │ ├── MBartForCausalLM_training.txt │ │ │ │ ├── MBartForConditionalGeneration_training.txt │ │ │ │ ├── MegatronBertForCausalLM_training.txt │ │ │ │ ├── MegatronBertForQuestionAnswering_training.txt │ │ │ │ ├── MobileBertForMaskedLM_training.txt │ │ │ │ ├── MobileBertForQuestionAnswering_training.txt │ │ │ │ ├── OPTForCausalLM_training.txt │ │ │ │ ├── PLBartForCausalLM_training.txt │ │ │ │ ├── PLBartForConditionalGeneration_training.txt │ │ │ │ ├── PegasusForCausalLM_training.txt │ │ │ │ ├── PegasusForConditionalGeneration_training.txt │ │ │ │ ├── RobertaForCausalLM_training.txt │ │ │ │ ├── RobertaForQuestionAnswering_training.txt │ │ │ │ ├── Speech2Text2ForCausalLM_training.txt │ │ │ │ ├── TrOCRForCausalLM_training.txt │ │ │ │ ├── XGLMForCausalLM_training.txt │ │ │ │ ├── XLNetLMHeadModel_training.txt │ │ │ │ └── YituTechConvBert_training.txt │ │ │ ├── timm_train │ │ │ │ ├── adv_inception_v3_training.txt │ │ │ │ ├── beit_base_patch16_224_training.txt │ │ │ │ ├── botnet26t_256_training.txt │ │ │ │ ├── cait_m36_384_training.txt │ │ │ │ ├── coat_lite_mini_training.txt │ │ │ │ ├── convmixer_768_32_training.txt │ │ │ │ ├── convnext_base_training.txt │ │ │ │ ├── crossvit_9_240_training.txt │ │ │ │ ├── cspdarknet53_training.txt │ │ │ │ ├── deit_base_distilled_patch16_224_training.txt │ │ │ │ ├── densenet121_training.txt │ │ │ │ ├── dla102_training.txt │ │ │ │ ├── dm_nfnet_f0_training.txt │ │ │ │ ├── dpn107_training.txt │ │ │ │ ├── eca_botnext26ts_256_training.txt │ │ │ │ ├── eca_halonext26ts_training.txt │ │ │ │ ├── ecaresnet101d_training.txt │ │ │ │ ├── ese_vovnet19b_dw_training.txt │ │ │ │ ├── fbnetc_100_training.txt │ │ │ │ ├── fbnetv3_b_training.txt │ │ │ │ ├── gernet_l_training.txt │ │ │ │ ├── ghostnet_100_training.txt │ │ │ │ ├── gluon_inception_v3_training.txt │ │ │ │ ├── gluon_senet154_training.txt │ │ │ │ ├── gluon_xception65_training.txt │ │ │ │ ├── gmixer_24_224_training.txt │ │ │ │ ├── gmlp_s16_224_training.txt │ │ │ │ ├── hardcorenas_a_training.txt │ │ │ │ ├── hrnet_w18_training.txt │ │ │ │ ├── inception_v3_training.txt │ │ │ │ ├── jx_nest_base_training.txt │ │ │ │ ├── lcnet_050_training.txt │ │ │ │ ├── legacy_senet154_training.txt │ │ │ │ ├── levit_128_training.txt │ │ │ │ ├── mixer_b16_224_training.txt │ │ │ │ ├── mixnet_l_training.txt │ │ │ │ ├── mnasnet_100_training.txt │ │ │ │ ├── mobilenetv2_100_training.txt │ │ │ │ ├── mobilenetv3_large_100_training.txt │ │ │ │ ├── mobilevit_s_training.txt │ │ │ │ ├── nasnetalarge_training.txt │ │ │ │ ├── nfnet_l0_training.txt │ │ │ │ ├── pit_b_224_training.txt │ │ │ │ ├── pnasnet5large_training.txt │ │ │ │ ├── poolformer_m36_training.txt │ │ │ │ ├── regnety_002_training.txt │ │ │ │ ├── repvgg_a2_training.txt │ │ │ │ ├── res2net101_26w_4s_training.txt │ │ │ │ ├── res2net50_14w_8s_training.txt │ │ │ │ ├── res2next50_training.txt │ │ │ │ ├── resmlp_12_224_training.txt │ │ │ │ ├── resnest101e_training.txt │ │ │ │ ├── resnet18_training.txt │ │ │ │ ├── rexnet_100_training.txt │ │ │ │ ├── sebotnet33ts_256_training.txt │ │ │ │ ├── selecsls42b_training.txt │ │ │ │ ├── spnasnet_100_training.txt │ │ │ │ ├── swin_base_patch4_window7_224_training.txt │ │ │ │ ├── swsl_resnext101_32x16d_training.txt │ │ │ │ ├── tf_efficientnet_b0_training.txt │ │ │ │ ├── tf_mixnet_l_training.txt │ │ │ │ ├── tinynet_a_training.txt │ │ │ │ ├── tnt_s_patch16_224_training.txt │ │ │ │ ├── twins_pcpvt_base_training.txt │ │ │ │ ├── visformer_small_training.txt │ │ │ │ ├── vit_base_patch16_224_training.txt │ │ │ │ └── volo_d1_224_training.txt │ │ │ └── torchbench_train │ │ │ │ ├── BERT_pytorch_training.txt │ │ │ │ ├── Background_Matting_training.txt │ │ │ │ ├── LearningToPaint_training.txt │ │ │ │ ├── Super_SloMo_training.txt │ │ │ │ ├── alexnet_training.txt │ │ │ │ ├── attention_is_all_you_need_pytorch_training.txt │ │ │ │ ├── dcgan_training.txt │ │ │ │ ├── densenet121_training.txt │ │ │ │ ├── fambench_dlrm_training.txt │ │ │ │ ├── fastNLP_Bert_training.txt │ │ │ │ ├── hf_Albert_training.txt │ │ │ │ ├── hf_Bart_training.txt │ │ │ │ ├── hf_Bert_training.txt │ │ │ │ ├── hf_BigBird_training.txt │ │ │ │ ├── hf_DistilBert_training.txt │ │ │ │ ├── hf_GPT2_training.txt │ │ │ │ ├── hf_Longformer_training.txt │ │ │ │ ├── maml_omniglot_training.txt │ │ │ │ ├── mnasnet1_0_training.txt │ │ │ │ ├── mobilenet_v2_training.txt │ │ │ │ ├── mobilenet_v3_large_training.txt │ │ │ │ ├── nvidia_deeprecommender_training.txt │ │ │ │ ├── pytorch_CycleGAN_and_pix2pix_training.txt │ │ │ │ ├── pytorch_stargan_training.txt │ │ │ │ ├── pytorch_struct_training.txt │ │ │ │ ├── pytorch_unet_training.txt │ │ │ │ ├── resnet18_training.txt │ │ │ │ ├── resnet50_training.txt │ │ │ │ ├── resnext50_32x4d_training.txt │ │ │ │ ├── shufflenet_v2_x1_0_training.txt │ │ │ │ ├── speech_transformer_training.txt │ │ │ │ ├── squeezenet1_1_training.txt │ │ │ │ ├── timm_efficientdet_training.txt │ │ │ │ ├── timm_efficientnet_training.txt │ │ │ │ ├── timm_nfnet_training.txt │ │ │ │ ├── timm_regnet_training.txt │ │ │ │ ├── timm_resnest_training.txt │ │ │ │ ├── timm_vision_transformer_training.txt │ │ │ │ ├── timm_vovnet_training.txt │ │ │ │ ├── tts_angular_training.txt │ │ │ │ ├── vgg16_training.txt │ │ │ │ ├── vision_maskrcnn_training.txt │ │ │ │ └── yolov3_training.txt │ │ ├── operator_inp_utils.py │ │ ├── operatorbench.py │ │ ├── overheads.py │ │ ├── tensor_layout_mini_benchmark.py │ │ └── utils.py │ ├── parse_logs.py │ ├── pr_time_benchmarks │ │ ├── README.md │ │ ├── __init__.py │ │ ├── benchmark_runner.sh │ │ ├── benchmarks │ │ │ ├── add_loop.py │ │ │ ├── aotdispatcher.py │ │ │ ├── aotdispatcher_partitioner.py │ │ │ ├── aotdispatcher_partitioner2.py │ │ │ ├── basic_modules_benchmarks.py │ │ │ ├── benchmark_base.py │ │ │ ├── float_args.py │ │ │ ├── sum_floordiv.py │ │ │ ├── symint_sum.py │ │ │ └── update_hint_benchmark.py │ │ ├── check_results.py │ │ ├── expected_results.csv │ │ ├── log_benchmarking_time.py │ │ └── test_check_result │ │ │ ├── expected_test.csv │ │ │ └── result_test.csv │ ├── run_all.sh │ ├── run_delta.sh │ ├── runner.py │ ├── summarize_perf.py │ ├── test.py │ ├── timm_models.py │ ├── timm_models.yaml │ ├── timm_models_list.txt │ ├── timm_models_list_cpu.txt │ ├── torchao_backend.py │ ├── torchbench.py │ ├── torchbench.yaml │ ├── torchbench_models_list.txt │ ├── torchbench_models_list_cpu.txt │ └── training_loss.py ├── fastrnns │ ├── README.md │ ├── __init__.py │ ├── bench.py │ ├── cells.py │ ├── conftest.py │ ├── custom_lstms.py │ ├── factory.py │ ├── fuser.py │ ├── profile.py │ ├── runner.py │ ├── scratch.py │ ├── test.py │ └── test_bench.py ├── framework_overhead_benchmark │ ├── SimpleAddModule.py │ ├── framework_overhead_benchmark.py │ ├── pt_wrapper_module.py │ └── utils.py ├── functional_autograd_benchmark │ ├── README.md │ ├── audio_text_models.py │ ├── compare.py │ ├── functional_autograd_benchmark.py │ ├── ppl_models.py │ ├── torchaudio_models.py │ ├── torchvision_models.py │ ├── utils.py │ └── vision_models.py ├── fuser │ ├── plot_speedups.py │ └── run_benchmarks.py ├── gpt_fast │ ├── benchmark.py │ ├── common.py │ ├── generate.py │ ├── mixtral_moe_model.py │ ├── mixtral_moe_quantize.py │ ├── model.py │ └── quantize.py ├── inference │ ├── CHANGELOG.md │ ├── README.md │ ├── process_metrics.py │ ├── results │ │ ├── output_128_false.md │ │ ├── output_128_true.md │ │ ├── output_1_false.md │ │ ├── output_1_true.md │ │ ├── output_256_false.md │ │ ├── output_256_true.md │ │ ├── output_32_false.md │ │ ├── output_32_true.md │ │ ├── output_64_false.md │ │ └── output_64_true.md │ ├── runner.sh │ ├── server.py │ └── src │ │ ├── avg_latency_plot.png │ │ └── throughput_plot.png ├── instruction_counts │ ├── README.md │ ├── applications │ │ ├── __init__.py │ │ └── ci.py │ ├── core │ │ ├── __init__.py │ │ ├── api.py │ │ ├── expand.py │ │ ├── types.py │ │ └── utils.py │ ├── definitions │ │ ├── __init__.py │ │ ├── setup.py │ │ └── standard.py │ ├── execution │ │ ├── __init__.py │ │ ├── runner.py │ │ └── work.py │ ├── main.py │ └── worker │ │ ├── __init__.py │ │ └── main.py ├── nested │ └── nested_bmm_bench.py ├── operator_benchmark │ ├── README.md │ ├── __init__.py │ ├── benchmark_all_other_test.py │ ├── benchmark_all_quantized_test.py │ ├── benchmark_all_test.py │ ├── benchmark_core.py │ ├── benchmark_pytorch.py │ ├── benchmark_runner.py │ ├── benchmark_test_generator.py │ ├── benchmark_utils.py │ ├── common │ │ ├── __init__.py │ │ ├── repeat_benchmark.py │ │ └── tests │ │ │ ├── add_ops_list_test.py │ │ │ ├── jit_forward_test.py │ │ │ ├── pt_backward_test.py │ │ │ ├── pt_configs_list_test.py │ │ │ ├── pt_cpu_gpu_forward_backward_test.py │ │ │ └── random_sample_test.py │ ├── operator_benchmark.py │ ├── pt │ │ ├── __init__.py │ │ ├── add_test.py │ │ ├── ao_sparsifier_test.py │ │ ├── arange_test.py │ │ ├── as_strided_test.py │ │ ├── batchnorm_test.py │ │ ├── binary_inplace_test.py │ │ ├── binary_test.py │ │ ├── bmm_test.py │ │ ├── cat_test.py │ │ ├── channel_shuffle_test.py │ │ ├── chunk_test.py │ │ ├── clip_ranges_test.py │ │ ├── configs.py │ │ ├── conv_test.py │ │ ├── diag_test.py │ │ ├── embeddingbag_test.py │ │ ├── fill_test.py │ │ ├── gather_test.py │ │ ├── gelu_test.py │ │ ├── groupnorm_test.py │ │ ├── hardsigmoid_test.py │ │ ├── hardswish_test.py │ │ ├── index_add__test.py │ │ ├── index_select_test.py │ │ ├── instancenorm_test.py │ │ ├── interpolate_test.py │ │ ├── layernorm_test.py │ │ ├── linear_prepack_fp16_test.py │ │ ├── linear_test.py │ │ ├── linear_unpack_fp16_test.py │ │ ├── matmul_test.py │ │ ├── matrix_mult_test.py │ │ ├── mm_test.py │ │ ├── nan_to_num_test.py │ │ ├── pool_test.py │ │ ├── qactivation_test.py │ │ ├── qarithmetic_test.py │ │ ├── qatembedding_ops_test.py │ │ ├── qbatchnorm_test.py │ │ ├── qcat_test.py │ │ ├── qcomparators_test.py │ │ ├── qconv_test.py │ │ ├── qembedding_bag_lookups_test.py │ │ ├── qembedding_pack_test.py │ │ ├── qembeddingbag_test.py │ │ ├── qgroupnorm_test.py │ │ ├── qinstancenorm_test.py │ │ ├── qinterpolate_test.py │ │ ├── qlayernorm_test.py │ │ ├── qlinear_test.py │ │ ├── qobserver_test.py │ │ ├── qpool_test.py │ │ ├── qrnn_test.py │ │ ├── qtensor_method_test.py │ │ ├── quantization_test.py │ │ ├── qunary_test.py │ │ ├── remainder_test.py │ │ ├── softmax_test.py │ │ ├── split_test.py │ │ ├── stack_test.py │ │ ├── sum_test.py │ │ ├── tensor_to_test.py │ │ ├── ternary_test.py │ │ ├── topk_test.py │ │ ├── unary_test.py │ │ └── where_test.py │ └── pt_extension │ │ ├── cpp_extension_test.py │ │ ├── extension.cpp │ │ └── setup.py ├── overrides_benchmark │ ├── README.md │ ├── bench.py │ ├── common.py │ └── pyspybench.py ├── profiler_benchmark │ ├── profiler_bench.py │ └── resnet_memory_profiler.py ├── record_function_benchmark │ └── record_function_bench.py ├── serialization │ ├── nested_annotation_str.py │ └── simple_measurement.py ├── sparse │ ├── README.md │ ├── __init__.py │ ├── dlmc │ │ ├── README.md │ │ ├── __init__.py │ │ ├── matmul_bench.py │ │ ├── test.sh │ │ └── utils.py │ ├── spmm.py │ ├── spmv.py │ ├── test_csr.sh │ ├── triton_ops.py │ └── utils.py ├── static_runtime │ ├── CMakeLists.txt │ ├── deep_wide_pt.cc │ ├── deep_wide_pt.h │ ├── deep_wide_pt_bench.cc │ ├── test_cpu_fusion.cc │ ├── test_generated_ops.cc │ ├── test_static_module.cc │ ├── test_static_runtime.cc │ ├── test_utils.cc │ └── test_utils.h ├── tensorexpr │ ├── HowToRun.md │ ├── __main__.py │ ├── attention.py │ ├── benchmark.py │ ├── broadcast.py │ ├── concat.py │ ├── conv.py │ ├── elementwise.py │ ├── matmul.py │ ├── microbenchmarks.py │ ├── nnc.png │ ├── normalization.py │ ├── pooling.py │ ├── pt_engine.py │ ├── reduction.py │ ├── rnn_eltwise.py │ ├── softmax.py │ ├── swish.py │ └── tensor_engine.py ├── transformer │ ├── attention_bias_benchmarks.py │ ├── better_transformer_vs_mha_functional.py │ ├── score_mod.py │ ├── sdp.py │ └── sdpa.py └── upload_scribe.py ├── binaries ├── CMakeLists.txt ├── aot_model_compiler.cc ├── at_launch_benchmark.cc ├── compare_models_torch.cc ├── core_overhead_benchmark.cc ├── dump_operator_names.cc ├── lite_interpreter_model_load.cc ├── load_benchmark_torch.cc ├── optimize_for_mobile.cc ├── parallel_info.cc ├── record_function_benchmark.cc └── speed_benchmark_torch.cc ├── buckbuild.bzl ├── c10 ├── BUCK.oss ├── BUILD.bazel ├── CMakeLists.txt ├── benchmark │ ├── BUILD.bazel │ ├── CMakeLists.txt │ ├── build.bzl │ └── intrusive_ptr_benchmark.cpp ├── build.bzl ├── core │ ├── Allocator.cpp │ ├── Allocator.h │ ├── AutogradState.cpp │ ├── AutogradState.h │ ├── BUILD.bazel │ ├── Backend.h │ ├── CPUAllocator.cpp │ ├── CPUAllocator.h │ ├── CachingDeviceAllocator.h │ ├── CompileTimeFunctionPointer.h │ ├── ConstantSymNodeImpl.cpp │ ├── ConstantSymNodeImpl.h │ ├── Contiguity.h │ ├── CopyBytes.cpp │ ├── CopyBytes.h │ ├── DefaultDtype.cpp │ ├── DefaultDtype.h │ ├── DefaultTensorOptions.h │ ├── Device.cpp │ ├── Device.h │ ├── DeviceArray.h │ ├── DeviceGuard.h │ ├── DeviceType.cpp │ ├── DeviceType.h │ ├── DispatchKey.cpp │ ├── DispatchKey.h │ ├── DispatchKeySet.cpp │ ├── DispatchKeySet.h │ ├── DynamicCast.h │ ├── Event.h │ ├── GeneratorImpl.cpp │ ├── GeneratorImpl.h │ ├── GradMode.cpp │ ├── GradMode.h │ ├── InferenceMode.cpp │ ├── InferenceMode.h │ ├── Layout.h │ ├── MemoryFormat.h │ ├── OptionalRef.h │ ├── PyHandleCache.h │ ├── QEngine.h │ ├── QScheme.h │ ├── RefcountedDeleter.cpp │ ├── RefcountedDeleter.h │ ├── SafePyObject.cpp │ ├── SafePyObject.h │ ├── Scalar.cpp │ ├── Scalar.h │ ├── ScalarType.cpp │ ├── ScalarType.h │ ├── ScalarTypeToTypeMeta.h │ ├── Storage.cpp │ ├── Storage.h │ ├── StorageImpl.cpp │ ├── StorageImpl.h │ ├── Stream.cpp │ ├── Stream.h │ ├── StreamGuard.h │ ├── SymBool.cpp │ ├── SymBool.h │ ├── SymFloat.cpp │ ├── SymFloat.h │ ├── SymInt.cpp │ ├── SymInt.h │ ├── SymIntArrayRef.cpp │ ├── SymIntArrayRef.h │ ├── SymNodeImpl.cpp │ ├── SymNodeImpl.h │ ├── SymbolicShapeMeta.cpp │ ├── SymbolicShapeMeta.h │ ├── TensorImpl.cpp │ ├── TensorImpl.h │ ├── TensorOptions.cpp │ ├── TensorOptions.h │ ├── UndefinedTensorImpl.cpp │ ├── UndefinedTensorImpl.h │ ├── WrapDimMinimal.cpp │ ├── WrapDimMinimal.h │ ├── alignment.h │ ├── build.bzl │ ├── impl │ │ ├── COW.cpp │ │ ├── COW.h │ │ ├── COWDeleter.cpp │ │ ├── COWDeleter.h │ │ ├── DeviceGuardImplInterface.cpp │ │ ├── DeviceGuardImplInterface.h │ │ ├── FakeGuardImpl.h │ │ ├── GPUTrace.cpp │ │ ├── GPUTrace.h │ │ ├── HermeticPyObjectTLS.cpp │ │ ├── HermeticPyObjectTLS.h │ │ ├── InlineDeviceGuard.h │ │ ├── InlineEvent.h │ │ ├── InlineStreamGuard.h │ │ ├── LocalDispatchKeySet.cpp │ │ ├── LocalDispatchKeySet.h │ │ ├── PyInterpreter.cpp │ │ ├── PyInterpreter.h │ │ ├── PyObjectSlot.cpp │ │ ├── PyObjectSlot.h │ │ ├── PythonDispatcherTLS.cpp │ │ ├── PythonDispatcherTLS.h │ │ ├── README-cow.md │ │ ├── README.md │ │ ├── SizesAndStrides.cpp │ │ ├── SizesAndStrides.h │ │ ├── TorchDispatchModeTLS.cpp │ │ ├── TorchDispatchModeTLS.h │ │ ├── VirtualGuardImpl.h │ │ ├── alloc_cpu.cpp │ │ └── alloc_cpu.h │ ├── thread_pool.cpp │ └── thread_pool.h ├── cuda │ ├── BUILD.bazel │ ├── CMakeLists.txt │ ├── CUDAAlgorithm.h │ ├── CUDAAllocatorConfig.cpp │ ├── CUDAAllocatorConfig.h │ ├── CUDACachingAllocator.cpp │ ├── CUDACachingAllocator.h │ ├── CUDADeviceAssertion.h │ ├── CUDADeviceAssertionHost.cpp │ ├── CUDADeviceAssertionHost.h │ ├── CUDAException.cpp │ ├── CUDAException.h │ ├── CUDAFunctions.cpp │ ├── CUDAFunctions.h │ ├── CUDAGraphsC10Utils.h │ ├── CUDAGuard.h │ ├── CUDAMacros.h │ ├── CUDAMallocAsyncAllocator.cpp │ ├── CUDAMathCompat.h │ ├── CUDAMiscFunctions.cpp │ ├── CUDAMiscFunctions.h │ ├── CUDAStream.cpp │ ├── CUDAStream.h │ ├── README.md │ ├── build.bzl │ ├── driver_api.cpp │ ├── driver_api.h │ ├── impl │ │ ├── CUDAGuardImpl.cpp │ │ ├── CUDAGuardImpl.h │ │ ├── CUDATest.cpp │ │ ├── CUDATest.h │ │ └── cuda_cmake_macros.h.in │ └── test │ │ ├── BUILD.bazel │ │ ├── CMakeLists.txt │ │ ├── build.bzl │ │ └── impl │ │ ├── CUDAAssertionsTest_1_var_test.cu │ │ ├── CUDAAssertionsTest_catches_stream.cu │ │ ├── CUDAAssertionsTest_catches_thread_and_block_and_device.cu │ │ ├── CUDAAssertionsTest_from_2_processes.cu │ │ ├── CUDAAssertionsTest_multiple_writes_from_blocks_and_threads.cu │ │ ├── CUDAAssertionsTest_multiple_writes_from_multiple_blocks.cu │ │ ├── CUDAAssertionsTest_multiple_writes_from_same_block.cu │ │ └── CUDATest.cpp ├── hip │ └── CMakeLists.txt ├── macros │ ├── BUILD.bazel │ ├── Export.h │ ├── Macros.h │ ├── build.bzl │ ├── cmake_configure_file.bzl │ └── cmake_macros.h.in ├── metal │ ├── indexing.h │ ├── random.h │ ├── reduction_utils.h │ ├── special_math.h │ └── utils.h ├── mobile │ ├── BUILD.bazel │ ├── CPUCachingAllocator.cpp │ ├── CPUCachingAllocator.h │ ├── CPUProfilingAllocator.cpp │ ├── CPUProfilingAllocator.h │ └── build.bzl ├── ovrsource_defs.bzl ├── test │ ├── BUILD.bazel │ ├── CMakeLists.txt │ ├── build.bzl │ ├── core │ │ ├── CompileTimeFunctionPointer_test.cpp │ │ ├── DeviceGuard_test.cpp │ │ ├── Device_test.cpp │ │ ├── DispatchKeySet_test.cpp │ │ ├── Scalar_test.cpp │ │ ├── StreamGuard_test.cpp │ │ ├── SymInt_test.cpp │ │ └── impl │ │ │ ├── InlineDeviceGuard_test.cpp │ │ │ ├── InlineStreamGuard_test.cpp │ │ │ ├── SizesAndStrides_test.cpp │ │ │ └── cow_test.cpp │ └── util │ │ ├── ArrayRef_test.cpp │ │ ├── Bitset_test.cpp │ │ ├── ConstexprCrc_test.cpp │ │ ├── DeadlockDetection_test.cpp │ │ ├── Half_test.cpp │ │ ├── LeftRight_test.cpp │ │ ├── Macros.h │ │ ├── Metaprogramming_test.cpp │ │ ├── NetworkFlow_test.cpp │ │ ├── Synchronized_test.cpp │ │ ├── ThreadLocal_test.cpp │ │ ├── TypeIndex_test.cpp │ │ ├── TypeList_test.cpp │ │ ├── TypeTraits_test.cpp │ │ ├── accumulate_test.cpp │ │ ├── bfloat16_test.cpp │ │ ├── bit_cast_test.cpp │ │ ├── complex_math_test.cpp │ │ ├── complex_math_test_common.h │ │ ├── complex_test.cpp │ │ ├── complex_test_common.h │ │ ├── error_test.cpp │ │ ├── exception_test.cpp │ │ ├── flags_test.cpp │ │ ├── generic_math_test.cpp │ │ ├── intrusive_ptr_test.cpp │ │ ├── irange_test.cpp │ │ ├── lazy_test.cpp │ │ ├── logging_test.cpp │ │ ├── optional_test.cpp │ │ ├── ordered_preserving_dict_test.cpp │ │ ├── registry_test.cpp │ │ ├── small_vector_test.cpp │ │ ├── ssize_test.cpp │ │ ├── string_util_test.cpp │ │ ├── string_view_test.cpp │ │ ├── tempfile_test.cpp │ │ └── typeid_test.cpp ├── util │ ├── AbortHandler.h │ ├── AlignOf.h │ ├── ApproximateClock.cpp │ ├── ApproximateClock.h │ ├── Array.h │ ├── ArrayRef.h │ ├── BFloat16-inl.h │ ├── BFloat16-math.h │ ├── BFloat16.h │ ├── BUILD.bazel │ ├── Backtrace.cpp │ ├── Backtrace.h │ ├── Bfloat16.cpp │ ├── Bitset.h │ ├── C++17.cpp │ ├── C++17.h │ ├── CallOnce.h │ ├── ConstexprCrc.h │ ├── DeadlockDetection.cpp │ ├── DeadlockDetection.h │ ├── Deprecated.h │ ├── DimVector.h │ ├── DynamicCounter.cpp │ ├── DynamicCounter.h │ ├── Exception.cpp │ ├── Exception.h │ ├── ExclusivelyOwned.h │ ├── ExclusivelyOwnedTensorTraits.h │ ├── FbcodeMaps.h │ ├── Flags.h │ ├── Float8_e4m3fn-inl.h │ ├── Float8_e4m3fn.cpp │ ├── Float8_e4m3fn.h │ ├── Float8_e4m3fnuz-inl.h │ ├── Float8_e4m3fnuz.cpp │ ├── Float8_e4m3fnuz.h │ ├── Float8_e5m2-inl.h │ ├── Float8_e5m2.cpp │ ├── Float8_e5m2.h │ ├── Float8_e5m2fnuz-inl.h │ ├── Float8_e5m2fnuz.cpp │ ├── Float8_e5m2fnuz.h │ ├── Float8_e8m0fnu-inl.h │ ├── Float8_e8m0fnu.cpp │ ├── Float8_e8m0fnu.h │ ├── Float8_fnuz_cvt.h │ ├── FunctionRef.h │ ├── Gauge.cpp │ ├── Gauge.h │ ├── Half-inl.h │ ├── Half.cpp │ ├── Half.h │ ├── IdWrapper.h │ ├── Lazy.h │ ├── LeftRight.cpp │ ├── LeftRight.h │ ├── Load.h │ ├── Logging.cpp │ ├── Logging.h │ ├── MathConstants.cpp │ ├── MathConstants.h │ ├── MaybeOwned.h │ ├── Metaprogramming.cpp │ ├── Metaprogramming.h │ ├── NetworkFlow.cpp │ ├── NetworkFlow.h │ ├── Optional.cpp │ ├── Optional.h │ ├── OptionalArrayRef.h │ ├── ParallelGuard.cpp │ ├── ParallelGuard.h │ ├── Registry.h │ ├── ScopeExit.h │ ├── SmallBuffer.h │ ├── SmallVector.cpp │ ├── SmallVector.h │ ├── StringUtil.cpp │ ├── StringUtil.h │ ├── Synchronized.h │ ├── ThreadLocal.h │ ├── ThreadLocalDebugInfo.cpp │ ├── ThreadLocalDebugInfo.h │ ├── Type.h │ ├── TypeCast.cpp │ ├── TypeCast.h │ ├── TypeIndex.h │ ├── TypeList.cpp │ ├── TypeList.h │ ├── TypeSafeSignMath.h │ ├── TypeTraits.cpp │ ├── TypeTraits.h │ ├── Type_demangle.cpp │ ├── Type_no_demangle.cpp │ ├── Unicode.cpp │ ├── Unicode.h │ ├── UniqueVoidPtr.cpp │ ├── UniqueVoidPtr.h │ ├── Unroll.h │ ├── WaitCounter.cpp │ ├── WaitCounter.h │ ├── WaitCounterDynamicBackend.h │ ├── accumulate.h │ ├── bit_cast.h │ ├── bits.h │ ├── build.bzl │ ├── complex.h │ ├── complex_math.cpp │ ├── complex_math.h │ ├── complex_utils.h │ ├── copysign.h │ ├── env.cpp │ ├── env.h │ ├── error.cpp │ ├── error.h │ ├── flags_use_gflags.cpp │ ├── flags_use_no_gflags.cpp │ ├── flat_hash_map.h │ ├── floating_point_utils.h │ ├── generic_math.h │ ├── hash.h │ ├── int128.cpp │ ├── int128.h │ ├── intrusive_ptr.cpp │ ├── intrusive_ptr.h │ ├── irange.h │ ├── llvmMathExtras.h │ ├── logging_is_google_glog.h │ ├── logging_is_not_google_glog.h │ ├── numa.cpp │ ├── numa.h │ ├── order_preserving_flat_hash_map.h │ ├── overflows.h │ ├── overloaded.h │ ├── python_stub.h │ ├── qint32.h │ ├── qint8.h │ ├── quint2x4.h │ ├── quint4x2.h │ ├── quint8.h │ ├── safe_numerics.h │ ├── signal_handler.cpp │ ├── signal_handler.h │ ├── sparse_bitset.h │ ├── ssize.h │ ├── static_tracepoint.h │ ├── static_tracepoint_elfx86.h │ ├── strides.h │ ├── string_utils.h │ ├── string_view.h │ ├── strong_type.h │ ├── tempfile.cpp │ ├── tempfile.h │ ├── thread_name.cpp │ ├── thread_name.h │ ├── typeid.cpp │ ├── typeid.h │ └── win32-headers.h └── xpu │ ├── CMakeLists.txt │ ├── XPUCachingAllocator.cpp │ ├── XPUCachingAllocator.h │ ├── XPUDeviceProp.h │ ├── XPUException.h │ ├── XPUFunctions.cpp │ ├── XPUFunctions.h │ ├── XPUMacros.h │ ├── XPUStream.cpp │ ├── XPUStream.h │ ├── impl │ ├── XPUGuardImpl.cpp │ ├── XPUGuardImpl.h │ └── xpu_cmake_macros.h.in │ └── test │ ├── CMakeLists.txt │ └── impl │ ├── XPUCachingAllocatorTest.cpp │ ├── XPUDeviceTest.cpp │ ├── XPUGuardTest.cpp │ ├── XPUStreamTest.cpp │ └── XPUTest.h ├── caffe2 ├── .clang-format ├── CMakeLists.txt ├── core │ ├── CMakeLists.txt │ ├── common.cc │ ├── common.h │ ├── macros.h │ ├── macros.h.in │ └── timer.h ├── perfkernels │ ├── CMakeLists.txt │ ├── batch_box_cox_avx512.cc │ ├── batch_box_cox_vec.h │ ├── common.h │ ├── common_avx.cc │ ├── common_avx2.cc │ ├── common_sve.cc │ ├── embedding_lookup_idx.cc │ ├── embedding_lookup_idx.h │ ├── embedding_lookup_idx_avx2.cc │ ├── embedding_lookup_idx_sve.cc │ ├── hp_emblookup_codegen.py │ └── sve_emblookup_codegen.py ├── serialize │ ├── CMakeLists.txt │ ├── crc.cc │ ├── crc_alt.h │ ├── file_adapter.cc │ ├── file_adapter.h │ ├── in_memory_adapter.h │ ├── inline_container.cc │ ├── inline_container.h │ ├── inline_container_test.cc │ ├── istream_adapter.cc │ ├── istream_adapter.h │ ├── read_adapter_interface.cc │ ├── read_adapter_interface.h │ └── versions.h ├── unexported_symbols.lds ├── utils │ ├── CMakeLists.txt │ ├── fixed_divisor.h │ ├── proto_wrap.cc │ ├── proto_wrap.h │ ├── string_utils.cc │ ├── string_utils.h │ └── threadpool │ │ ├── ThreadPool.cc │ │ ├── ThreadPool.h │ │ ├── ThreadPoolCommon.h │ │ ├── WorkersPool.h │ │ ├── pthreadpool-cpp.cc │ │ ├── pthreadpool-cpp.h │ │ ├── pthreadpool.cc │ │ ├── pthreadpool.h │ │ ├── pthreadpool_impl.cc │ │ ├── thread_pool_guard.cpp │ │ └── thread_pool_guard.h └── version_script.lds ├── cmake ├── Allowlist.cmake ├── BuildVariables.cmake ├── Caffe2Config.cmake.in ├── CheckAbi.cmake ├── Codegen.cmake ├── DebugHelper.cmake ├── Dependencies.cmake ├── External │ ├── EigenBLAS.cmake │ ├── aotriton.cmake │ ├── nccl.cmake │ ├── nnpack.cmake │ ├── rccl.cmake │ └── ucc.cmake ├── FlatBuffers.cmake ├── GoogleTestPatch.cmake ├── IncludeSource.cpp.in ├── Metal.cmake ├── MiscCheck.cmake ├── Modules │ ├── FindAPL.cmake │ ├── FindARM.cmake │ ├── FindAVX.cmake │ ├── FindAtlas.cmake │ ├── FindBLAS.cmake │ ├── FindBLIS.cmake │ ├── FindBenchmark.cmake │ ├── FindCUB.cmake │ ├── FindCUDAToolkit.cmake │ ├── FindCUDSS.cmake │ ├── FindCUSPARSELT.cmake │ ├── FindFlexiBLAS.cmake │ ├── FindGloo.cmake │ ├── FindITT.cmake │ ├── FindLAPACK.cmake │ ├── FindMAGMA.cmake │ ├── FindMKL.cmake │ ├── FindMKLDNN.cmake │ ├── FindNCCL.cmake │ ├── FindNuma.cmake │ ├── FindOpenBLAS.cmake │ ├── FindOpenMP.cmake │ ├── FindOpenTelemetryApi.cmake │ ├── FindSYCLToolkit.cmake │ ├── FindSanitizer.cmake │ ├── FindVSX.cmake │ ├── FindZVECTOR.cmake │ ├── Findpybind11.cmake │ ├── FindvecLib.cmake │ └── README.md ├── Modules_CUDA_fix │ ├── FindCUDA.cmake │ ├── FindCUDNN.cmake │ ├── README.md │ └── upstream │ │ ├── CMakeInitializeConfigs.cmake │ │ ├── FindCUDA.cmake │ │ ├── FindCUDA │ │ ├── make2cmake.cmake │ │ ├── parse_cubin.cmake │ │ ├── run_nvcc.cmake │ │ └── select_compute_arch.cmake │ │ ├── FindPackageHandleStandardArgs.cmake │ │ ├── FindPackageMessage.cmake │ │ └── README.md ├── ProtoBuf.cmake ├── ProtoBufPatch.cmake ├── Summary.cmake ├── TorchConfig.cmake.in ├── TorchConfigVersion.cmake.in ├── VulkanCodegen.cmake ├── VulkanDependencies.cmake ├── cmake_uninstall.cmake.in ├── iOS.cmake ├── prioritized_text.txt └── public │ ├── ComputeLibrary.cmake │ ├── LoadHIP.cmake │ ├── cuda.cmake │ ├── gflags.cmake │ ├── glog.cmake │ ├── mkl.cmake │ ├── mkldnn.cmake │ ├── protobuf.cmake │ ├── utils.cmake │ └── xpu.cmake ├── defs.bzl ├── docker.Makefile ├── docs ├── .gitignore ├── Makefile ├── README.md ├── cpp │ ├── Makefile │ └── source │ │ ├── Doxyfile │ │ ├── _static │ │ └── cpp_theme.css │ │ ├── check-doxygen.sh │ │ ├── conf.py │ │ ├── frontend.rst │ │ ├── index.rst │ │ ├── installing.rst │ │ ├── library.rst │ │ └── notes │ │ ├── faq.rst │ │ ├── inference_mode.rst │ │ ├── maybe_owned.rst │ │ ├── tensor_basics.rst │ │ ├── tensor_creation.rst │ │ ├── tensor_cuda_stream.rst │ │ ├── tensor_indexing.rst │ │ └── versioning.rst ├── libtorch.rst ├── make.bat ├── requirements.txt └── source │ ├── _static │ ├── css │ │ ├── copybutton.css │ │ ├── custom.css │ │ └── jit.css │ └── img │ │ ├── aliastracker_graph.png │ │ ├── distributed_autograd │ │ ├── distributed_dependencies.png │ │ ├── distributed_dependencies_computed.png │ │ ├── local_dependencies.png │ │ └── send_recv_functions.png │ │ ├── dynamic_graph.gif │ │ ├── dynamo │ │ ├── TorchDynamo.png │ │ ├── flowchart.jpg │ │ ├── td_stack.png │ │ └── torchinductor_backend.png │ │ ├── fine_grained_apis │ │ ├── api_diagram.png │ │ └── call_stack_diagram.png │ │ ├── inductor_profiling │ │ ├── inductor_code.png │ │ ├── kernel_breakdown.png │ │ ├── terminal_printout.png │ │ └── trace.png │ │ ├── masked │ │ └── tensor_comparison.jpg │ │ ├── meshgrid.png │ │ ├── nested │ │ └── njt_visual.png │ │ ├── nn │ │ └── layer_norm.jpg │ │ ├── onnx │ │ ├── onnx_dynamo_mlp_model.png │ │ ├── torch_dynamo_exporter_memory_usage.png │ │ └── torch_script_exporter_memory_usage.png │ │ ├── pipeline_parallelism │ │ ├── no_pipe.png │ │ └── pipe.png │ │ ├── profiling_torch_compile │ │ ├── ac2g.png │ │ ├── basic_chrome_trace.png │ │ ├── compilation_profiling.png │ │ ├── cpu_bound.png │ │ ├── graph_breaks.png │ │ ├── graph_breaks_with_torch_compiled_region.png │ │ ├── kernel_launch_labeled.png │ │ ├── noninductor_triton_kernel.png │ │ └── triton_kernel_launch.png │ │ ├── pytorch-logo-dark-unstable.png │ │ ├── pytorch-logo-dark.png │ │ ├── pytorch-logo-dark.svg │ │ ├── pytorch-logo-flame.png │ │ ├── pytorch-logo-flame.svg │ │ ├── rpc_arch.png │ │ ├── tensor_illustration.png │ │ ├── tensorboard │ │ ├── add_histogram.png │ │ ├── add_histogram_raw.png │ │ ├── add_hparam.png │ │ ├── add_image.png │ │ ├── add_images.png │ │ ├── add_scalar.png │ │ ├── add_scalars.png │ │ └── hier_tags.png │ │ └── torch_cuda_memory │ │ ├── active_memory_timeline.png │ │ └── allocator_state_history.png │ ├── _templates │ ├── autosummary │ │ ├── class.rst │ │ └── classnoinheritance.rst │ ├── classtemplate.rst │ ├── layout.html │ └── sobolengine.rst │ ├── accelerator.rst │ ├── amp.rst │ ├── autograd.rst │ ├── backends.rst │ ├── benchmark_utils.rst │ ├── bottleneck.rst │ ├── checkpoint.rst │ ├── community │ ├── build_ci_governance.rst │ ├── contribution_guide.rst │ ├── design.rst │ ├── governance.rst │ └── persons_of_interest.rst │ ├── complex_numbers.rst │ ├── cond.rst │ ├── conf.py │ ├── config_mod.rst │ ├── cpp_extension.rst │ ├── cpp_index.rst │ ├── cpu.rst │ ├── cuda._sanitizer.rst │ ├── cuda.rst │ ├── cuda.tunable.rst │ ├── cuda_environment_variables.rst │ ├── cudnn_persistent_rnn.rst │ ├── cudnn_rnn_determinism.rst │ ├── data.rst │ ├── ddp_comm_hooks.rst │ ├── debugging_environment_variables.rst │ ├── deploy.rst │ ├── deterministic.rst │ ├── distributed.algorithms.join.rst │ ├── distributed.checkpoint.rst │ ├── distributed.elastic.rst │ ├── distributed.fsdp.fully_shard.rst │ ├── distributed.optim.rst │ ├── distributed.pipelining.rst │ ├── distributed.rst │ ├── distributed.tensor.parallel.rst │ ├── distributed.tensor.rst │ ├── distributions.rst │ ├── dlpack.rst │ ├── docutils.conf │ ├── elastic │ ├── agent.rst │ ├── agent_diagram.jpg │ ├── control_plane.rst │ ├── customization.rst │ ├── errors.rst │ ├── etcd_rdzv_diagram.png │ ├── events.rst │ ├── examples.rst │ ├── kubernetes.rst │ ├── metrics.rst │ ├── multiprocessing.rst │ ├── quickstart.rst │ ├── rendezvous.rst │ ├── run.rst │ ├── subprocess_handler.rst │ ├── timer.rst │ └── train_script.rst │ ├── export.ir_spec.rst │ ├── export.programming_model.rst │ ├── export.rst │ ├── fft.rst │ ├── fsdp.rst │ ├── func.api.rst │ ├── func.batch_norm.rst │ ├── func.migrating.rst │ ├── func.rst │ ├── func.ux_limitations.rst │ ├── func.whirlwind_tour.rst │ ├── future_mod.rst │ ├── futures.rst │ ├── fx.experimental.rst │ ├── fx.rst │ ├── hub.rst │ ├── index.rst │ ├── jit.rst │ ├── jit_builtin_functions.rst │ ├── jit_language_reference.rst │ ├── jit_language_reference_v2.rst │ ├── jit_python_reference.rst │ ├── jit_unsupported.rst │ ├── jit_utils.rst │ ├── library.rst │ ├── linalg.rst │ ├── logging.rst │ ├── masked.rst │ ├── math-quantizer-equation.png │ ├── meta.rst │ ├── miscellaneous_environment_variables.rst │ ├── mobile_optimizer.rst │ ├── model_zoo.rst │ ├── module_tracker.rst │ ├── monitor.rst │ ├── mps.rst │ ├── mps_environment_variables.rst │ ├── mtia.memory.rst │ ├── mtia.rst │ ├── multiprocessing.rst │ ├── name_inference.rst │ ├── named_tensor.rst │ ├── nested.rst │ ├── nn.attention.bias.rst │ ├── nn.attention.experimental.rst │ ├── nn.attention.flex_attention.rst │ ├── nn.attention.rst │ ├── nn.functional.rst │ ├── nn.init.rst │ ├── nn.rst │ ├── notes │ ├── amp_examples.rst │ ├── autograd.rst │ ├── broadcasting.rst │ ├── cpu_threading_runtimes.svg │ ├── cpu_threading_torchscript_inference.rst │ ├── cpu_threading_torchscript_inference.svg │ ├── cuda.rst │ ├── custom_operators.rst │ ├── ddp.rst │ ├── extending.func.rst │ ├── extending.rst │ ├── faq.rst │ ├── fsdp.rst │ ├── get_start_xpu.rst │ ├── gradcheck.rst │ ├── hip.rst │ ├── large_scale_deployments.rst │ ├── modules.rst │ ├── mps.rst │ ├── multiprocessing.rst │ ├── numerical_accuracy.rst │ ├── randomness.rst │ ├── serialization.rst │ └── windows.rst │ ├── onnx.rst │ ├── onnx_dynamo.rst │ ├── onnx_dynamo_memory_usage.rst │ ├── onnx_dynamo_onnxruntime_backend.rst │ ├── onnx_torchscript.rst │ ├── onnx_torchscript_supported_aten_ops.rst │ ├── optim.rst │ ├── package.rst │ ├── profiler.rst │ ├── quantization-accuracy-debugging.rst │ ├── quantization-backend-configuration.rst │ ├── quantization-support.rst │ ├── quantization.rst │ ├── random.rst │ ├── rpc.rst │ ├── rpc │ ├── distributed_autograd.rst │ └── rref.rst │ ├── scripts │ ├── build_activation_images.py │ ├── build_opsets.py │ ├── build_quantization_configs.py │ ├── exportdb │ │ ├── blurb.txt │ │ └── generate_example_rst.py │ └── onnx │ │ └── build_onnx_torchscript_supported_aten_op_csv_table.py │ ├── signal.rst │ ├── size.rst │ ├── sparse.rst │ ├── special.rst │ ├── storage.rst │ ├── tensor_attributes.rst │ ├── tensor_view.rst │ ├── tensorboard.rst │ ├── tensors.rst │ ├── testing.rst │ ├── threading_environment_variables.rst │ ├── torch.ao.ns._numeric_suite.rst │ ├── torch.ao.ns._numeric_suite_fx.rst │ ├── torch.compiler.config.rst │ ├── torch.compiler.rst │ ├── torch.compiler_aot_inductor.rst │ ├── torch.compiler_aot_inductor_minifier.rst │ ├── torch.compiler_api.rst │ ├── torch.compiler_best_practices_for_backends.rst │ ├── torch.compiler_cudagraph_trees.rst │ ├── torch.compiler_custom_backends.rst │ ├── torch.compiler_dynamic_shapes.rst │ ├── torch.compiler_dynamo_deepdive.rst │ ├── torch.compiler_dynamo_overview.rst │ ├── torch.compiler_fake_tensor.rst │ ├── torch.compiler_faq.rst │ ├── torch.compiler_fine_grain_apis.rst │ ├── torch.compiler_get_started.rst │ ├── torch.compiler_inductor_profiling.rst │ ├── torch.compiler_ir.rst │ ├── torch.compiler_nn_module.rst │ ├── torch.compiler_performance_dashboard.rst │ ├── torch.compiler_profiling_torch_compile.rst │ ├── torch.compiler_transformations.rst │ ├── torch.compiler_troubleshooting.rst │ ├── torch.compiler_troubleshooting_old.rst │ ├── torch.overrides.rst │ ├── torch.rst │ ├── torch_cuda_memory.rst │ ├── torch_environment_variables.rst │ ├── torch_nccl_environment_variables.rst │ ├── type_info.rst │ ├── utils.rst │ └── xpu.rst ├── example ├── README.md ├── __init__.py ├── bootstrap │ ├── README.md │ ├── __init__.py │ └── bootstrap.py ├── helr │ ├── README.md │ ├── data │ │ ├── MNIST_test.txt │ │ └── MNIST_train.txt │ └── helr.py ├── resnet │ ├── README.md │ ├── __init__.py │ ├── approx.py │ ├── convs.py │ ├── resnet20.py │ └── utils.py └── sort │ ├── README.md │ └── sort.py ├── functorch ├── .gitignore ├── CMakeLists.txt ├── COMPILE_README.md ├── README.md ├── __init__.py ├── _src │ ├── __init__.py │ ├── aot_autograd │ │ └── __init__.py │ ├── eager_transforms │ │ └── __init__.py │ ├── make_functional │ │ └── __init__.py │ └── vmap │ │ └── __init__.py ├── benchmarks │ ├── chrome_trace_parser.py │ ├── cse.py │ ├── operator_authoring.py │ ├── per_sample_grads.py │ ├── pointwise_scorecard.py │ └── process_scorecard.py ├── compile │ └── __init__.py ├── csrc │ ├── dim │ │ ├── arena.h │ │ ├── dim.cpp │ │ ├── dim.h │ │ ├── dim_opcode.c │ │ ├── minpybind.h │ │ └── python_variable_simple.h │ └── init_dim_only.cpp ├── dim │ ├── README.md │ ├── __init__.py │ ├── batch_tensor.py │ ├── delayed_mul_tensor.py │ ├── dim.py │ ├── magic_trace.py │ ├── op_properties.py │ ├── reference.py │ ├── tree_map.py │ └── wrap_type.py ├── docs │ ├── .gitignore │ ├── Makefile │ ├── README.md │ └── source │ │ ├── _static │ │ └── css │ │ │ └── custom.css │ │ ├── _templates │ │ ├── autosummary │ │ │ └── class.rst │ │ ├── classtemplate.rst │ │ └── layout.html │ │ ├── aot_autograd.rst │ │ ├── batch_norm.rst │ │ ├── conf.py │ │ ├── docutils.conf │ │ ├── experimental.rst │ │ ├── functorch.rst │ │ ├── index.rst │ │ ├── install.rst │ │ ├── notebooks │ │ ├── _src │ │ │ ├── plot_ensembling.py │ │ │ ├── plot_jacobians_and_hessians.py │ │ │ └── plot_per_sample_gradients.py │ │ ├── aot_autograd_optimizations.ipynb │ │ ├── ensembling.ipynb │ │ ├── jacobians_hessians.ipynb │ │ ├── minifier.ipynb │ │ ├── neural_tangent_kernels.ipynb │ │ ├── per_sample_grads.ipynb │ │ └── whirlwind_tour.ipynb │ │ └── ux_limitations.rst ├── einops │ ├── __init__.py │ ├── _parsing.py │ └── rearrange.py ├── examples │ ├── .gitignore │ ├── compilation │ │ ├── README.md │ │ ├── eager_fusion.py │ │ ├── fuse_module.py │ │ ├── linear_train.py │ │ └── simple_function.py │ ├── dp_cifar10 │ │ ├── README.md │ │ ├── cifar10_opacus.py │ │ └── cifar10_transforms.py │ ├── ensembling │ │ └── parallel_train.py │ ├── lennard_jones │ │ └── lennard_jones.py │ ├── maml_omniglot │ │ ├── .gitignore │ │ ├── README.md │ │ ├── maml-omniglot-higher.py │ │ ├── maml-omniglot-ptonly.py │ │ ├── maml-omniglot-transforms.py │ │ └── support │ │ │ └── omniglot_loaders.py │ └── maml_regression │ │ ├── evjang.py │ │ ├── evjang_transforms.py │ │ └── evjang_transforms_module.py ├── experimental │ ├── __init__.py │ ├── control_flow.py │ └── ops.py ├── notebooks │ ├── _src │ │ ├── plot_ensembling.py │ │ ├── plot_jacobians_and_hessians.py │ │ └── plot_per_sample_gradients.py │ ├── aot_autograd_optimizations.ipynb │ ├── ensembling.ipynb │ ├── jacobians_hessians.ipynb │ ├── minifier.ipynb │ ├── neural_tangent_kernels.ipynb │ ├── per_sample_grads.ipynb │ └── whirlwind_tour.ipynb ├── op_analysis │ ├── annotated_ops │ ├── gen_data.py │ └── public_api └── writing_batching_rules.md ├── mypy-strict.ini ├── mypy.ini ├── mypy_plugins ├── check_mypy_version.py └── sympy_mypy_plugin.py ├── pt_ops.bzl ├── pt_template_srcs.bzl ├── pyproject.toml ├── pytest.ini ├── requirements.txt ├── scripts ├── README.md ├── add_apache_header.sh ├── analysis │ ├── format_test_csv.py │ └── run_test_csv.sh ├── apache_header.txt ├── apache_python.txt ├── build_android.sh ├── build_android_gradle.sh ├── build_host_protoc.sh ├── build_ios.sh ├── build_local.sh ├── build_mobile.sh ├── build_pytorch_android.sh ├── build_raspbian.sh ├── build_tegra_x1.sh ├── build_tizen.sh ├── build_windows.bat ├── compile_tests │ ├── common.py │ ├── download_reports.py │ ├── failures_histogram.py │ ├── passrate.py │ └── update_failures.py ├── diagnose_protobuf.py ├── export │ └── update_schema.py ├── fbcode-dev-setup │ └── ccache_setup.sh ├── get_python_cmake_flags.py ├── install_triton_wheel.sh ├── jit │ └── log_extract.py ├── onnx │ ├── install-develop.sh │ ├── install.sh │ └── test.sh ├── proto.ps1 ├── release │ ├── README.md │ ├── apply-release-changes.sh │ ├── cut-release-branch.sh │ ├── promote │ │ ├── common_utils.sh │ │ ├── prep_binary_for_pypi.sh │ │ ├── s3_to_s3.sh │ │ └── wheel_to_pypi.sh │ ├── restore-backup.sh │ └── tag-docker-images.sh ├── release_notes │ ├── .gitignore │ ├── README.md │ ├── apply_categories.py │ ├── categorize.py │ ├── classifier.py │ ├── commitlist.py │ ├── common.py │ ├── explore.ipynb │ ├── namespace_check.py │ ├── requirements.txt │ └── test_release_notes.py ├── remove_apache_header.sh ├── temp.sh └── xcode_build.rb ├── setup.py ├── test ├── HowToWriteTestsUsingFileCheck.md ├── _test_bazel.py ├── allowlist_for_publicAPI.json ├── ao │ └── sparsity │ │ ├── test_activation_sparsifier.py │ │ ├── test_composability.py │ │ ├── test_data_scheduler.py │ │ ├── test_data_sparsifier.py │ │ ├── test_kernels.py │ │ ├── test_parametrization.py │ │ ├── test_qlinear_packed_params.py │ │ ├── test_scheduler.py │ │ ├── test_sparsifier.py │ │ ├── test_sparsity_utils.py │ │ └── test_structured_sparsifier.py ├── autograd │ ├── test_complex.py │ ├── test_functional.py │ └── test_logging.py ├── backends │ └── xeon │ │ └── test_launch.py ├── benchmark_utils │ ├── callgrind_artifacts.json │ └── test_benchmark_utils.py ├── bottleneck_test │ ├── test.py │ ├── test_args.py │ └── test_cuda.py ├── conftest.py ├── cpp │ ├── __init__.py │ ├── aoti_abi_check │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── main.cpp │ │ ├── test_cast.cpp │ │ ├── test_dtype.cpp │ │ ├── test_math.cpp │ │ ├── test_rand.cpp │ │ └── test_vec.cpp │ ├── aoti_inference │ │ ├── CMakeLists.txt │ │ ├── aoti_custom_class.cpp │ │ ├── aoti_custom_class.h │ │ ├── compile_model.py │ │ ├── generate_lowered_cpu.py │ │ ├── standalone_compile.sh │ │ ├── standalone_test.cpp │ │ ├── test.cpp │ │ └── test.py │ ├── api │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── any.cpp │ │ ├── autograd.cpp │ │ ├── dataloader.cpp │ │ ├── dispatch.cpp │ │ ├── enum.cpp │ │ ├── expanding-array.cpp │ │ ├── fft.cpp │ │ ├── functional.cpp │ │ ├── grad_mode.cpp │ │ ├── inference_mode.cpp │ │ ├── init.cpp │ │ ├── init_baseline.h │ │ ├── init_baseline.py │ │ ├── integration.cpp │ │ ├── ivalue.cpp │ │ ├── jit.cpp │ │ ├── memory.cpp │ │ ├── meta_tensor.cpp │ │ ├── misc.cpp │ │ ├── module.cpp │ │ ├── moduledict.cpp │ │ ├── modulelist.cpp │ │ ├── modules.cpp │ │ ├── namespace.cpp │ │ ├── nested.cpp │ │ ├── nested_int.cpp │ │ ├── nn_utils.cpp │ │ ├── operations.cpp │ │ ├── optim.cpp │ │ ├── optim_baseline.h │ │ ├── optim_baseline.py │ │ ├── ordered_dict.cpp │ │ ├── parallel.cpp │ │ ├── parallel_benchmark.cpp │ │ ├── parameterdict.cpp │ │ ├── parameterlist.cpp │ │ ├── rnn.cpp │ │ ├── sequential.cpp │ │ ├── serialize.cpp │ │ ├── special.cpp │ │ ├── static.cpp │ │ ├── support.cpp │ │ ├── support.h │ │ ├── tensor.cpp │ │ ├── tensor_cuda.cpp │ │ ├── tensor_flatten.cpp │ │ ├── tensor_indexing.cpp │ │ ├── tensor_options.cpp │ │ ├── tensor_options_cuda.cpp │ │ ├── torch_include.cpp │ │ └── transformer.cpp │ ├── c10d │ │ ├── BackoffTest.cpp │ │ ├── CMakeLists.txt │ │ ├── CUDATest.cu │ │ ├── CUDATest.hpp │ │ ├── FileStoreTest.cpp │ │ ├── HashStoreTest.cpp │ │ ├── ProcessGroupGlooAsyncTest.cpp │ │ ├── ProcessGroupGlooTest.cpp │ │ ├── ProcessGroupMPITest.cpp │ │ ├── ProcessGroupNCCLErrorsTest.cpp │ │ ├── ProcessGroupNCCLTest.cpp │ │ ├── ProcessGroupUCCTest.cpp │ │ ├── StoreTestCommon.hpp │ │ ├── TCPStoreTest.cpp │ │ ├── TestUtils.hpp │ │ └── example │ │ │ └── allreduce.cpp │ ├── common │ │ ├── main.cpp │ │ └── support.h │ ├── dist_autograd │ │ ├── CMakeLists.txt │ │ └── test_dist_autograd.cpp │ ├── jit │ │ ├── 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 │ │ └── upgrader_models │ │ │ ├── test_versioned_div_scalar_float_v2.ptl │ │ │ ├── test_versioned_div_scalar_float_v2.ptl.ff │ │ │ ├── test_versioned_div_scalar_inplace_float_v2.ptl │ │ │ ├── test_versioned_div_scalar_inplace_float_v2.ptl.ff │ │ │ ├── test_versioned_div_scalar_inplace_int_v2.ptl │ │ │ ├── test_versioned_div_scalar_inplace_int_v2.ptl.ff │ │ │ ├── test_versioned_div_scalar_int_v2.ptl │ │ │ ├── test_versioned_div_scalar_int_v2.ptl.ff │ │ │ ├── test_versioned_div_scalar_reciprocal_float_v2.ptl │ │ │ ├── test_versioned_div_scalar_reciprocal_float_v2.ptl.ff │ │ │ ├── test_versioned_div_scalar_reciprocal_int_v2.ptl │ │ │ ├── test_versioned_div_scalar_reciprocal_int_v2.ptl.ff │ │ │ ├── test_versioned_div_scalar_scalar_v2.ptl │ │ │ ├── test_versioned_div_scalar_scalar_v2.ptl.ff │ │ │ ├── test_versioned_div_tensor_inplace_v2.ptl │ │ │ ├── test_versioned_div_tensor_inplace_v2.ptl.ff │ │ │ ├── test_versioned_div_tensor_out_v2.ptl │ │ │ ├── test_versioned_div_tensor_out_v2.ptl.ff │ │ │ ├── test_versioned_div_tensor_v2.ptl │ │ │ └── test_versioned_div_tensor_v2.ptl.ff │ ├── lazy │ │ ├── CMakeLists.txt │ │ ├── test_backend_device.cpp │ │ ├── test_cache.cpp │ │ ├── test_ir.cpp │ │ ├── test_ir_util.cpp │ │ ├── test_lazy_graph_executor.cpp │ │ ├── test_lazy_ops.cpp │ │ ├── test_lazy_ops_util.cpp │ │ ├── test_lazy_ops_util.h │ │ ├── test_misc.cpp │ │ ├── test_permutation_util.cpp │ │ ├── test_shape.cpp │ │ ├── test_tensor_impl.cpp │ │ ├── test_trie_cache.cpp │ │ └── test_util.cpp │ ├── lite_interpreter_runtime │ │ ├── CMakeLists.txt │ │ ├── delegate_test.ptl │ │ ├── delegated_submodule_with_debug_info.ptl │ │ ├── light_model.ptl │ │ ├── main.cpp │ │ ├── resources.h │ │ ├── sequence.ptl │ │ ├── test_backend_for_profiling.ptl │ │ ├── test_lite_interpreter_runtime.cpp │ │ ├── test_mobile_profiler.cpp │ │ └── to_be_profiled_module.ptl │ ├── monitor │ │ ├── test_counters.cpp │ │ └── test_events.cpp │ ├── profiler │ │ ├── containers.cpp │ │ ├── perf_events.cpp │ │ └── record_function.cpp │ ├── rpc │ │ ├── CMakeLists.txt │ │ ├── e2e_test_base.cpp │ │ ├── e2e_test_base.h │ │ ├── test_e2e_tensorpipe.cpp │ │ ├── test_tensorpipe_serialization.cpp │ │ └── test_wire_serialization.cpp │ └── tensorexpr │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── gtest_assert_float_eq.h │ │ ├── padded_buffer.cpp │ │ ├── padded_buffer.h │ │ ├── test_approx.cpp │ │ ├── test_aten.cpp │ │ ├── test_base.h │ │ ├── test_boundsinference.cpp │ │ ├── test_conv.cpp │ │ ├── test_cpp_codegen.cpp │ │ ├── test_cuda.cpp │ │ ├── test_dynamic_shapes.cpp │ │ ├── test_expr.cpp │ │ ├── test_external_calls.cpp │ │ ├── test_graph_opt.cpp │ │ ├── test_ir_printer.cpp │ │ ├── test_ir_verifier.cpp │ │ ├── test_kernel.cpp │ │ ├── test_llvm.cpp │ │ ├── test_loopnest.cpp │ │ ├── test_memdependency.cpp │ │ ├── test_memplanning.cpp │ │ ├── test_ops.cpp │ │ ├── test_quantization.cpp │ │ ├── test_reductions.cpp │ │ ├── test_registerizer.cpp │ │ ├── test_simplify.cpp │ │ ├── test_te_fuser_pass.cpp │ │ ├── test_type.cpp │ │ ├── test_type_specializations.cpp │ │ ├── test_utils.h │ │ └── tutorial.cpp ├── cpp_api_parity │ ├── __init__.py │ ├── functional_impl_check.py │ ├── module_impl_check.py │ ├── parity-tracker.md │ ├── parity_table_parser.py │ ├── sample_functional.py │ ├── sample_module.py │ └── utils.py ├── cpp_extensions │ ├── cpp_c10d_extension.cpp │ ├── cpp_c10d_extension.hpp │ ├── cpp_frontend_extension.cpp │ ├── cublas_extension.cpp │ ├── cuda_dlink_extension.cpp │ ├── cuda_dlink_extension_add.cu │ ├── cuda_dlink_extension_add.cuh │ ├── cuda_dlink_extension_kernel.cu │ ├── cuda_extension.cpp │ ├── cuda_extension.cu │ ├── cuda_extension_kernel.cu │ ├── cuda_extension_kernel2.cu │ ├── cudnn_extension.cpp │ ├── cusolver_extension.cpp │ ├── dangling_impl_extension.cpp │ ├── doubler.h │ ├── extension.cpp │ ├── identity.cpp │ ├── jit_extension.cpp │ ├── jit_extension2.cpp │ ├── maia_extension.cpp │ ├── mps_extension.mm │ ├── mtia_extension.cpp │ ├── no_python_abi_suffix_test │ │ ├── no_python_abi_suffix_test.cpp │ │ └── setup.py │ ├── open_registration_extension.cpp │ ├── open_registration_extension │ │ ├── README.md │ │ ├── pytorch_openreg │ │ │ ├── __init__.py │ │ │ ├── _aten_impl.py │ │ │ ├── _device_daemon.py │ │ │ ├── _meta_parser.py │ │ │ └── csrc │ │ │ │ ├── Module.cpp │ │ │ │ ├── OpenReg.h │ │ │ │ ├── OpenRegHooks.cpp │ │ │ │ └── OpenRegMem.cpp │ │ ├── setup.py │ │ └── test │ │ │ └── test_openreg.py │ ├── python_agnostic_extension │ │ ├── python_agnostic │ │ │ ├── __init__.py │ │ │ ├── csrc │ │ │ │ └── ultra_norm.cu │ │ │ └── ops.py │ │ └── setup.py │ ├── rng_extension.cpp │ ├── self_compiler_include_dirs_test │ │ └── tmp.h │ ├── setup.py │ ├── torch_library.cu │ ├── torch_test_cpp_extension │ │ └── __init__.py │ └── xpu_extension.sycl ├── create_dummy_torchscript_model.py ├── custom_backend │ ├── CMakeLists.txt │ ├── backend.py │ ├── custom_backend.cpp │ ├── custom_backend.h │ ├── test_custom_backend.cpp │ └── test_custom_backend.py ├── custom_operator │ ├── CMakeLists.txt │ ├── model.py │ ├── my_custom_ops.py │ ├── my_custom_ops2.py │ ├── op.cpp │ ├── op.h │ ├── pointwise.py │ ├── test_custom_ops.cpp │ ├── test_custom_ops.py │ └── test_infer_schema_annotation.py ├── delete.py ├── distributed │ ├── _composable │ │ ├── fsdp │ │ │ ├── test_fully_shard_autograd.py │ │ │ ├── test_fully_shard_clip_grad_norm_.py │ │ │ ├── test_fully_shard_comm.py │ │ │ ├── test_fully_shard_compile.py │ │ │ ├── test_fully_shard_extensions.py │ │ │ ├── test_fully_shard_frozen.py │ │ │ ├── test_fully_shard_grad_scaler.py │ │ │ ├── test_fully_shard_ignore_params.py │ │ │ ├── test_fully_shard_init.py │ │ │ ├── test_fully_shard_logging.py │ │ │ ├── test_fully_shard_memory.py │ │ │ ├── test_fully_shard_mixed_precision.py │ │ │ ├── test_fully_shard_overlap.py │ │ │ ├── test_fully_shard_state.py │ │ │ ├── test_fully_shard_state_dict.py │ │ │ └── test_fully_shard_training.py │ │ ├── test_checkpoint.py │ │ ├── test_composability │ │ │ ├── test_2d_composability.py │ │ │ └── test_pp_composability.py │ │ ├── test_contract.py │ │ ├── test_replicate.py │ │ └── test_replicate_with_compiler.py │ ├── _shard │ │ ├── sharded_optim │ │ │ └── test_sharded_optim.py │ │ ├── sharded_tensor │ │ │ ├── ops │ │ │ │ ├── test_binary_cmp.py │ │ │ │ ├── test_embedding.py │ │ │ │ ├── test_embedding_bag.py │ │ │ │ ├── test_init.py │ │ │ │ └── test_tensor_ops.py │ │ │ ├── test_logger.py │ │ │ ├── test_sharded_tensor.py │ │ │ └── test_sharded_tensor_reshard.py │ │ ├── sharding_plan │ │ │ └── test_sharding_plan.py │ │ ├── sharding_spec │ │ │ └── test_sharding_spec.py │ │ └── test_sharder.py │ ├── _tools │ │ ├── test_fsdp2_mem_tracker.py │ │ ├── test_mem_tracker.py │ │ ├── test_memory_tracker.py │ │ ├── test_mod_tracker.py │ │ ├── test_runtime_estimator.py │ │ ├── test_sac_estimator.py │ │ └── test_sac_ilp.py │ ├── algorithms │ │ ├── ddp_comm_hooks │ │ │ └── test_ddp_hooks.py │ │ ├── quantization │ │ │ └── test_quantization.py │ │ └── test_join.py │ ├── argparse_util_test.py │ ├── bin │ │ └── test_script.py │ ├── checkpoint │ │ ├── e2e │ │ │ ├── test_e2e_save_and_load.py │ │ │ ├── test_fine_tuning.py │ │ │ ├── test_fsdp_ep.py │ │ │ └── test_pipeline.py │ │ ├── fsdp │ │ │ └── test_fsdp_dsd.py │ │ ├── test_checkpoint.py │ │ ├── test_compatibility.py │ │ ├── test_dedup_tensors.py │ │ ├── test_dtensor_checkpoint.py │ │ ├── test_dtensor_resharding.py │ │ ├── test_file_system_checkpoint.py │ │ ├── test_file_system_checkpoint_cpu.py │ │ ├── test_format_utils.py │ │ ├── test_fsdp_model_state.py │ │ ├── test_fsdp_optim_state.py │ │ ├── test_fsdp_tp_checkpoint_conversion.py │ │ ├── test_fsspec.py │ │ ├── test_hf_storage.py │ │ ├── test_hsdp_checkpoint.py │ │ ├── test_nested_dict.py │ │ ├── test_planner.py │ │ ├── test_save_load_api.py │ │ ├── test_state_dict.py │ │ ├── test_state_dict_utils.py │ │ ├── test_tp_checkpoint.py │ │ ├── test_traverse.py │ │ └── test_utils.py │ ├── elastic │ │ ├── agent │ │ │ └── server │ │ │ │ └── test │ │ │ │ ├── __init__.py │ │ │ │ ├── api_test.py │ │ │ │ └── local_elastic_agent_test.py │ │ ├── events │ │ │ └── lib_test.py │ │ ├── metrics │ │ │ ├── __init__.py │ │ │ └── api_test.py │ │ ├── multiprocessing │ │ │ ├── api_test.py │ │ │ ├── bin │ │ │ │ ├── echo1.py │ │ │ │ ├── echo2.py │ │ │ │ ├── echo3.py │ │ │ │ ├── test_script.py │ │ │ │ └── zombie_test.py │ │ │ ├── errors │ │ │ │ ├── api_test.py │ │ │ │ └── error_handler_test.py │ │ │ ├── redirects_test.py │ │ │ └── tail_log_test.py │ │ ├── rendezvous │ │ │ ├── __init__.py │ │ │ ├── api_test.py │ │ │ ├── c10d_rendezvous_backend_test.py │ │ │ ├── dynamic_rendezvous_test.py │ │ │ ├── etcd_rendezvous_backend_test.py │ │ │ ├── etcd_rendezvous_test.py │ │ │ ├── etcd_server_test.py │ │ │ ├── out_of_tree_rendezvous_test.py │ │ │ ├── out_of_tree_test_package │ │ │ │ ├── pyproject.toml │ │ │ │ └── src │ │ │ │ │ └── testbackend │ │ │ │ │ └── __init__.py │ │ │ ├── rendezvous_backend_test.py │ │ │ ├── static_rendezvous_test.py │ │ │ └── utils_test.py │ │ ├── test_control_plane.py │ │ ├── timer │ │ │ ├── __init__.py │ │ │ ├── api_test.py │ │ │ ├── file_based_local_timer_test.py │ │ │ ├── local_timer_example.py │ │ │ └── local_timer_test.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── data │ │ │ ├── __init__.py │ │ │ └── cycling_iterator_test.py │ │ │ ├── distributed_test.py │ │ │ ├── logging_test.py │ │ │ └── util_test.py │ ├── flight_recorder │ │ └── test_fr_analysis.py │ ├── fsdp │ │ ├── test_checkpoint_wrapper.py │ │ ├── test_distributed_checkpoint.py │ │ ├── test_fsdp_apply.py │ │ ├── test_fsdp_backward_prefetch.py │ │ ├── test_fsdp_checkpoint.py │ │ ├── test_fsdp_clip_grad_norm.py │ │ ├── test_fsdp_comm.py │ │ ├── test_fsdp_comm_hooks.py │ │ ├── test_fsdp_core.py │ │ ├── test_fsdp_dtensor_state_dict.py │ │ ├── test_fsdp_exec_order.py │ │ ├── test_fsdp_fine_tune.py │ │ ├── test_fsdp_flatten_params.py │ │ ├── test_fsdp_freezing_weights.py │ │ ├── test_fsdp_fx.py │ │ ├── test_fsdp_grad_acc.py │ │ ├── test_fsdp_hybrid_shard.py │ │ ├── test_fsdp_ignored_modules.py │ │ ├── test_fsdp_input.py │ │ ├── test_fsdp_memory.py │ │ ├── test_fsdp_meta.py │ │ ├── test_fsdp_misc.py │ │ ├── test_fsdp_mixed_precision.py │ │ ├── test_fsdp_multiple_forward.py │ │ ├── test_fsdp_multiple_wrapping.py │ │ ├── test_fsdp_optim_state.py │ │ ├── test_fsdp_overlap.py │ │ ├── test_fsdp_pure_fp16.py │ │ ├── test_fsdp_sharded_grad_scaler.py │ │ ├── test_fsdp_state_dict.py │ │ ├── test_fsdp_tp_integration.py │ │ ├── test_fsdp_traversal.py │ │ ├── test_fsdp_uneven.py │ │ ├── test_fsdp_unshard_params.py │ │ ├── test_fsdp_use_orig_params.py │ │ ├── test_hsdp_dtensor_state_dict.py │ │ ├── test_shard_utils.py │ │ ├── test_utils.py │ │ └── test_wrap.py │ ├── launcher │ │ ├── __init__.py │ │ ├── api_test.py │ │ ├── bin │ │ │ ├── test_script.py │ │ │ ├── test_script.sh │ │ │ ├── test_script_init_method.py │ │ │ ├── test_script_is_torchelastic_launched.py │ │ │ └── test_script_local_rank.py │ │ ├── launch_test.py │ │ └── test_run.py │ ├── nn │ │ └── jit │ │ │ ├── __init__.py │ │ │ └── test_instantiator.py │ ├── optim │ │ ├── test_apply_optimizer_in_backward.py │ │ ├── test_named_optimizer.py │ │ └── test_zero_redundancy_optimizer.py │ ├── pipelining │ │ ├── __init__.py │ │ ├── artifacts │ │ │ ├── zb1p_2rank_2stagep_comms.csv │ │ │ └── zb1p_2rank_2stagep_compute.csv │ │ ├── model_registry.py │ │ ├── schedule_registry.py │ │ ├── test_backward.py │ │ ├── test_microbatch.py │ │ ├── test_pipe.py │ │ ├── test_schedule.py │ │ ├── test_schedule_multiproc.py │ │ ├── test_stage.py │ │ ├── test_transformer.py │ │ └── test_unflatten.py │ ├── rpc │ │ ├── cuda │ │ │ └── test_tensorpipe_agent.py │ │ ├── test_faulty_agent.py │ │ ├── test_share_memory.py │ │ └── test_tensorpipe_agent.py │ ├── tensor │ │ ├── README.md │ │ ├── __init__.py │ │ ├── debug │ │ │ ├── test_comm_mode.py │ │ │ ├── test_comm_mode_features.py │ │ │ └── test_op_coverage.py │ │ ├── experimental │ │ │ ├── test_local_map.py │ │ │ ├── test_register_sharding.py │ │ │ └── test_tp_transform.py │ │ ├── parallel │ │ │ ├── __init__.py │ │ │ ├── test_micro_pipeline_tp.py │ │ │ ├── test_parallelize_api.py │ │ │ ├── test_tp_examples.py │ │ │ ├── test_tp_random_state.py │ │ │ └── test_tp_style.py │ │ ├── test_api.py │ │ ├── test_attention.py │ │ ├── test_common_rules.py │ │ ├── test_convolution_ops.py │ │ ├── test_dtensor.py │ │ ├── test_dtensor_compile.py │ │ ├── test_dtensor_ops.py │ │ ├── test_embedding_ops.py │ │ ├── test_experimental_ops.py │ │ ├── test_init.py │ │ ├── test_math_ops.py │ │ ├── test_matrix_ops.py │ │ ├── test_op_strategy.py │ │ ├── test_optimizers.py │ │ ├── test_pointwise_ops.py │ │ ├── test_random_ops.py │ │ ├── test_redistribute.py │ │ ├── test_tensor_ops.py │ │ ├── test_utils.py │ │ ├── test_view_ops.py │ │ └── test_xla_integration.py │ ├── test_backends.py │ ├── test_c10d_common.py │ ├── test_c10d_functional_native.py │ ├── test_c10d_gloo.py │ ├── test_c10d_logger.py │ ├── test_c10d_nccl.py │ ├── test_c10d_object_collectives.py │ ├── test_c10d_ops_nccl.py │ ├── test_c10d_pypg.py │ ├── test_c10d_spawn.py │ ├── test_c10d_spawn_gloo.py │ ├── test_c10d_spawn_nccl.py │ ├── test_c10d_spawn_ucc.py │ ├── test_c10d_ucc.py │ ├── test_collective_utils.py │ ├── test_composability.py │ ├── test_compute_comm_reordering.py │ ├── test_control_collectives.py │ ├── test_data_parallel.py │ ├── test_device_mesh.py │ ├── test_distributed_spawn.py │ ├── test_dynamo_distributed.py │ ├── test_fake_pg.py │ ├── test_functional_api.py │ ├── test_inductor_collectives.py │ ├── test_launcher.py │ ├── test_multi_threaded_pg.py │ ├── test_nccl.py │ ├── test_pg_wrapper.py │ ├── test_serialization.py │ ├── test_store.py │ └── test_symmetric_memory.py ├── distributions │ ├── test_constraints.py │ ├── test_distributions.py │ ├── test_transforms.py │ └── test_utils.py ├── dynamo │ ├── __init__.py │ ├── mock_modules │ │ ├── __init__.py │ │ ├── mock_module1.py │ │ ├── mock_module2.py │ │ └── mock_module3.py │ ├── mock_store_global_crossfile_inline.py │ ├── test_activation_checkpointing.py │ ├── test_after_aot.py │ ├── test_aot_autograd.py │ ├── test_aot_autograd_cache.py │ ├── test_autograd_function.py │ ├── test_backends.py │ ├── test_backward_higher_order_ops.py │ ├── test_base_hop.py │ ├── test_base_output.py │ ├── test_bytecode_utils.py │ ├── test_callback.py │ ├── test_compile.py │ ├── test_compiler_bisector.py │ ├── test_comptime.py │ ├── test_config.py │ ├── test_ctx_manager.py │ ├── test_cudagraphs.py │ ├── test_cudagraphs_expandable_segments.py │ ├── test_debug_utils.py │ ├── test_decorators.py │ ├── test_deviceguard.py │ ├── test_dicts.py │ ├── test_dynamic_shapes.py │ ├── test_exc.py │ ├── test_exceptions.py │ ├── test_export.py │ ├── test_export_mutations.py │ ├── test_flat_apply.py │ ├── test_frame_init.py │ ├── test_functions.py │ ├── test_fx_passes_pre_grad.py │ ├── test_generator.py │ ├── test_global.py │ ├── test_graph_break_messages.py │ ├── test_graph_deduplication.py │ ├── test_graph_region_tracker.py │ ├── test_guard_manager.py │ ├── test_higher_order_ops.py │ ├── test_hooks.py │ ├── test_input_attr_tracking.py │ ├── test_interop.py │ ├── test_logging.py │ ├── test_metrics_context.py │ ├── test_minifier.py │ ├── test_misc.py │ ├── test_model_output.py │ ├── test_modes.py │ ├── test_modules.py │ ├── test_nops.py │ ├── test_optimizers.py │ ├── test_pgo.py │ ├── test_pre_dispatch.py │ ├── test_profiler.py │ ├── test_python_autograd.py │ ├── test_python_dispatcher.py │ ├── test_recompile_ux.py │ ├── test_recompiles.py │ ├── test_reconstruct.py │ ├── test_reorder_logs.py │ ├── test_repros.py │ ├── test_resume.py │ ├── test_sdpa.py │ ├── test_skip_guard_eval_unsafe.py │ ├── test_skip_non_tensor.py │ ├── test_sources.py │ ├── test_structured_trace.py │ ├── test_subclasses.py │ ├── test_subgraphs.py │ ├── test_sys.py │ ├── test_torchrec.py │ ├── test_trace_rules.py │ ├── test_unspec.py │ ├── test_utils.py │ ├── test_verify_correctness.py │ ├── test_view.py │ └── utils.py ├── dynamo_expected_failures │ ├── ActivationCheckpointingTests.test_cond_with_kwargs │ ├── AotAutogradFallbackTests.test_aot_sequence_nr │ ├── AutogradFunctionTests.test_graph_break_if_lifted_free_variable │ ├── AutogradFunctionTests.test_print_in_bwd │ ├── AutogradFunctionTests.test_stride_in_bwd │ ├── ComptimeTests.test_graph_break │ ├── CtxManagerTests.test_autograd_profiler_enabled │ ├── CtxManagerTests.test_disable_saved_tensors_hooks │ ├── CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled │ ├── CtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled_nested │ ├── DecoratorTests.test_mark_static_address_guarded │ ├── DecoratorTests.test_mark_static_address_unguarded │ ├── DefaultsTests.test_cast_tensor_single_elem │ ├── DefaultsTests.test_compare_constant_and_tensor │ ├── DefaultsTests.test_func_default_torch_args │ ├── DefaultsTests.test_in_set_inplace │ ├── DefaultsTests.test_in_set_would_fail_broadcast │ ├── DynamoProfilerTests.test_dynamo_timed_profiling_isolated │ ├── DynamoProfilerTests.test_profiler_cache_lookup │ ├── DynamoProfilerTests.test_profiler_dynamo_compiled_region │ ├── End2EndTests.test_init_group │ ├── ExampleTests.test_exportdb_supported_case_assume_constant_result │ ├── ExampleTests.test_exportdb_supported_case_constrain_as_size_example │ ├── ExcTests.test_backend_suppress_line │ ├── ExcTests.test_graph_break_log │ ├── ExcTests.test_internal_error_no_suppress │ ├── ExcTests.test_internal_error_suppress_errors │ ├── ExcTests.test_not_implemented_error │ ├── ExcTests.test_trigger_on_error │ ├── ExportTests.test_cond_raise_user_error_on_non_list_operands │ ├── ExportTests.test_cond_raise_user_error_on_non_tensor_operands │ ├── ExportTests.test_cond_raise_user_error_on_unsupported_pred │ ├── ExportTests.test_export_decomp │ ├── ExportTests.test_export_dynamic_dim_cleanup │ ├── ExportTests.test_export_dynamic_dim_range_constraint │ ├── ExportTests.test_export_mark_dynamic_conflict_dynamic_dim │ ├── ExportTests.test_export_multi_dynamic_dim_constraint │ ├── ExportTests.test_export_multi_dynamic_dim_unsafe_relationship │ ├── ExportTests.test_export_no_raise │ ├── ExportTests.test_export_preserve_constraints_as_metadata_scalar │ ├── ExportTests.test_export_raise_on_relationship │ ├── ExportTests.test_export_with_constant_dict_values │ ├── ExportTests.test_export_with_constant_free_function │ ├── ExportTests.test_export_with_constant_free_function_and_class_method │ ├── ExportTests.test_export_with_constant_free_function_and_class_method_multiarg │ ├── ExportTests.test_export_with_constant_free_function_and_class_method_multiarg_diff │ ├── ExportTests.test_export_with_constant_list_nonzero │ ├── ExportTests.test_export_with_constant_list_nonzero_free_function │ ├── ExportTests.test_export_with_constant_method_on_module │ ├── ExportTests.test_export_with_constant_method_on_module_invoke_twice │ ├── ExportTests.test_export_with_constant_none_control_flow │ ├── ExportTests.test_export_with_constant_none_control_flow_free_func │ ├── ExportTests.test_export_with_constant_not_none_control_flow │ ├── ExportTests.test_export_with_constant_not_none_control_flow_free_func │ ├── ExportTests.test_export_with_constant_not_none_control_flow_pos │ ├── ExportTests.test_export_with_constant_not_return_const │ ├── ExportTests.test_export_with_constant_tuple_nonzero │ ├── ExportTests.test_export_with_map_cond │ ├── ExportTests.test_exported_graph_serialization │ ├── ExportTests.test_map_cond_param_buffer_lifted │ ├── ExportTests.test_torch_inference_mode_ctx │ ├── ExportTests.test_untracked_inputs_in_constraints │ ├── FuncTorchHigherOrderOpTests.test_grad │ ├── FuncTorchHigherOrderOpTests.test_grad_disable_capture │ ├── FuncTorchHigherOrderOpTests.test_grad_fn_with_kwargs │ ├── FuncTorchHigherOrderOpTests.test_grad_freevar_python_scalar │ ├── FuncTorchHigherOrderOpTests.test_grad_has_aux │ ├── FuncTorchHigherOrderOpTests.test_grad_non_tensor_input │ ├── FuncTorchHigherOrderOpTests.test_grad_pytree │ ├── FuncTorchHigherOrderOpTests.test_grad_two_tensor_all_grad_has_aux │ ├── FuncTorchHigherOrderOpTests.test_grad_two_tensor_has_aux │ ├── FuncTorchHigherOrderOpTests.test_grad_with_graph_break │ ├── FuncTorchHigherOrderOpTests.test_grad_with_side_effect │ ├── FuncTorchHigherOrderOpTests.test_vmap │ ├── FuncTorchHigherOrderOpTests.test_vmap_disable_capture │ ├── FuncTorchHigherOrderOpTests.test_vmap_free_const │ ├── FuncTorchHigherOrderOpTests.test_vmap_free_tensor │ ├── FuncTorchHigherOrderOpTests.test_vmap_kwargs │ ├── FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_in_dims │ ├── FuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_out_dims │ ├── FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs │ ├── FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_diff_dims │ ├── FuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_out_dims_tuple │ ├── FuncTorchHigherOrderOpTests.test_vmap_over_vmap_captured │ ├── FuncTorchHigherOrderOpTests.test_vmap_over_vmap_two_inputs │ ├── FuncTorchHigherOrderOpTests.test_vmap_previous_illegal_op_no_graph_break │ ├── FuncTorchHigherOrderOpTests.test_vmap_pytree_inputs │ ├── FuncTorchHigherOrderOpTests.test_vmap_recompile_same_config │ ├── FuncTorchHigherOrderOpTests.test_vmap_recompile_with_randomness │ ├── FuncTorchHigherOrderOpTests.test_vmap_side_effects │ ├── FuncTorchHigherOrderOpTests.test_vmap_two_inputs │ ├── FuncTorchHigherOrderOpTests.test_vmap_two_inputs_tuple_in_dims │ ├── FuncTorchHigherOrderOpTests.test_vmap_with_conditional_graph_break │ ├── FuncTorchHigherOrderOpTests.test_vmap_with_graph_break │ ├── FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_2 │ ├── FuncTorchHigherOrderOpTests.test_vmap_with_graph_break_lambda │ ├── FunctionTests.test_default_dict │ ├── FunctionTests.test_default_dict_closure │ ├── FunctionTests.test_default_dict_lambda │ ├── FunctionTests.test_is_contiguous_frame_counts │ ├── FunctionTests.test_math_radians │ ├── FunctionTests.test_partials_as_input_partials_lambda │ ├── FunctionTests.test_partials_as_input_partials_mod │ ├── HigherOrderOpTests.test_capture_untracked_global_nested │ ├── HigherOrderOpTests.test_capture_value_created_in_subgraph │ ├── HigherOrderOpTests.test_cond_branches_no_arguments │ ├── HigherOrderOpTests.test_cond_branches_no_arguments_no_closure │ ├── HigherOrderOpTests.test_cond_pytree_operands │ ├── HigherOrderOpTests.test_cond_pytree_operands_with_non_tensor_leaves │ ├── HigherOrderOpTests.test_cond_source_fn_stack │ ├── HigherOrderOpTests.test_cond_subgraph_name_is_valid │ ├── HigherOrderOpTests.test_fallback_on_python_primitives_output │ ├── HigherOrderOpTests.test_map_lowers_to_graph │ ├── HigherOrderOpTests.test_map_multi_return │ ├── HigherOrderOpTests.test_map_pytree_return │ ├── HigherOrderOpTests.test_map_source_fn_stack │ ├── HigherOrderOpTests.test_map_subgraph_name_is_valid │ ├── HigherOrderOpTests.test_map_symint_input │ ├── HigherOrderOpTests.test_modules │ ├── HigherOrderOpTests.test_nested_tuple_output │ ├── HigherOrderOpTests.test_output_with_dict │ ├── HigherOrderOpTests.test_side_effect_in_body │ ├── HigherOrderOpTests.test_vmap_source_fn_stack │ ├── HigherOrderOpTests.test_wrap_kwarg_recompile │ ├── HigherOrderOpTests.test_wrap_subgraph_name_is_valid │ ├── HooksTests.test_functools_arg_vary │ ├── HooksTests.test_post_acc_grad_hook │ ├── HooksTests.test_tensor_register_global_hook │ ├── HooksTests.test_tensor_register_hook_in_graph_local │ ├── HooksTests.test_tensor_register_multiple_hooks │ ├── HooksTests.test_tensor_register_multiple_hooks_handles_in_list │ ├── InteropTests.test_vmap_in_graph │ ├── LoggingTests.test_aot │ ├── LoggingTests.test_aot_graphs │ ├── LoggingTests.test_aot_joint_graph │ ├── LoggingTests.test_bytecode │ ├── LoggingTests.test_custom_format │ ├── LoggingTests.test_custom_format_exc │ ├── LoggingTests.test_dynamo_debug │ ├── LoggingTests.test_dynamo_error │ ├── LoggingTests.test_dynamo_info │ ├── LoggingTests.test_graph │ ├── LoggingTests.test_graph_breaks │ ├── LoggingTests.test_graph_code │ ├── LoggingTests.test_graph_sizes │ ├── LoggingTests.test_graph_sizes_dynamic │ ├── LoggingTests.test_guards_recompiles │ ├── LoggingTests.test_inductor_debug │ ├── LoggingTests.test_inductor_error │ ├── LoggingTests.test_inductor_info │ ├── LoggingTests.test_multiline_format │ ├── LoggingTests.test_output_code │ ├── LoggingTests.test_recompiles │ ├── LoggingTests.test_trace_source_cond │ ├── LoggingTests.test_trace_source_nested │ ├── MiscTests.test_add_to_set │ ├── MiscTests.test_any_all_symnode │ ├── MiscTests.test_backend_match_guard │ ├── MiscTests.test_boolarg │ ├── MiscTests.test_callpacked │ ├── MiscTests.test_compare_shapes_with_constant │ ├── MiscTests.test_cond_nested │ ├── MiscTests.test_cond_side_effects │ ├── MiscTests.test_dataclass_fields │ ├── MiscTests.test_deque_append_left │ ├── MiscTests.test_deque_input │ ├── MiscTests.test_deterministic_algorithms_mutated │ ├── MiscTests.test_dict_order_keys │ ├── MiscTests.test_dict_order_keys_modules │ ├── MiscTests.test_dict_order_keys_tensors │ ├── MiscTests.test_dictcomp │ ├── MiscTests.test_dtypes_no_graphbreaks │ ├── MiscTests.test_dunder_new_function_inlining │ ├── MiscTests.test_frozenset_torch_func_contains │ ├── MiscTests.test_get_cache_entry │ ├── MiscTests.test_guard_failure_fn │ ├── MiscTests.test_guard_failure_fn2 │ ├── MiscTests.test_guard_failure_fn_shape_control │ ├── MiscTests.test_guard_failure_fn_tensor_iter │ ├── MiscTests.test_inference_mode │ ├── MiscTests.test_inplace_param_update │ ├── MiscTests.test_inplace_view_on_graph_input │ ├── MiscTests.test_intermediary_tensor_grad_access │ ├── MiscTests.test_is_compiling │ ├── MiscTests.test_iter_set │ ├── MiscTests.test_itertools_accumulate_symint_default_sum │ ├── MiscTests.test_itertools_accumulate_tensors_default_sum │ ├── MiscTests.test_itertools_groupby_pure_python_default_identify_func │ ├── MiscTests.test_itertools_groupby_pure_python_key_func │ ├── MiscTests.test_itertools_infinite_cycle │ ├── MiscTests.test_itertools_infinite_repeat │ ├── MiscTests.test_itertools_infinite_repeat_mutation │ ├── MiscTests.test_itertools_repeat │ ├── MiscTests.test_listcomp │ ├── MiscTests.test_mandelbrot_numpy │ ├── MiscTests.test_mark_static │ ├── MiscTests.test_namedtuple1 │ ├── MiscTests.test_namedtuple2 │ ├── MiscTests.test_nan │ ├── MiscTests.test_nested_optimize │ ├── MiscTests.test_nested_optimize_decorator │ ├── MiscTests.test_nested_optimize_run │ ├── MiscTests.test_nn_module_getattr │ ├── MiscTests.test_no_raise_guard_partial_constraint │ ├── MiscTests.test_no_raise_guard_partial_constraint_across_break │ ├── MiscTests.test_numpy_array_of_arrays │ ├── MiscTests.test_numpy_force │ ├── MiscTests.test_numpy_int_constant │ ├── MiscTests.test_numpy_iter │ ├── MiscTests.test_numpy_non_torch_dtype │ ├── MiscTests.test_numpy_readonly │ ├── MiscTests.test_numpy_recompilation_scalar │ ├── MiscTests.test_numpy_size_attr │ ├── MiscTests.test_numpy_subdtype │ ├── MiscTests.test_numpy_tolist │ ├── MiscTests.test_numpy_torch_operators │ ├── MiscTests.test_numpy_with_builtin_type │ ├── MiscTests.test_out_variants_with_resizing_on_graph_inputs │ ├── MiscTests.test_pure_python_accumulate │ ├── MiscTests.test_py_guards_mark_dynamic │ ├── MiscTests.test_raise_guard_full_constraint │ ├── MiscTests.test_raise_guard_partial_constraint_no_graph_break │ ├── MiscTests.test_raise_on_backend_error │ ├── MiscTests.test_recompile_on_global_state_change │ ├── MiscTests.test_release_input_memory │ ├── MiscTests.test_release_module_memory │ ├── MiscTests.test_release_scope_memory │ ├── MiscTests.test_repeat_interleave_graphbreaks │ ├── MiscTests.test_return_nested_function │ ├── MiscTests.test_set_aliasing_recompiles │ ├── MiscTests.test_setattr_mutation1 │ ├── MiscTests.test_simple_set_usage │ ├── MiscTests.test_size_input │ ├── MiscTests.test_str_format_assert2 │ ├── MiscTests.test_tensor_build_list_unpack │ ├── MiscTests.test_tensor_dict1 │ ├── MiscTests.test_tensor_dict2 │ ├── MiscTests.test_tensor_dict3 │ ├── MiscTests.test_tensor_item_capture │ ├── MiscTests.test_tolist_0d │ ├── MiscTests.test_tolist_1d │ ├── MiscTests.test_tolist_kd │ ├── MiscTests.test_tolist_kd_dynamic │ ├── MiscTests.test_tolist_scalar │ ├── MiscTests.test_torch_seed │ ├── MiscTests.test_tracing_nested_py_tree │ ├── MiscTests.test_tracing_nested_py_tree_dicts │ ├── MiscTests.test_tracing_nested_py_tree_mixed_all │ ├── MiscTests.test_tracing_nested_py_tree_tuples │ ├── MiscTests.test_tracing_py_tree │ ├── MiscTests.test_tracing_py_tree_tensor_subclass │ ├── MiscTests.test_tracing_tree_map_only │ ├── MiscTests.test_type_copy │ ├── MiscTests.test_typing_typevar │ ├── MiscTests.test_user_getattribute │ ├── MiscTests.test_with_builtin_type │ ├── MiscTests.test_yield_from │ ├── MiscTests.test_yield_gen_and_from │ ├── MiscTests.test_yield_send_to_subgenerator_graph_break │ ├── NNModuleTests.test_lazy_module1 │ ├── NNModuleTests.test_lazy_module2 │ ├── NNModuleTests.test_lazy_module4 │ ├── NNModuleTests.test_lazy_module5 │ ├── NNModuleTests.test_lazy_module6 │ ├── NNModuleTests.test_lazy_module_no_cls_to_become │ ├── NNModuleTests.test_self_mutating1 │ ├── NNModuleTests.test_unsupportedmethod │ ├── NNModuleTests.test_unsupportedmodule │ ├── NonStrictExportTestExport.test__scaled_dot_product_flash_attention_non_strict │ ├── NonStrictExportTestExport.test_basic_non_strict_fake_tensor_non_strict │ ├── NonStrictExportTestExport.test_basic_non_strict_real_tensor_non_strict │ ├── NonStrictExportTestExport.test_buffer_util_non_strict │ ├── NonStrictExportTestExport.test_cond_with_module_stack_export_with_non_strict │ ├── NonStrictExportTestExport.test_export_decomps_dynamic_non_strict │ ├── NonStrictExportTestExport.test_export_decomps_simple_non_strict │ ├── NonStrictExportTestExport.test_export_with_wrong_inputs_non_strict │ ├── NonStrictExportTestExport.test_external_call_non_strict_real_tensor_non_strict │ ├── NonStrictExportTestExport.test_fqn_non_strict │ ├── NonStrictExportTestExport.test_nn_module_stack_non_strict │ ├── NonStrictExportTestExport.test_nn_module_stack_shared_submodule_non_strict │ ├── NonStrictExportTestExport.test_non_strict_dynamic_shapes_non_strict │ ├── NonStrictExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_non_strict │ ├── NonStrictExportTestExport.test_param_util_non_strict │ ├── NonStrictExportTestExport.test_raise_user_error_when_guard_on_data_dependent_operation_non_strict │ ├── NonStrictExportTestExport.test_sym_sqrt_non_strict │ ├── NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_non_strict │ ├── NonStrictExportTestExport.test_to_module_with_mutated_buffer_multiple_update_sub_later_non_strict │ ├── NonStrictExportTestExport.test_to_module_with_mutated_buffer_non_strict │ ├── NumpyTestsCPU.test_boolean_indexing_weirdness_cpu │ ├── NumpyTestsCPU.test_boolean_shape_mismatch_cpu │ ├── NumpyTestsCPU.test_empty_fancy_index_cpu │ ├── NumpyTestsCPU.test_index_no_floats_cpu │ ├── OptimizedModuleTest.test_backward_hooks │ ├── OptimizedModuleTest.test_cache_size_limit_on_guarded_nn_modules │ ├── OptimizedModuleTest.test_composition_with_opt_mod │ ├── OptimizedModuleTest.test_hooks_inner │ ├── OptimizedModuleTest.test_hooks_outer │ ├── OptimizedModuleTest.test_hooks_skip_guards │ ├── OptimizedModuleTest.test_module_dict_iter_name │ ├── OptimizedModuleTest.test_nn_module │ ├── OptimizedModuleTest.test_no_recompile_on_nn_guarded_modules │ ├── OptimizedModuleTest.test_to │ ├── OptimizerTests.test_adagrad │ ├── OptimizerTests.test_adam │ ├── OptimizerTests.test_adamax │ ├── OptimizerTests.test_adamw │ ├── OptimizerTests.test_asgd │ ├── OptimizerTests.test_nadam │ ├── OptimizerTests.test_rmsprop │ ├── OptimizerTests.test_rprop │ ├── OptimizerTests.test_sgd │ ├── PackedSequenceTest.test_pack_sequence │ ├── PackedSequenceTest.test_to │ ├── PackedSequenceTest.test_total_length │ ├── PackedSequenceTest.test_type_casts │ ├── PackedSequenceTest.test_unpack_sequence │ ├── RecompileTests.test_aliasing_guard_failures_with_globals │ ├── RecompileTests.test_automatic_dynamic_reduce_recompiles │ ├── RecompileTests.test_automatic_dynamic_tensor_scalar_change │ ├── RecompileTests.test_dynamic_shape_parameter_recompile │ ├── RecompileTests.test_recompiles_true_false_flop │ ├── RecompileUxTests.test_mismatched_type │ ├── RecompileUxTests.test_verbose_tensor_check │ ├── ReproTests.test_add_sub_alpha_out │ ├── ReproTests.test_addr_alpha_beta_out │ ├── ReproTests.test_boxes_len │ ├── ReproTests.test_chunk_reformer_ff │ ├── ReproTests.test_convert_boxes_to_pooler_format │ ├── ReproTests.test_create_rand_mask_from_inputs │ ├── ReproTests.test_do_paste_mask │ ├── ReproTests.test_dynamic_shapes_float_guard │ ├── ReproTests.test_dynamic_shapes_implicit_guard │ ├── ReproTests.test_empty_list_contains_with_jump │ ├── ReproTests.test_function_in_skipfiles │ ├── ReproTests.test_functools_wraps │ ├── ReproTests.test_hf_t5_forward │ ├── ReproTests.test_hf_xsoftmax_training │ ├── ReproTests.test_issue175 │ ├── ReproTests.test_jit_trace_errors │ ├── ReproTests.test_list_aliasing │ ├── ReproTests.test_list_self_reference │ ├── ReproTests.test_longformer_chunk │ ├── ReproTests.test_merge_criteria_processor_list1 │ ├── ReproTests.test_merge_criteria_processor_list2 │ ├── ReproTests.test_module_in_skipfiles │ ├── ReproTests.test_multi_import │ ├── ReproTests.test_negative_shape_guard │ ├── ReproTests.test_numpy_not_ndarray_recompiles │ ├── ReproTests.test_recursive_map │ ├── ReproTests.test_reformer_eval │ ├── ReproTests.test_reformer_min_chunk_len │ ├── ReproTests.test_reformer_sorting │ ├── ReproTests.test_relative_import │ ├── ReproTests.test_relative_import_no_modulename │ ├── ReproTests.test_restricted_list_subclass1 │ ├── ReproTests.test_rewrite_assert_noop │ ├── ReproTests.test_rewrite_assert_with_msg │ ├── ReproTests.test_rewrite_assert_with_non_string_msg │ ├── ReproTests.test_seq_append_list │ ├── ReproTests.test_size_typematch │ ├── ReproTests.test_tensor_data_kwarg │ ├── ReproTests.test_threading_local │ ├── ReproTests.test_validate_model_kwargs │ ├── RetraceExportTestDynamismExpression.test_export_inline_constraints_retraceability │ ├── RetraceExportTestExport.test_cond_with_module_stack_export_with_retraceability │ ├── RetraceExportTestExport.test_constrain_size_in_eager_retraceability │ ├── RetraceExportTestExport.test_constrain_size_with_constrain_value_retraceability │ ├── RetraceExportTestExport.test_constrain_size_with_various_cases_retraceability │ ├── RetraceExportTestExport.test_nn_module_stack_retraceability │ ├── RetraceExportTestExport.test_nn_module_stack_shared_submodule_retraceability │ ├── RetraceExportTestExport.test_non_strict_dynamic_shapes_retraceability │ ├── RetraceExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_retraceability │ ├── SerDesExportTestDynamismExpression.test_export_inline_constraints_serdes │ ├── SerDesExportTestExport.test_basic_non_strict_fake_tensor_serdes │ ├── SerDesExportTestExport.test_basic_non_strict_real_tensor_serdes │ ├── SerDesExportTestExport.test_cond_with_module_stack_export_with_serdes │ ├── SerDesExportTestExport.test_constrain_size_in_eager_serdes │ ├── SerDesExportTestExport.test_constrain_size_with_constrain_value_serdes │ ├── SerDesExportTestExport.test_constrain_size_with_various_cases_serdes │ ├── SerDesExportTestExport.test_external_call_non_strict_real_tensor_serdes │ ├── SerDesExportTestExport.test_nn_module_stack_serdes │ ├── SerDesExportTestExport.test_nn_module_stack_shared_submodule_serdes │ ├── SerDesExportTestExport.test_non_strict_dynamic_shapes_serdes │ ├── SerDesExportTestExport.test_non_strict_dynamic_shapes_suggested_fixes_serdes │ ├── SkipNonTensorTests.test_add_tensor1 │ ├── SkipNonTensorTests.test_add_tensor2 │ ├── SkipNonTensorTests.test_add_tensor_dict │ ├── SkipNonTensorTests.test_add_tensor_list │ ├── SkipNonTensorTests.test_do_not_skip_side_effects │ ├── SkipNonTensorTests.test_recursive_list │ ├── SubGraphTests.test_dynamic_duck_size │ ├── SubGraphTests.test_dynamic_getitem │ ├── SubGraphTests.test_dynamic_kwarg │ ├── SubGraphTests.test_dynamic_order_dependence │ ├── SubGraphTests.test_dynamic_zero_inference │ ├── SubGraphTests.test_enumerate_not_break_graph │ ├── SubGraphTests.test_no_graph_break_on_item │ ├── SubclassTests.test_compile_higher_order_with_functionalization │ ├── SubclassTests.test_compile_with_functionalization │ ├── SubclassTests.test_torch_function_state_guards │ ├── SubclassTests.test_wrapper_subclass_guards_on_inner_tensor │ ├── SubclassTests.test_wrapper_subclass_with_differently_sized_inner_tensor │ ├── SubclassTests.test_wrapper_subclass_with_same_sized_inner_tensor │ ├── TestAOTAutograd.test_input_mutation_aliases_bases_out_of_order │ ├── TestAOTAutograd.test_input_mutation_false_aliasing │ ├── TestAOTDispatch.test_aot_dispatch_input_mutation │ ├── TestAOTDispatch.test_aot_dispatch_input_mutation_and_output_alias │ ├── TestAOTDispatch.test_aot_dispatch_output_alias │ ├── TestAOTDispatch.test_aot_dispatch_simple │ ├── TestAOTModuleSimplified.test_aot_module_simplified_dynamic │ ├── TestAnalyze.test_trace_dependencies │ ├── TestApplyAlongAxis.test_0d_array │ ├── TestApplyAlongAxis.test_axis_insertion │ ├── TestApplyAlongAxis.test_scalar_array │ ├── TestArange.test_infinite │ ├── TestArange.test_nan_step │ ├── TestArange.test_require_range │ ├── TestArange.test_zero_step │ ├── TestArgmax.test_combinations_data62 │ ├── TestArgmax.test_combinations_data63 │ ├── TestArgmax.test_combinations_data65 │ ├── TestArgmax.test_combinations_data66 │ ├── TestArrayAttributeDeletion.test_multiarray_not_writable_attributes_deletion │ ├── TestArrayAttributeDeletion.test_multiarray_writable_attributes_deletion │ ├── TestArrayCreationCopyArgument.test_scalars │ ├── TestArrayCreationCopyArgument.test_striding_not_ok │ ├── TestArrayFromScalar.test_integers_np_byte_np_longlong │ ├── TestArrayFromScalar.test_integers_np_int__np_longlong │ ├── TestArrayFromScalar.test_integers_np_intc_np_longlong │ ├── TestArrayFromScalar.test_integers_np_longlong_np_longlong │ ├── TestArrayFromScalar.test_integers_np_longlong_t26 │ ├── TestArrayFromScalar.test_integers_np_short_np_longlong │ ├── TestArrayFromScalar.test_integers_t15_np_longlong │ ├── TestArraySplit.test_integer_0_split │ ├── TestAssignment.test_assignment_broadcasting │ ├── TestAttributes.test_fill_max_uint64 │ ├── TestAttributes.test_fill_readonly │ ├── TestAttributes.test_fill_struct_array │ ├── TestAutograd.test_access_saved_tensor_twice_without_recomputation_works │ ├── TestAutograd.test_anomaly_mode_no_check_nan │ ├── TestAutograd.test_autograd_simple_views_python │ ├── TestAutograd.test_autograd_views_codegen │ ├── TestAutograd.test_backward_with_inputs │ ├── TestAutograd.test_checkpoint_detects_non_determinism │ ├── TestAutograd.test_checkpointing_non_reentrant_autocast_cpu │ ├── TestAutograd.test_checkpointing_without_reentrant_custom_function_works │ ├── TestAutograd.test_custom_function_forward_mode_inplace_checks │ ├── TestAutograd.test_custom_function_forward_mode_view_checks │ ├── TestAutograd.test_custom_function_saved_tensors │ ├── TestAutograd.test_gradcheck_forward_ad │ ├── TestAutograd.test_gradcheck_nondeterministic │ ├── TestAutograd.test_nested_anomaly_detect_nan │ ├── TestAutograd.test_nested_anomaly_printstack_cleanup │ ├── TestAutograd.test_record_function │ ├── TestAutograd.test_return_duplicate │ ├── TestAutograd.test_return_duplicate_inplace │ ├── TestAutograd.test_save_on_cpu_and_checkpoint │ ├── TestAutograd.test_set_grad_coroutines │ ├── TestAutograd.test_set_grad_coroutines_exit │ ├── TestAutograd.test_set_grad_generator_functions │ ├── TestAutograd.test_set_grad_generator_functions_recursive │ ├── TestAutogradDeviceTypeCPU.test_inplace_on_view_modify_base_cpu │ ├── TestAutogradDeviceTypeCPU.test_inplace_on_view_python_cpu │ ├── TestAutogradForwardMode.test_create_new_zeros_with_same_meta │ ├── TestAutogradForwardMode.test_detach_view_tracking │ ├── TestAutogradForwardMode.test_forward_level_cleanup │ ├── TestAutogradForwardMode.test_make_dual_inference_tensor_in_inference_mode │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_False_save_for_jvp_save_tensors_input_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_False_save_for_jvp_save_tensors_output_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_False_save_for_vjp_save_tensors_input_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_False_save_for_vjp_save_tensors_output_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_True_save_for_jvp_save_tensors_input_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_True_save_for_jvp_save_tensors_output_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_True_save_for_vjp_save_tensors_input_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_function_returns_input_inner_requires_grad_True_save_for_vjp_save_tensors_output_mark_dirty_False_cpu │ ├── TestAutogradFunctionCPU.test_once_differentiable_autograd_vjp_cpu │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_False_save_for_jvp_save_tensors_input_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_False_save_for_jvp_save_tensors_output_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_False_save_for_vjp_save_tensors_input_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_False_save_for_vjp_save_tensors_output_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_True_save_for_jvp_save_tensors_input_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_True_save_for_jvp_save_tensors_output_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_True_save_for_vjp_save_tensors_input_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_function_returns_input_inner_requires_grad_True_save_for_vjp_save_tensors_output_mark_dirty_False_cuda │ ├── TestAutogradFunctionCUDA.test_once_differentiable_autograd_vjp_cuda │ ├── TestAutogradFunctionVmapAPICUDA.test_has_vmap_staticmethod_and_has_generate_vmap_rule_cuda │ ├── TestAutogradFunctionVmapAPICUDA.test_no_vmap_staticmethod_and_no_generate_vmap_rule_cuda │ ├── TestAutogradInferenceMode.test_inference_mode_decorator │ ├── TestAutogradInferenceMode.test_inference_mode_inf_tensor_in_inf_mode_inplace_op │ ├── TestBaseMath.test_lower_align │ ├── TestBaseSparsifier.test_state_dict │ ├── TestBitsCPU.test_cat_cpu │ ├── TestBool.test_sum_2 │ ├── TestBooleanIndexing.test_bool_as_int_argument_errors │ ├── TestBooleanIndexing.test_boolean_indexing_weirdness │ ├── TestBufferProtocolCPU.test_byte_to_int_cpu │ ├── TestClip.test_clip_func_takes_out │ ├── TestClip.test_simple_complex │ ├── TestColumnStack.test_non_iterable │ ├── TestComposabilityCPU.test_autograd_function_no_setup_context_transform_hessian_cpu │ ├── TestComposabilityCPU.test_autograd_function_no_setup_context_transform_jacfwd_cpu │ ├── TestComposabilityCPU.test_deprecation_transforms_transform_functionalize_cpu │ ├── TestComposabilityCPU.test_requires_grad_inside_transform_cpu │ ├── TestComposabilityCUDA.test_autograd_function_no_setup_context_transform_hessian_cuda │ ├── TestComposabilityCUDA.test_autograd_function_no_setup_context_transform_jacfwd_cuda │ ├── TestComposabilityCUDA.test_jvp_supports_saved_tensor_hooks_cuda │ ├── TestComposabilityCUDA.test_requires_grad_inside_transform_cuda │ ├── TestConcatenate.test_bad_out_shape │ ├── TestConcatenate.test_concatenate │ ├── TestConcatenate.test_exceptions │ ├── TestConcatenate.test_large_concatenate_axis_None │ ├── TestCond.test_empty_sq_cases │ ├── TestCond.test_sq_cases │ ├── TestContentStoreCPU.test_repeated_hash_cpu │ ├── TestControlFlow.test_map_autograd_nested_list │ ├── TestControlFlow.test_map_autograd_no_grad_output │ ├── TestControlFlow.test_map_dict_in_out │ ├── TestControlFlow.test_map_list_in_out │ ├── TestControlFlowTraced.test_map_functionalized │ ├── TestControlFlowTraced.test_map_functionalized_aot_func │ ├── TestControlFlowTraced.test_nested_cond_map_cond_symbolic │ ├── TestControlFlowTraced.test_nested_map_cond_real │ ├── TestControlFlowTraced.test_nested_map_cond_symbolic │ ├── TestControlFlowTraced.test_tracing_map_autograd_aot_functionalized │ ├── TestControlFlowTraced.test_tracing_map_autograd_symbolic_dict │ ├── TestControlFlowTraced.test_tracing_map_autograd_symbolic_list │ ├── TestControlFlowTraced.test_tracing_map_autograd_symbolic_simple │ ├── TestControlFlowTraced.test_tracing_map_real │ ├── TestControlFlowTraced.test_tracing_map_symbolic_dict │ ├── TestControlFlowTraced.test_tracing_map_symbolic_list │ ├── TestControlFlowTraced.test_tracing_map_symbolic_simple │ ├── TestConversion.test_to_int_scalar │ ├── TestCov.test_aweights │ ├── TestCov.test_fweights │ ├── TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attribute │ ├── TestCppExtensionJIT.test_cpp_frontend_module_has_up_to_date_attributes │ ├── TestCppExtensionOpenRgistration.test_open_device_registration │ ├── TestCross.test_broadcasting_shapes │ ├── TestCustomBackendAPI.test_aot_autograd_api │ ├── TestCustomBackendAPI.test_lookup_backend │ ├── TestCustomBackendAPI.test_register_backend_api │ ├── TestCustomOp.test_impl_device_cpu │ ├── TestCustomOp.test_legacy_define │ ├── TestDLPack.test_dlpack_device │ ├── TestDLPack.test_dtype_passthrough_dtype0 │ ├── TestDLPack.test_dtype_passthrough_dtype1 │ ├── TestDLPack.test_dtype_passthrough_dtype2 │ ├── TestDLPack.test_dtype_passthrough_dtype3 │ ├── TestDLPack.test_dtype_passthrough_dtype4 │ ├── TestDLPack.test_dtype_passthrough_dtype5 │ ├── TestDLPack.test_dtype_passthrough_dtype6 │ ├── TestDLPack.test_dtype_passthrough_dtype7 │ ├── TestDLPack.test_dtype_passthrough_dtype8 │ ├── TestDLPack.test_dtype_passthrough_dtype9 │ ├── TestDLPack.test_dunder_dlpack_refcount │ ├── TestDLPack.test_from_dlpack_refcount │ ├── TestDLPack.test_from_torch │ ├── TestDLPack.test_higher_dims_ndim_0 │ ├── TestDLPack.test_higher_dims_ndim_1 │ ├── TestDLPack.test_higher_dims_ndim_10 │ ├── TestDLPack.test_higher_dims_ndim_11 │ ├── TestDLPack.test_higher_dims_ndim_12 │ ├── TestDLPack.test_higher_dims_ndim_13 │ ├── TestDLPack.test_higher_dims_ndim_14 │ ├── TestDLPack.test_higher_dims_ndim_15 │ ├── TestDLPack.test_higher_dims_ndim_16 │ ├── TestDLPack.test_higher_dims_ndim_17 │ ├── TestDLPack.test_higher_dims_ndim_18 │ ├── TestDLPack.test_higher_dims_ndim_19 │ ├── TestDLPack.test_higher_dims_ndim_2 │ ├── TestDLPack.test_higher_dims_ndim_20 │ ├── TestDLPack.test_higher_dims_ndim_21 │ ├── TestDLPack.test_higher_dims_ndim_22 │ ├── TestDLPack.test_higher_dims_ndim_23 │ ├── TestDLPack.test_higher_dims_ndim_24 │ ├── TestDLPack.test_higher_dims_ndim_25 │ ├── TestDLPack.test_higher_dims_ndim_26 │ ├── TestDLPack.test_higher_dims_ndim_27 │ ├── TestDLPack.test_higher_dims_ndim_28 │ ├── TestDLPack.test_higher_dims_ndim_29 │ ├── TestDLPack.test_higher_dims_ndim_3 │ ├── TestDLPack.test_higher_dims_ndim_30 │ ├── TestDLPack.test_higher_dims_ndim_31 │ ├── TestDLPack.test_higher_dims_ndim_32 │ ├── TestDLPack.test_higher_dims_ndim_4 │ ├── TestDLPack.test_higher_dims_ndim_5 │ ├── TestDLPack.test_higher_dims_ndim_6 │ ├── TestDLPack.test_higher_dims_ndim_7 │ ├── TestDLPack.test_higher_dims_ndim_8 │ ├── TestDLPack.test_higher_dims_ndim_9 │ ├── TestDLPack.test_ndim0 │ ├── TestDelete.test_slices │ ├── TestDeserialize.test_tensor_tensor_list │ ├── TestDiag.test_failure │ ├── TestDiff.test_append │ ├── TestDiff.test_axis │ ├── TestDiff.test_n │ ├── TestDiff.test_prepend │ ├── TestDropoutNN.test_invalid_dropout_p │ ├── TestDsplit.test_0D_array │ ├── TestDsplit.test_1D_array │ ├── TestDsplit.test_2D_array │ ├── TestDsplit.test_non_iterable │ ├── TestDstack.test_non_iterable │ ├── TestDynamicQuantizedOps.test_qlinear │ ├── TestDynamismExpression.test_export_inline_constraints │ ├── TestEigh.test_invalid │ ├── TestEigvalsh.test_invalid │ ├── TestExpandDims.test_axis_out_of_range │ ├── TestExpandDims.test_repeated_axis │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv1d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv2d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_conv3d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_group_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_instance_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_mean_nn_functional_layer_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv1d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv2d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_conv3d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_group_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_instance_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weight_per_sample_grad_sum_nn_functional_layer_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv1d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv2d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_conv3d_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_group_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_instance_norm_cpu_float64 │ ├── TestExpandedWeightFunctionalCPU.test_expanded_weights_per_sample_grad_input_no_grad_nn_functional_layer_norm_cpu_float64 │ ├── TestExperiment.test_with_buffer_as_submodule │ ├── TestExperimentalUtils.test_profiler_optimizer_single_tensor_pattern │ ├── TestExperimentalUtils.test_profiler_synchronized_dataloader_pattern │ ├── TestExport.test_basic_non_strict_fake_tensor │ ├── TestExport.test_basic_non_strict_real_tensor │ ├── TestExport.test_cond_with_module_stack_export_with │ ├── TestExport.test_constrain_size_in_eager │ ├── TestExport.test_constrain_size_with_constrain_value │ ├── TestExport.test_constrain_size_with_various_cases │ ├── TestExport.test_external_call_non_strict_real_tensor │ ├── TestExport.test_nn_module_stack │ ├── TestExport.test_nn_module_stack_shared_submodule │ ├── TestExport.test_non_strict_dynamic_shapes │ ├── TestExport.test_non_strict_dynamic_shapes_suggested_fixes │ ├── TestFakeQuantizeOps.test_backward_per_channel │ ├── TestFakeQuantizeOps.test_backward_per_tensor │ ├── TestFakeQuantizeOps.test_learnable_backward_per_channel_cpu │ ├── TestFakeQuantizeOps.test_learnable_backward_per_channel_cuda │ ├── TestFakeQuantizeOps.test_learnable_backward_per_tensor_cuda │ ├── TestFakeQuantizeOps.test_learnable_forward_per_channel_cuda │ ├── TestFakeQuantizeOps.test_learnable_forward_per_tensor_cpu │ ├── TestFakeQuantizeOps.test_learnable_forward_per_tensor_cuda │ ├── TestFlag.test_writeable_from_readonly │ ├── TestFlip.test_axes │ ├── TestFliplr.test_basic │ ├── TestFloatNonIntegerArgument.test_non_integer_argument_errors │ ├── TestFloatNonIntegerArgument.test_reduce_axis_float_index │ ├── TestFromBuffer.test_empty │ ├── TestFunctionalizeCPU.test_multioutput_view_cpu │ ├── TestFunctionalizeCPU.test_simple_view_cpu │ ├── TestFunctionalizeCPU.test_vmap_functionalize_jvp_cpu │ ├── TestGenericCumSumProd.test_bad_axis_func0 │ ├── TestGenericCumSumProd.test_bad_axis_func1 │ ├── TestGenericPytree.test_flatten_unflatten_deque_cxx │ ├── TestGenericPytree.test_flatten_unflatten_deque_py │ ├── TestGenericReductions.test_bad_axis_func0 │ ├── TestGenericReductions.test_bad_axis_func1 │ ├── TestGenericReductions.test_bad_axis_func10 │ ├── TestGenericReductions.test_bad_axis_func11 │ ├── TestGenericReductions.test_bad_axis_func2 │ ├── TestGenericReductions.test_bad_axis_func3 │ ├── TestGenericReductions.test_bad_axis_func4 │ ├── TestGenericReductions.test_bad_axis_func5 │ ├── TestGenericReductions.test_bad_axis_func6 │ ├── TestGenericReductions.test_bad_axis_func7 │ ├── TestGenericReductions.test_bad_axis_func8 │ ├── TestGenericReductions.test_bad_axis_func9 │ ├── TestGradNewOnesOverride.test_newones │ ├── TestGradient.test_badargs │ ├── TestGradient.test_specific_axes │ ├── TestGradient.test_values │ ├── TestHessianCPU.test_jacfwd_different_levels_cpu │ ├── TestHessianCUDA.test_jacfwd_different_levels_cuda │ ├── TestHigherOrderOperatorInteractionCPU.test_grad_name_wrapping_cpu │ ├── TestHistogram.test_error_binnum_type │ ├── TestHistogram.test_finite_range │ ├── TestHistogram.test_one_bin │ ├── TestHistogram2d.test_binparameter_combination │ ├── TestHistogramdd.test_bins_error_2 │ ├── TestHistogramdd.test_bins_errors │ ├── TestHistogramdd.test_equal_edges │ ├── TestHistogramdd.test_inf_edges │ ├── TestHistogramdd.test_weights │ ├── TestHsplit.test_0D_array │ ├── TestHsplit.test_non_iterable │ ├── TestHstack.test_empty_input │ ├── TestHstack.test_non_iterable │ ├── TestIndexing.test_index_no_floats │ ├── TestIndexing.test_slicing_no_floats │ ├── TestIndexingCPU.test_byte_mask_cpu │ ├── TestIndexingCPU.test_empty_ndim_index_bool_cpu │ ├── TestIndexingCPU.test_index_cpu │ ├── TestIndexingCPU.test_index_limits_cpu │ ├── TestIndexingCPU.test_out_of_bound_index_cpu │ ├── TestIndexingCPU.test_zero_dim_index_cpu │ ├── TestInputAttrTracking.test_complex_attr_access_without_graph_breaks │ ├── TestInputAttrTracking.test_const_property_on_tensor │ ├── TestInputAttrTracking.test_set_data_on_input_tensor │ ├── TestInputAttrTracking.test_tensor_property_assigned_on_tensor │ ├── TestInputAttrTracking.test_tensor_property_on_tensor │ ├── TestIsScalar.test_is_not_scalar_value6 │ ├── TestIterator.test_iterator │ ├── TestJacCPU.test_against_reference_correctness_different_devices_cpu │ ├── TestJacCPU.test_against_reference_default_arg_cpu │ ├── TestJacCPU.test_against_reference_multi_input_cpu │ ├── TestJacCPU.test_against_reference_multi_input_multi_output_cpu │ ├── TestJacCPU.test_against_reference_simple_cpu │ ├── TestJacCPU.test_against_reference_unrelated_outputs_cpu │ ├── TestJacCPU.test_against_reference_zero_dim_cpu │ ├── TestJacCPU.test_argnums_defaults_to_zero_cpu │ ├── TestJacCPU.test_aux_pytree_cpu │ ├── TestJacCPU.test_dimensionality_cpu │ ├── TestJacCPU.test_empty_output_cpu │ ├── TestJacCPU.test_inplace_cpu │ ├── TestJacCPU.test_jac_with_non_tensor_args_cpu │ ├── TestJacCPU.test_multiple_inputs_outputs_pytree_cpu │ ├── TestJacCPU.test_multiple_inputs_pytree_cpu │ ├── TestJacCPU.test_multiple_outputs_multiple_argnums_cpu │ ├── TestJacCPU.test_multiple_outputs_single_argnums_cpu │ ├── TestJacCPU.test_outputs_can_any_pytree_cpu │ ├── TestJacCPU.test_unrelated_input_cpu │ ├── TestJacCPU.test_unrelated_output_cpu │ ├── TestJit.test_batchnorm │ ├── TestLazyModules.test_lazy_batchnorm_with_dict_input │ ├── TestLexsort.test_datetime │ ├── TestLexsort.test_mixed │ ├── TestLinalgCPU.test_addmm_sizes_cpu_float32 │ ├── TestLinalgCPU.test_addmm_sizes_cpu_float64 │ ├── TestLinalgCPU.test_addr_integral_cpu_int16 │ ├── TestLinalgCPU.test_addr_integral_cpu_int32 │ ├── TestLinalgCPU.test_addr_integral_cpu_int64 │ ├── TestLinalgCPU.test_addr_integral_cpu_int8 │ ├── TestLinalgCPU.test_addr_integral_cpu_uint8 │ ├── TestLinalgCPU.test_geqrf_cpu_complex128 │ ├── TestLinalgCPU.test_geqrf_cpu_complex64 │ ├── TestLinalgCPU.test_geqrf_cpu_float32 │ ├── TestLinalgCPU.test_geqrf_cpu_float64 │ ├── TestLinalgCPU.test_householder_product_cpu_complex128 │ ├── TestLinalgCPU.test_householder_product_cpu_complex64 │ ├── TestLinalgCPU.test_householder_product_cpu_float32 │ ├── TestLinalgCPU.test_householder_product_cpu_float64 │ ├── TestLinalgCPU.test_pinv_cpu_complex128 │ ├── TestLinalgCPU.test_pinv_cpu_complex64 │ ├── TestLinalgCPU.test_pinv_cpu_float32 │ ├── TestLinalgCPU.test_pinv_cpu_float64 │ ├── TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex128 │ ├── TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_complex64 │ ├── TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float32 │ ├── TestLinalgCPU.test_slogdet_errors_and_warnings_cpu_float64 │ ├── TestLinalgCPU.test_solve_cpu_complex128 │ ├── TestLinalgCPU.test_solve_cpu_complex64 │ ├── TestLinalgCPU.test_solve_cpu_float32 │ ├── TestLinalgCPU.test_solve_cpu_float64 │ ├── TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_complex128 │ ├── TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_complex64 │ ├── TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_float32 │ ├── TestLinalgCPU.test_tensorinv_errors_and_warnings_cpu_float64 │ ├── TestMatmul.test_exceptions │ ├── TestMedian.test_extended_axis_invalid │ ├── TestMeshgrid.test_indexing │ ├── TestMetaKernel.test_addmm_invalid_dtype │ ├── TestMin.test_adapt │ ├── TestMin.test_attn │ ├── TestMin.test_big_split │ ├── TestMin.test_c │ ├── TestMin.test_compare_dims │ ├── TestMin.test_diag │ ├── TestMin.test_dim_args │ ├── TestMin.test_dims_with_size │ ├── TestMin.test_dir │ ├── TestMin.test_doc │ ├── TestMin.test_embed │ ├── TestMin.test_eq │ ├── TestMin.test_expand │ ├── TestMin.test_functorch │ ├── TestMin.test_hello │ ├── TestMin.test_index │ ├── TestMin.test_index_placement │ ├── TestMin.test_inplace │ ├── TestMin.test_manual_stuff │ ├── TestMin.test_mask │ ├── TestMin.test_max │ ├── TestMin.test_mm │ ├── TestMin.test_mm_fuse │ ├── TestMin.test_monkey │ ├── TestMin.test_network │ ├── TestMin.test_order │ ├── TestMin.test_order_keyword │ ├── TestMin.test_parse │ ├── TestMin.test_permute_orig │ ├── TestMin.test_seg │ ├── TestMin.test_simple │ ├── TestMin.test_softmax_split │ ├── TestMin.test_stack │ ├── TestMin.test_time_mm_fuse │ ├── TestMin.test_with_dims_split │ ├── TestMinFunctorchOnly.test_adapt │ ├── TestMinFunctorchOnly.test_attn │ ├── TestMinFunctorchOnly.test_big_split │ ├── TestMinFunctorchOnly.test_c │ ├── TestMinFunctorchOnly.test_compare_dims │ ├── TestMinFunctorchOnly.test_diag │ ├── TestMinFunctorchOnly.test_dim_args │ ├── TestMinFunctorchOnly.test_dims_with_size │ ├── TestMinFunctorchOnly.test_dir │ ├── TestMinFunctorchOnly.test_doc │ ├── TestMinFunctorchOnly.test_embed │ ├── TestMinFunctorchOnly.test_eq │ ├── TestMinFunctorchOnly.test_expand │ ├── TestMinFunctorchOnly.test_functorch │ ├── TestMinFunctorchOnly.test_hello │ ├── TestMinFunctorchOnly.test_index │ ├── TestMinFunctorchOnly.test_index_placement │ ├── TestMinFunctorchOnly.test_inplace │ ├── TestMinFunctorchOnly.test_manual_stuff │ ├── TestMinFunctorchOnly.test_mask │ ├── TestMinFunctorchOnly.test_max │ ├── TestMinFunctorchOnly.test_mm │ ├── TestMinFunctorchOnly.test_mm_fuse │ ├── TestMinFunctorchOnly.test_monkey │ ├── TestMinFunctorchOnly.test_network │ ├── TestMinFunctorchOnly.test_order │ ├── TestMinFunctorchOnly.test_order_keyword │ ├── TestMinFunctorchOnly.test_parse │ ├── TestMinFunctorchOnly.test_permute_orig │ ├── TestMinFunctorchOnly.test_seg │ ├── TestMinFunctorchOnly.test_simple │ ├── TestMinFunctorchOnly.test_softmax_split │ ├── TestMinFunctorchOnly.test_stack │ ├── TestMinFunctorchOnly.test_with_dims_split │ ├── TestMinMax.test_axis │ ├── TestMinMax.test_scalar │ ├── TestMisc.test_byteorder_check │ ├── TestMisc.test_generalized_raise_multiloop │ ├── TestMkldnnCPU.test_add_cpu │ ├── TestMkldnnCPU.test_copy_cpu │ ├── TestMkldnnCPU.test_mul_cpu │ ├── TestMkldnnCPU.test_sigmoid_cpu │ ├── TestMkldnnCPU.test_tanh_cpu │ ├── TestModuleHookNN.test_hook_inplace │ ├── TestMoveaxis.test_errors │ ├── TestMultiDot.test_dynamic_programming_optimization_and_out │ ├── TestMultiDot.test_three_arguments_and_out │ ├── TestMultiDot.test_too_few_input_arrays │ ├── TestMultiDot.test_two_arguments_and_out │ ├── TestMultiprocessing.test_empty_shared │ ├── TestMultiprocessing.test_fs_is_shared │ ├── TestMultiprocessing.test_inherit_tensor │ ├── TestMultiprocessing.test_is_shared │ ├── TestNN.test_fb_fc_packed │ ├── TestNN.test_linear_autograd_device_cuda_bias_weightCOO │ ├── TestNN.test_linear_autograd_device_cuda_bias_weightCSC │ ├── TestNN.test_linear_autograd_device_cuda_bias_weightCSR │ ├── TestNN.test_linear_autograd_device_cuda_nobias_weightCOO │ ├── TestNNDeviceTypeCPU.test_invalid_reduction_strings_cpu │ ├── TestNNDeviceTypeCPU.test_module_to_empty_cpu_float32 │ ├── TestNNDeviceTypeCPU.test_module_to_empty_cpu_float64 │ ├── TestNNDeviceTypeCPU.test_nll_loss_byte_target_matches_long_cpu │ ├── TestNNDeviceTypeCPU.test_threshold_inplace_overlap_cpu │ ├── TestNNDeviceTypeCUDA.test_nll_loss_byte_target_matches_long_cuda │ ├── TestNNParametrization.test_deepcopy_after_parametrization_swap_True │ ├── TestNNParametrization.test_errors_unparametrized_tensor_parametrization_swap_True │ ├── TestNNParametrization.test_initialization_parametrization_swap_True │ ├── TestNNParametrization.test_new_spectral_norm_forward_swap_True │ ├── TestNNParametrization.test_new_spectral_norm_swap_True │ ├── TestNNParametrization.test_register_and_remove_buffer_parametrization_swap_True │ ├── TestNNParametrization.test_register_and_remove_nested_parametrization_swap_True │ ├── TestNNParametrization.test_register_and_remove_parametrization_swap_True │ ├── TestNNParametrization.test_serialization_parametrization_swap_True │ ├── TestNNParametrization.test_transfer_parametrizations_and_params_right_inverse_swap_True │ ├── TestNNParametrization.test_transfer_parametrizations_and_params_swap_True │ ├── TestNNParametrization.test_wrapper_subclass_parametrization_swap_True │ ├── TestNNParametrizationDeviceCUDA.test_weight_norm_parametrization_swap_False_cuda │ ├── TestNNParametrizationDeviceCUDA.test_weight_norm_parametrization_swap_True_cuda │ ├── TestNamedTensor.test_flatten │ ├── TestNamedTensor.test_reduction_fns │ ├── TestNamedTensor.test_set_names_property │ ├── TestNamedTensor.test_transpose_variants │ ├── TestNamedTensor.test_unary_propagate_names_fns │ ├── TestNamedTensor.test_unflatten │ ├── TestNamedTensor.test_using_unseen_interned_string_bumps_refcount_permanently │ ├── TestNamedTensor.test_using_unseen_uninterned_string_refcounts │ ├── TestNamedTuple.test_max │ ├── TestNegative.test_exceptions │ ├── TestNestedCheckpoint.test_nested_checkpoint_early_stop_False │ ├── TestNestedCheckpoint.test_nested_checkpoint_early_stop_True │ ├── TestNestedCheckpoint.test_nested_checkpoint_kwargs_early_stop_True │ ├── TestNestedCheckpoint.test_nested_checkpoint_non_tensor_inputs_and_outputs_early_stop_True │ ├── TestNestedCheckpoint.test_nested_checkpoint_reentrant_backwards_early_stop_True │ ├── TestNestedCheckpoint.test_nested_checkpoint_same_graph_early_stop_True │ ├── TestNestedCheckpoint.test_nested_checkpoint_set_early_stop │ ├── TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_False │ ├── TestNestedCheckpoint.test_nested_checkpoint_two_children_early_stop_True │ ├── TestNestedTensor.test_binary_recompiles │ ├── TestNestedTensor.test_unbind │ ├── TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-1 │ ├── TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-10 │ ├── TestNonarrayArgs.test_dunder_round_edgecases_val_2147483647_ndigits_-9 │ ├── TestNonzeroAndCountNonzero.test_count_nonzero_axis │ ├── TestNormDouble.test_axis │ ├── TestNormDouble.test_bad_args │ ├── TestNormDouble.test_matrix_2x2 │ ├── TestNormInt64.test_axis │ ├── TestNormInt64.test_bad_args │ ├── TestNormInt64.test_matrix_2x2 │ ├── TestNormSingle.test_axis │ ├── TestNormSingle.test_bad_args │ ├── TestNormSingle.test_matrix_2x2 │ ├── TestNumPyInteropCPU.test_numpy_non_writeable_cpu │ ├── TestObserver.test_per_channel_observers │ ├── TestObserver.test_per_tensor_observers │ ├── TestOptimizations.test_example_inputs │ ├── TestOptimizations.test_example_inputs_runtime_use │ ├── TestOutDtypeOp.test_out_dtype_non_op_overload │ ├── TestOutDtypeOp.test_out_dtype_wrong_output │ ├── TestPadding.test_constant_padNd │ ├── TestPasses.test_functionalize_inline_contraints │ ├── TestPasses.test_views_op_having_view_copy │ ├── TestPercentile.test_extended_axis_invalid │ ├── TestPercentile.test_scalar_q │ ├── TestPickle.test_pickle │ ├── TestPoolingNN.test_MaxUnpool2d_output_size │ ├── TestProfiler.test_flops │ ├── TestProfiler.test_high_level_trace │ ├── TestProfiler.test_is_profiler_enabled │ ├── TestProfiler.test_source │ ├── TestProfiler.test_source_multithreaded_basic_work_in_main_thread_True │ ├── TestProfiler.test_source_multithreaded_close_in_scope_work_in_main_thread_True │ ├── TestProfiler.test_source_multithreaded_complex_work_in_main_thread_True │ ├── TestProfiler.test_source_multithreaded_multiple_preexisting_work_in_main_thread_True │ ├── TestProfiler.test_source_multithreaded_open_in_scope_work_in_main_thread_True │ ├── TestProfilerTree.test_profiler_experimental_tree_with_memory │ ├── TestProfilerTree.test_profiler_experimental_tree_with_memory_and_stack │ ├── TestProfilerTree.test_profiler_experimental_tree_with_record_function │ ├── TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_dispatch │ ├── TestProfilerTree.test_profiler_experimental_tree_with_stack_and_torch_function │ ├── TestPutAlongAxis.test_broadcast │ ├── TestPutmask.test_byteorder_greater_False │ ├── TestPutmask.test_byteorder_greater_True │ ├── TestPutmask.test_record_array │ ├── TestPythonAutograd.test_backwards2 │ ├── TestPythonAutograd.test_forwards1 │ ├── TestPythonAutograd.test_forwards2 │ ├── TestPythonAutograd.test_split │ ├── TestPythonRegistration.test_finalizer │ ├── TestQR.test_mode_raw │ ├── TestQR.test_qr_empty_m_0_n_0 │ ├── TestQR.test_qr_empty_m_0_n_3 │ ├── TestQR.test_qr_empty_m_3_n_0 │ ├── TestQuantile.test_quantile_monotonic_method_averaged_inverted_cdf │ ├── TestQuantile.test_quantile_monotonic_method_closest_observation │ ├── TestQuantile.test_quantile_monotonic_method_hazen │ ├── TestQuantile.test_quantile_monotonic_method_interpolated_inverted_cdf │ ├── TestQuantile.test_quantile_monotonic_method_inverted_cdf │ ├── TestQuantile.test_quantile_monotonic_method_median_unbiased │ ├── TestQuantile.test_quantile_monotonic_method_normal_unbiased │ ├── TestQuantile.test_quantile_monotonic_method_weibull │ ├── TestQuantizedEmbeddingOps.test_embedding_bag_2d_indices │ ├── TestQuantizedOps.test_custom_module_lstm │ ├── TestQuantizedOps.test_max_pool2d_cudnn │ ├── TestQuantizedOps.test_qtanh │ ├── TestQuantizedTensor.test_choose_qparams │ ├── TestQuantizedTensor.test_choose_qparams_optimized │ ├── TestQuantizedTensor.test_decomposed_dequantize_per_tensor │ ├── TestQuantizedTensor.test_decomposed_quantize_per_tensor │ ├── TestQuantizedTensor.test_decomposed_quantize_per_tensor_bfloat16_input │ ├── TestQuantizedTensor.test_fp16_saturate_op │ ├── TestQuantizedTensor.test_qtensor_cpu │ ├── TestQuantizedTensor.test_quantize_per_channel_sub_byte │ ├── TestReductionsCPU.test_std_vs_numpy_cpu_complex128 │ ├── TestReductionsCPU.test_std_vs_numpy_cpu_complex64 │ ├── TestReductionsCPU.test_std_vs_numpy_cpu_float32 │ ├── TestReductionsCPU.test_std_vs_numpy_cpu_float64 │ ├── TestReductionsCPU.test_var_vs_numpy_cpu_complex128 │ ├── TestReductionsCPU.test_var_vs_numpy_cpu_complex64 │ ├── TestReductionsCPU.test_var_vs_numpy_cpu_float32 │ ├── TestReductionsCPU.test_var_vs_numpy_cpu_float64 │ ├── TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_complex128 │ ├── TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_complex64 │ ├── TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_float32 │ ├── TestReductionsCPU.test_warn_invalid_degrees_of_freedom_cpu_float64 │ ├── TestRequire.test_C_and_F_simul │ ├── TestRequire.test_require_each │ ├── TestRequire.test_unknown_requirement │ ├── TestRollaxis.test_exceptions │ ├── TestRot90.test_basic │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_bfloat16_cpu_bfloat16 │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float16_cpu_float16 │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float32_cpu_float32 │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_0_float64_cpu_float64 │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_bfloat16_cpu_bfloat16 │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float16_cpu_float16 │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float32_cpu_float32 │ ├── TestSDPACPU.test_fused_sdp_choice_cpu_type_dense_dropout_0_7_float64_cpu_float64 │ ├── TestScalarIndexing.test_invalid_newaxis │ ├── TestScalarIndexing.test_invalid_subscript │ ├── TestScalarIndexing.test_invalid_subscript_assignment │ ├── TestScript.test_conv_error │ ├── TestScript.test_error_stacktrace_interface │ ├── TestScript.test_infer_size │ ├── TestScript.test_is_scripting │ ├── TestScript.test_method_overloading │ ├── TestScript.test_namedtuple_default_values_using_factory_constructor │ ├── TestScript.test_parse_nested_names │ ├── TestScript.test_parse_tensor_constants │ ├── TestScript.test_python_call │ ├── TestScript.test_script_pack_padded_sequence │ ├── TestScript.test_string_device_implicit_conversion │ ├── TestScript.test_torch_functional_tensordot_int │ ├── TestShapeOpsCUDA.test_flip_cuda_float32 │ ├── TestSortComplex.test_sort_real_type_in_g_type_out_G │ ├── TestSplit.test_unequal_split │ ├── TestStackMisc.test_stack │ ├── TestStats.test_dtype_from_dtype │ ├── TestStats.test_out │ ├── TestSubclassSerialization.test_allowlist_for_weights_only │ ├── TestTEFuserDynamic.test_remove_output_used_only_in_size │ ├── TestTEFuserDynamic.test_to_dtype │ ├── TestTEFuserDynamic.test_torch_to │ ├── TestTEFuserStatic.test_inlined_optimized_graph │ ├── TestTEFuserStatic.test_remove_output_used_only_in_size │ ├── TestTEFuserStatic.test_to_dtype │ ├── TestTEFuserStatic.test_torch_to │ ├── TestTake.test_raise │ ├── TestTakeAlongAxis.test_invalid │ ├── TestTensorBoardUtils.test_numpy_vid_uint8 │ ├── TestTensorCreationCPU.test_block_diag_cpu │ ├── TestTensorCreationCPU.test_constructor_dtypes_cpu │ ├── TestTensorProtoSummary.test_half_tensor_proto_bfloat16_proto_type_14 │ ├── TestTensorProtoSummary.test_half_tensor_proto_float16_proto_type_19 │ ├── TestTorch.test_map │ ├── TestTorch.test_new │ ├── TestTorch.test_parsing_intlist │ ├── TestTorch.test_type │ ├── TestTorch.test_upsample_nearest2d_meta │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_add_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_addcdiv_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_addcmul_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_atan2_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_div_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_eq_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_fmod_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_ge_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_gt_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_le_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_lerp_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_lt_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_masked_fill_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_masked_scatter_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_mul_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_ne_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_pow_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_remainder_cpu │ ├── TestTorchDeviceTypeCPU.test_broadcast_fn_sub_cpu │ ├── TestTorchDeviceTypeCPU.test_deterministic_empty_cpu_uint64 │ ├── TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float32 │ ├── TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool1d_cpu_float64 │ ├── TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float32 │ ├── TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool2d_cpu_float64 │ ├── TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float32 │ ├── TestTorchDeviceTypeCPU.test_nondeterministic_alert_MaxUnpool3d_cpu_float64 │ ├── TestTorchFunctionMode.test_mode_notimplemented_loop │ ├── TestTorchFunctionMode.test_modes_return_notimplemented │ ├── TestTorchFunctionMode.test_nested_modes_with_python_has_torch_function │ ├── TestTorchFunctionOverride.test_Tensor___cuda_array_interface_____get__ │ ├── TestTorchTidyProfiler.test_module_and_optimizer_ids │ ├── TestTorchTidyProfiler.test_nnmodule_params │ ├── TestTorchTidyProfiler.test_tensorimpl_invalidation_full │ ├── TestTorchTidyProfiler.test_tensorimpl_invalidation_keep_alive │ ├── TestTorchTidyProfiler.test_tensorimpl_invalidation_scalar_args │ ├── TestTorchTidyProfiler.test_tensorimpl_invalidation_set │ ├── TestTorchbind.test_torchbind_inductor │ ├── TestTrilIndicesFrom.test_exceptions │ ├── TestTriuIndicesFrom.test_exceptions │ ├── TestTypeHints.test_doc_examples │ ├── TestTypePromotionCPU.test_alpha_mismatch_cpu │ ├── TestTypePromotionCPU.test_alternate_result_cpu │ ├── TestUnflatten.test_unflatten_container_type │ ├── TestUnique.test_unique_axis │ ├── TestUnique.test_unique_axis_errors │ ├── TestVerifyCorrectness.test_incorrect_verify_true │ ├── TestVerifyCorrectness.test_torchscript │ ├── TestVmapAPI.test_fallback_does_not_warn_by_default │ ├── TestVmapAPI.test_fallback_warns_when_warnings_are_enabled │ ├── TestVmapAPI.test_out_dim_out_of_bounds_err_msg │ ├── TestVmapOperators.test_new_empty_strided │ ├── TestVsplit.test_0D_array │ ├── TestVsplit.test_1D_array │ ├── TestVsplit.test_non_iterable │ ├── TestVstack.test_empty_input │ ├── TestVstack.test_non_iterable │ ├── TestWhere.test_error │ ├── TraceRuleTests.test_skipfiles_inlinelist │ ├── UnspecTests.test_builtin_max_min │ ├── UnspecTests.test_conv1d_symint_padding │ ├── UnspecTests.test_isinstance_symint │ ├── UnspecTests.test_mark_01_dynamic │ ├── UnspecTests.test_no_recompilations │ ├── UnspecTests.test_no_recompiles │ ├── UnspecTests.test_propagate_dynamic_dim │ └── UnspecTests.test_use_and_specialize ├── dynamo_skips │ ├── DecoratorTests.test_allow_in_graph │ ├── DynamicShapesAotAutogradFallbackTests.test_aot_sequence_nr_dynamic_shapes │ ├── DynamicShapesCtxManagerTests.test_autograd_profiler_enabled_dynamic_shapes │ ├── DynamicShapesCtxManagerTests.test_disable_saved_tensors_hooks_dynamic_shapes │ ├── DynamicShapesCtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled_dynamic_shapes │ ├── DynamicShapesCtxManagerTests.test_disable_saved_tensors_hooks_prev_disabled_nested_dynamic_shapes │ ├── DynamicShapesExportTests.test_cond_raise_user_error_on_non_list_operands_dynamic_shapes │ ├── DynamicShapesExportTests.test_cond_raise_user_error_on_non_tensor_operands_dynamic_shapes │ ├── DynamicShapesExportTests.test_cond_raise_user_error_on_unsupported_pred_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_dynamic_dim_range_constraint_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_mark_dynamic_conflict_dynamic_dim_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_dict_values_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_free_function_and_class_method_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_free_function_and_class_method_multiarg_diff_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_free_function_and_class_method_multiarg_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_free_function_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_list_nonzero_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_list_nonzero_free_function_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_method_on_module_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_method_on_module_invoke_twice_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_none_control_flow_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_none_control_flow_free_func_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_not_none_control_flow_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_not_none_control_flow_free_func_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_not_none_control_flow_pos_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_not_return_const_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_constant_tuple_nonzero_dynamic_shapes │ ├── DynamicShapesExportTests.test_export_with_map_cond_dynamic_shapes │ ├── DynamicShapesExportTests.test_map_cond_param_buffer_lifted_dynamic_shapes │ ├── DynamicShapesExportTests.test_retracibility_dynamic_shapes │ ├── DynamicShapesExportTests.test_torch_inference_mode_ctx_dynamic_shapes │ ├── DynamicShapesExportTests.test_untracked_inputs_in_constraints_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_disable_capture_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_fn_with_kwargs_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_freevar_python_scalar_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_has_aux_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_non_tensor_input_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_pytree_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_two_tensor_all_grad_has_aux_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_two_tensor_has_aux_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_with_graph_break_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_grad_with_side_effect_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_disable_capture_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_free_const_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_free_tensor_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_illegal_op_graph_break_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_kwargs_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_in_dims_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_multiple_invocation_out_dims_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_diff_dims_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_multiple_outputs_out_dims_tuple_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_over_vmap_captured_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_over_vmap_two_inputs_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_pytree_inputs_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_side_effects_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_two_inputs_dynamic_shapes │ ├── DynamicShapesFuncTorchHigherOrderOpTests.test_vmap_two_inputs_tuple_in_dims_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_default_dict_closure_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_default_dict_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_default_dict_lambda_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_fstrings2_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_is_contiguous_frame_counts_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_math_radians_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_partials_as_input_partials_lambda_dynamic_shapes │ ├── DynamicShapesFunctionTests.test_partials_as_input_partials_mod_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_access_module_attr_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_capture_untracked_global_nested_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_capture_value_created_in_subgraph_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_cond_branches_no_arguments_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_cond_branches_no_arguments_no_closure_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_cond_pytree_operands_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_cond_pytree_operands_with_non_tensor_leaves_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_cond_source_fn_stack_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_cond_subgraph_name_is_valid_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_fallback_on_python_primitives_output_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_map_lowers_to_graph_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_map_multi_return_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_map_pytree_return_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_map_source_fn_stack_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_map_subgraph_name_is_valid_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_map_symint_input_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_modules_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_nested_tuple_output_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_output_with_dict_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_side_effect_in_body_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_vmap_source_fn_stack_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_wrap_kwarg_recompile_dynamic_shapes │ ├── DynamicShapesHigherOrderOpTests.test_wrap_subgraph_name_is_valid_dynamic_shapes │ ├── DynamicShapesMiscTests.test_add_to_set_dynamic_shapes │ ├── DynamicShapesMiscTests.test_any_all_symnode_dynamic_shapes │ ├── DynamicShapesMiscTests.test_backend_match_guard_dynamic_shapes │ ├── DynamicShapesMiscTests.test_boolarg_dynamic_shapes │ ├── DynamicShapesMiscTests.test_callpacked_dynamic_shapes │ ├── DynamicShapesMiscTests.test_closure_out_of_scope_cell_dynamic_shapes │ ├── DynamicShapesMiscTests.test_compare_shapes_with_constant_dynamic_shapes │ ├── DynamicShapesMiscTests.test_cond_nested_dynamic_shapes │ ├── DynamicShapesMiscTests.test_cond_side_effects_dynamic_shapes │ ├── DynamicShapesMiscTests.test_dataclass_fields_dynamic_shapes │ ├── DynamicShapesMiscTests.test_deque_append_left_dynamic_shapes │ ├── DynamicShapesMiscTests.test_deque_input_dynamic_shapes │ ├── DynamicShapesMiscTests.test_deterministic_algorithms_mutated_dynamic_shapes │ ├── DynamicShapesMiscTests.test_dict_order_keys_dynamic_shapes │ ├── DynamicShapesMiscTests.test_dict_order_keys_modules_dynamic_shapes │ ├── DynamicShapesMiscTests.test_dict_order_keys_tensors_dynamic_shapes │ ├── DynamicShapesMiscTests.test_dictcomp_dynamic_shapes │ ├── DynamicShapesMiscTests.test_dtypes_no_graphbreaks_dynamic_shapes │ ├── DynamicShapesMiscTests.test_dunder_new_function_inlining_dynamic_shapes │ ├── DynamicShapesMiscTests.test_frozenset_torch_func_contains_dynamic_shapes │ ├── DynamicShapesMiscTests.test_get_cache_entry_dynamic_shapes │ ├── DynamicShapesMiscTests.test_grad_state_mutated_dynamic_shapes │ ├── DynamicShapesMiscTests.test_guard_failure_fn_shape_control_dynamic_shapes │ ├── DynamicShapesMiscTests.test_guard_failure_fn_tensor_iter_dynamic_shapes │ ├── DynamicShapesMiscTests.test_inline_closure_not_loaded_by_parent_dynamic_shapes │ ├── DynamicShapesMiscTests.test_inplace_param_update_dynamic_shapes │ ├── DynamicShapesMiscTests.test_intermediary_tensor_grad_access_dynamic_shapes │ ├── DynamicShapesMiscTests.test_is_compiling_dynamic_shapes │ ├── DynamicShapesMiscTests.test_iter_set_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_accumulate_symint_default_sum_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_accumulate_tensors_builtins_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_accumulate_tensors_default_sum_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_accumulate_tensors_kwargs_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_accumulate_tensors_user_defined_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_groupby_pure_python_default_identify_func_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_groupby_pure_python_key_func_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_infinite_count_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_infinite_cycle_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_infinite_repeat_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_infinite_repeat_mutation_dynamic_shapes │ ├── DynamicShapesMiscTests.test_itertools_repeat_dynamic_shapes │ ├── DynamicShapesMiscTests.test_linetable_311_writer1_dynamic_shapes │ ├── DynamicShapesMiscTests.test_linetable_311_writer2_dynamic_shapes │ ├── DynamicShapesMiscTests.test_list_slice_mul_dynamic_shapes │ ├── DynamicShapesMiscTests.test_listcomp_dynamic_shapes │ ├── DynamicShapesMiscTests.test_mandelbrot_numpy_dynamic_shapes │ ├── DynamicShapesMiscTests.test_mark_static_dynamic_shapes │ ├── DynamicShapesMiscTests.test_namedtuple1_dynamic_shapes │ ├── DynamicShapesMiscTests.test_namedtuple2_dynamic_shapes │ ├── DynamicShapesMiscTests.test_nan_dynamic_shapes │ ├── DynamicShapesMiscTests.test_nested_closure_dynamic_shapes │ ├── DynamicShapesMiscTests.test_nested_closure_mutation_dynamic_shapes │ ├── DynamicShapesMiscTests.test_nested_optimize_decorator_dynamic_shapes │ ├── DynamicShapesMiscTests.test_nested_optimize_dynamic_shapes │ ├── DynamicShapesMiscTests.test_nested_optimize_run_dynamic_shapes │ ├── DynamicShapesMiscTests.test_nn_module_getattr_dynamic_shapes │ ├── DynamicShapesMiscTests.test_no_raise_guard_partial_constraint_across_break_dynamic_shapes │ ├── DynamicShapesMiscTests.test_no_raise_guard_partial_constraint_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_array_of_arrays_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_force_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_int_constant_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_iter_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_non_torch_dtype_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_readonly_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_recompilation_scalar_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_size_attr_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_subdtype_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_tolist_dynamic_shapes │ ├── DynamicShapesMiscTests.test_numpy_with_builtin_type_dynamic_shapes │ ├── DynamicShapesMiscTests.test_out_variants_with_resizing_on_graph_inputs_dynamic_shapes │ ├── DynamicShapesMiscTests.test_pure_python_accumulate_dynamic_shapes │ ├── DynamicShapesMiscTests.test_py_guards_mark_dynamic_dynamic_shapes │ ├── DynamicShapesMiscTests.test_raise_guard_full_constraint_dynamic_shapes │ ├── DynamicShapesMiscTests.test_raise_guard_partial_constraint_no_graph_break_dynamic_shapes │ ├── DynamicShapesMiscTests.test_raise_on_backend_error_dynamic_shapes │ ├── DynamicShapesMiscTests.test_recompile_on_global_state_change_dynamic_shapes │ ├── DynamicShapesMiscTests.test_release_input_memory_dynamic_shapes │ ├── DynamicShapesMiscTests.test_release_module_memory_dynamic_shapes │ ├── DynamicShapesMiscTests.test_release_scope_memory_dynamic_shapes │ ├── DynamicShapesMiscTests.test_repeat_interleave_graphbreaks_dynamic_shapes │ ├── DynamicShapesMiscTests.test_return_nested_function_dynamic_shapes │ ├── DynamicShapesMiscTests.test_set_aliasing_recompiles_dynamic_shapes │ ├── DynamicShapesMiscTests.test_setattr_mutation1_dynamic_shapes │ ├── DynamicShapesMiscTests.test_simple_set_usage_dynamic_shapes │ ├── DynamicShapesMiscTests.test_size_input_dynamic_shapes │ ├── DynamicShapesMiscTests.test_slice_input_dynamic_shapes │ ├── DynamicShapesMiscTests.test_str_format_assert2_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tensor_build_list_unpack_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tensor_dict1_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tensor_dict2_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tensor_dict3_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tensor_item_capture_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tolist_0d_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tolist_1d_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tolist_kd_dynamic_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tolist_kd_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tolist_scalar_dynamic_shapes │ ├── DynamicShapesMiscTests.test_torch_seed_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tracing_nested_py_tree_dicts_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tracing_nested_py_tree_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tracing_nested_py_tree_mixed_all_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tracing_nested_py_tree_tuples_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tracing_py_tree_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tracing_py_tree_tensor_subclass_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tracing_tree_map_only_dynamic_shapes │ ├── DynamicShapesMiscTests.test_tuple_mul_dynamic_shapes │ ├── DynamicShapesMiscTests.test_type_copy_dynamic_shapes │ ├── DynamicShapesMiscTests.test_typing_typevar_dynamic_shapes │ ├── DynamicShapesMiscTests.test_user_getattribute_dynamic_shapes │ ├── DynamicShapesMiscTests.test_with_builtin_type_dynamic_shapes │ ├── DynamicShapesMiscTests.test_yield_from_dynamic_shapes │ ├── DynamicShapesMiscTests.test_yield_gen_and_from_dynamic_shapes │ ├── DynamicShapesMiscTests.test_yield_send_to_subgenerator_graph_break_dynamic_shapes │ ├── DynamicShapesNNModuleTests.test_self_mutating1_dynamic_shapes │ ├── DynamicShapesNNModuleTests.test_unsupportedmethod_dynamic_shapes │ ├── DynamicShapesNNModuleTests.test_unsupportedmodule_dynamic_shapes │ ├── DynamicShapesReproTests.test_add_sub_alpha_out_dynamic_shapes │ ├── DynamicShapesReproTests.test_addr_alpha_beta_out_dynamic_shapes │ ├── DynamicShapesReproTests.test_boxes_len_dynamic_shapes │ ├── DynamicShapesReproTests.test_chunk_reformer_ff_dynamic_shapes │ ├── DynamicShapesReproTests.test_convert_boxes_to_pooler_format_dynamic_shapes │ ├── DynamicShapesReproTests.test_create_rand_mask_from_inputs_dynamic_shapes │ ├── DynamicShapesReproTests.test_do_paste_mask_dynamic_shapes │ ├── DynamicShapesReproTests.test_dynamic_shapes_implicit_guard_dynamic_shapes │ ├── DynamicShapesReproTests.test_empty_list_contains_with_jump_dynamic_shapes │ ├── DynamicShapesReproTests.test_function_in_skipfiles_dynamic_shapes │ ├── DynamicShapesReproTests.test_hf_t5_forward_dynamic_shapes │ ├── DynamicShapesReproTests.test_hf_xsoftmax_training_dynamic_shapes │ ├── DynamicShapesReproTests.test_issue175_dynamic_shapes │ ├── DynamicShapesReproTests.test_list_aliasing_dynamic_shapes │ ├── DynamicShapesReproTests.test_list_self_reference_dynamic_shapes │ ├── DynamicShapesReproTests.test_longformer_chunk_dynamic_shapes │ ├── DynamicShapesReproTests.test_many_views_with_mutation_dynamic_shapes │ ├── DynamicShapesReproTests.test_merge_criteria_processor_list1_dynamic_shapes │ ├── DynamicShapesReproTests.test_merge_criteria_processor_list2_dynamic_shapes │ ├── DynamicShapesReproTests.test_module_in_skipfiles_dynamic_shapes │ ├── DynamicShapesReproTests.test_negative_shape_guard_dynamic_shapes │ ├── DynamicShapesReproTests.test_numpy_not_ndarray_recompiles_dynamic_shapes │ ├── DynamicShapesReproTests.test_optim_state_references_cleared_dynamic_shapes │ ├── DynamicShapesReproTests.test_recursive_map_dynamic_shapes │ ├── DynamicShapesReproTests.test_reformer_eval_dynamic_shapes │ ├── DynamicShapesReproTests.test_reformer_min_chunk_len_dynamic_shapes │ ├── DynamicShapesReproTests.test_reformer_sorting_dynamic_shapes │ ├── DynamicShapesReproTests.test_reformer_train_dynamic_shapes │ ├── DynamicShapesReproTests.test_relative_import_dynamic_shapes │ ├── DynamicShapesReproTests.test_relative_import_no_modulename_dynamic_shapes │ ├── DynamicShapesReproTests.test_restricted_list_subclass1_dynamic_shapes │ ├── DynamicShapesReproTests.test_rewrite_assert_noop_dynamic_shapes │ ├── DynamicShapesReproTests.test_rewrite_assert_with_msg_dynamic_shapes │ ├── DynamicShapesReproTests.test_rewrite_assert_with_non_string_msg_dynamic_shapes │ ├── DynamicShapesReproTests.test_seq_append_list_dynamic_shapes │ ├── DynamicShapesReproTests.test_size_typematch_dynamic_shapes │ ├── DynamicShapesReproTests.test_tensor_data_kwarg_dynamic_shapes │ ├── DynamicShapesReproTests.test_threading_local_dynamic_shapes │ ├── DynamicShapesReproTests.test_validate_model_kwargs_dynamic_shapes │ ├── DynamicShapesSubGraphTests.test_dynamic_duck_size_dynamic_shapes │ ├── DynamicShapesSubGraphTests.test_dynamic_getitem_dynamic_shapes │ ├── DynamicShapesSubGraphTests.test_dynamic_kwarg_dynamic_shapes │ ├── DynamicShapesSubGraphTests.test_dynamic_order_dependence_dynamic_shapes │ ├── DynamicShapesSubGraphTests.test_dynamic_zero_inference_dynamic_shapes │ ├── DynamicShapesSubGraphTests.test_enumerate_not_break_graph_dynamic_shapes │ ├── DynamicShapesSubGraphTests.test_no_graph_break_on_item_dynamic_shapes │ ├── DynamoProfilerTests.test_dynamo_timed_profiling_backend_compile │ ├── ExportTests.test_predispatch_with_for_out_dtype │ ├── ExportTests.test_predispatch_with_for_out_dtype_nested │ ├── FrameInitTests.test_frame_init │ ├── HigherOrderOpTests.test_access_module_attr │ ├── InPlaceCompilationTests.test_compilation │ ├── InteropTests.test_fx_fn │ ├── LoggingTests.test_distributed_rank_logging │ ├── LoggingTests.test_logs_out │ ├── LoggingTests.test_trace_call │ ├── LoggingTests.test_trace_call_graph_break │ ├── LoggingTests.test_trace_call_inline_call │ ├── MiscTests.test_auto_functionalize_on_view │ ├── MiscTests.test_auto_functionalize_optional │ ├── MiscTests.test_exception_table_e2e │ ├── MiscTests.test_exception_table_e2e_2 │ ├── MiscTests.test_exception_table_parsing │ ├── MiscTests.test_generate_trivial_abstract_impl │ ├── MiscTests.test_itertools_accumulate_tensors_builtins │ ├── MiscTests.test_itertools_accumulate_tensors_kwargs │ ├── MiscTests.test_itertools_accumulate_tensors_user_defined │ ├── MiscTests.test_itertools_infinite_count │ ├── MiscTests.test_linetable_311_writer1 │ ├── MiscTests.test_linetable_311_writer2 │ ├── MiscTests.test_py311_jump_offset │ ├── NopTests.test_extended_args │ ├── OptimizerTests.test_adadelta │ ├── RecompileTests.test_aliasing_guard_failures │ ├── RecompileUxTests.test_drop_cache_on_skip │ ├── ReproTests.test_optim_state_references_cleared │ ├── ReproTests.test_reformer_train │ ├── TestAOTAutograd.test_view_detach │ ├── TestArgmax.test_combinations_data58 │ ├── TestArgmax.test_combinations_data61 │ ├── TestAutograd.test_hook_closure_cycle_use_custom_function_True_use_tensor_hook_True │ ├── TestAutograd.test_post_accumulate_grad_hook_gets_cleaned_up │ ├── TestAwait.test_await_python │ ├── TestBackends.test_execution │ ├── TestBackends.test_save_load │ ├── TestBackendsWithCompiler.test_execution │ ├── TestBasicsCPU.test_invalid_sparse_coo_values_cpu │ ├── TestBasicsCPU.test_invalid_sparse_csr_values_cpu │ ├── TestBinaryUfuncsCPU.test_add_cpu │ ├── TestBinaryUfuncsCPU.test_int_tensor_pow_neg_ints_cpu │ ├── TestBinaryUfuncsCPU.test_long_tensor_pow_floats_cpu │ ├── TestBinaryUfuncsCPU.test_shift_limits_cpu_uint8 │ ├── TestCheckpoint.test_checkpoint_trigger │ ├── TestConfigModule.test_env_name_semantics │ ├── TestConstant.test_check_constant │ ├── TestConvolutionNN.test_Conv1d_module_same_padding │ ├── TestConvolutionNN.test_Conv2d_backward_twice │ ├── TestConvolutionNN.test_Conv2d_module_same_padding │ ├── TestConvolutionNN.test_Conv3d_module_same_padding │ ├── TestConvolutionNN.test_ConvTranspose2d_output_size_downsample_upsample │ ├── TestConvolutionNN.test_ConvTranspose3d_correct_output_size │ ├── TestConvolutionNNDeviceTypeCPU.test_conv2d_no_grad_cpu_float32 │ ├── TestCppExtensionJIT.test_cpp_frontend_module_has_same_output_as_python │ ├── TestCustomOp.test_impl_device_function │ ├── TestCustomOpTestingCPU.test_opcheck_fails_basic_cpu │ ├── TestDLPack.test_non_contiguous │ ├── TestDtypeAnalysis.test_custom_rules │ ├── TestDtypeAnalysis.test_unary │ ├── TestExperiment.test_mark_strict_with_container_type │ ├── TestExperimentalUtils.test_profiler_for_loop_indexing_pattern │ ├── TestFFTShift.test_fft_n │ ├── TestFX.test_annotations_empty_tuple │ ├── TestFX.test_assert │ ├── TestFX.test_custom_traceback_raised_when_exception_source_is_graphmodule │ ├── TestFXExperimental.test_optimize_for_inference_cpu │ ├── TestFXExperimental.test_optimize_for_inference_cpu_torchvision │ ├── TestForeachCPU.test_add_scalar_with_empty_list_and_empty_tensor_cpu_int16 │ ├── TestForeachCPU.test_add_scalar_with_empty_list_and_empty_tensor_cpu_int32 │ ├── TestForeachCPU.test_add_scalar_with_empty_list_and_empty_tensor_cpu_int64 │ ├── TestForeachCPU.test_add_scalar_with_empty_list_and_empty_tensor_cpu_int8 │ ├── TestForeachCPU.test_add_scalar_with_empty_list_and_empty_tensor_cpu_uint8 │ ├── TestFreezing.test_freeze_module_with_fork2 │ ├── TestFreezing.test_freeze_module_with_fork_calling_module_method │ ├── TestFrozenOptimizations.test_collapse_adjacent_conversions │ ├── TestFrozenOptimizations.test_conv_add_folding │ ├── TestFrozenOptimizations.test_conv_bn_folding │ ├── TestGenericProxyTensorFake.test_amp_cache │ ├── TestGenericProxyTensorFake.test_inplace_metadata │ ├── TestGenericProxyTensorReal.test_inplace_metadata │ ├── TestGenericProxyTensorSymbolic.test_amp_cache │ ├── TestGenericProxyTensorSymbolic.test_inplace_metadata │ ├── TestHistogramdd.test_bins_array │ ├── TestIndexingCPU.test_invalid_index_cpu │ ├── TestJitGeneratedModule.test_nn_Conv1d │ ├── TestJitGeneratedModule.test_nn_Conv1d_circular_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv1d_dilated │ ├── TestJitGeneratedModule.test_nn_Conv1d_groups │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad1 │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad1size1 │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad2size1 │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad_same │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad_same2 │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad_same_dilated │ ├── TestJitGeneratedModule.test_nn_Conv1d_pad_valid │ ├── TestJitGeneratedModule.test_nn_Conv1d_reflect_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv1d_replicate_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv1d_stride │ ├── TestJitGeneratedModule.test_nn_Conv1d_zero_batch │ ├── TestJitGeneratedModule.test_nn_Conv1d_zeros_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv2d │ ├── TestJitGeneratedModule.test_nn_Conv2d_circular_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv2d_depthwise │ ├── TestJitGeneratedModule.test_nn_Conv2d_depthwise_dilated │ ├── TestJitGeneratedModule.test_nn_Conv2d_depthwise_padded │ ├── TestJitGeneratedModule.test_nn_Conv2d_depthwise_strided │ ├── TestJitGeneratedModule.test_nn_Conv2d_depthwise_with_multiplier │ ├── TestJitGeneratedModule.test_nn_Conv2d_dilated │ ├── TestJitGeneratedModule.test_nn_Conv2d_groups │ ├── TestJitGeneratedModule.test_nn_Conv2d_groups_thnn │ ├── TestJitGeneratedModule.test_nn_Conv2d_pad_same │ ├── TestJitGeneratedModule.test_nn_Conv2d_pad_same_dilated │ ├── TestJitGeneratedModule.test_nn_Conv2d_pad_valid │ ├── TestJitGeneratedModule.test_nn_Conv2d_padding │ ├── TestJitGeneratedModule.test_nn_Conv2d_reflect_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv2d_replicate_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv2d_strided │ ├── TestJitGeneratedModule.test_nn_Conv2d_zero_batch │ ├── TestJitGeneratedModule.test_nn_Conv2d_zeros_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv3d │ ├── TestJitGeneratedModule.test_nn_Conv3d_circular_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv3d_dilated │ ├── TestJitGeneratedModule.test_nn_Conv3d_dilated_strided │ ├── TestJitGeneratedModule.test_nn_Conv3d_groups │ ├── TestJitGeneratedModule.test_nn_Conv3d_pad_same │ ├── TestJitGeneratedModule.test_nn_Conv3d_pad_same_dilated │ ├── TestJitGeneratedModule.test_nn_Conv3d_pad_valid │ ├── TestJitGeneratedModule.test_nn_Conv3d_replicate_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_Conv3d_stride │ ├── TestJitGeneratedModule.test_nn_Conv3d_stride_padding │ ├── TestJitGeneratedModule.test_nn_Conv3d_zero_batch │ ├── TestJitGeneratedModule.test_nn_Conv3d_zeros_stride2_pad2 │ ├── TestJitGeneratedModule.test_nn_ConvTranspose1d │ ├── TestJitGeneratedModule.test_nn_ConvTranspose1d_dilated │ ├── TestJitGeneratedModule.test_nn_ConvTranspose1d_groups │ ├── TestJitGeneratedModule.test_nn_ConvTranspose2d │ ├── TestJitGeneratedModule.test_nn_ConvTranspose2d_groups │ ├── TestJitGeneratedModule.test_nn_ConvTranspose3d │ ├── TestJitGeneratedModule.test_nn_ConvTranspose3d_dilated │ ├── TestLazyDynamicOps.test_nonzero_dynamic │ ├── TestLazyReuseIr.testAdd │ ├── TestLazyReuseIr.testAddSub │ ├── TestLazyReuseIr.testAddSubFallback │ ├── TestLazyReuseIr.testBatchNorm │ ├── TestLazyTensor.test_tensor_ctr │ ├── TestLazyTensor.test_view_mark_step_preserved │ ├── TestLinalgCPU.test_lobpcg_torchscript_cpu_float64 │ ├── TestList.test_comprehension_iterable │ ├── TestLoadStateDict.test_load_state_dict_BC_swap_True │ ├── TestLogging.test_trace_numeric_counter │ ├── TestMKLDNNReinplacing.test_always_alive_values │ ├── TestMatmulOperator.test_exceptions │ ├── TestMatmulOperator.test_matmul_raises │ ├── TestMethods.test_compress │ ├── TestMethods.test_dot_out_mem_overlap │ ├── TestMethods.test_matmul_out │ ├── TestMethods.test_partition_iterative │ ├── TestMethods.test_round │ ├── TestMethods.test_searchsorted_complex │ ├── TestMethods.test_searchsorted_type_specific_2 │ ├── TestMethods.test_searchsorted_with_invalid_sorter │ ├── TestMethods.test_transpose │ ├── TestMisc.test_broadcasting_list │ ├── TestMisc.test_parse_ir_annotate │ ├── TestMisc.test_parse_ir_single_element_tensor_negative │ ├── TestMisc.test_parse_ir_single_element_tensor_positive │ ├── TestMkldnnFusion.test_single_conv │ ├── TestModels.test_snli_quantized │ ├── TestModels.test_time_sequence_prediction │ ├── TestModels.test_vae_quantized │ ├── TestNN.test_padding_list │ ├── TestNN.test_vector_to_parameters │ ├── TestNNDeviceTypeCPU.test_conv_empty_input_cpu_complex128 │ ├── TestNNParametrization.test_new_spectral_norm_dim_swap_False │ ├── TestNNParametrization.test_new_spectral_norm_dim_swap_True │ ├── TestNativeFunctions.test_intlist_error_with_overload │ ├── TestNnapiBackend.test_avg_pool2d │ ├── TestNnapiBackend.test_cat │ ├── TestNnapiBackend.test_compile_spec_santiy │ ├── TestNnapiBackend.test_conv2d │ ├── TestNnapiBackend.test_conv2d_transpose │ ├── TestNnapiBackend.test_dequantize │ ├── TestNnapiBackend.test_detach │ ├── TestNnapiBackend.test_flatten │ ├── TestNnapiBackend.test_hardtanh │ ├── TestNnapiBackend.test_linear │ ├── TestNnapiBackend.test_log_softmax │ ├── TestNnapiBackend.test_max_pool2d │ ├── TestNnapiBackend.test_mean │ ├── TestNnapiBackend.test_multi_output │ ├── TestNnapiBackend.test_pointwise_binary │ ├── TestNnapiBackend.test_pointwise_binary_const │ ├── TestNnapiBackend.test_pointwise_unary │ ├── TestNnapiBackend.test_prelu │ ├── TestNnapiBackend.test_qadd │ ├── TestNnapiBackend.test_qlinear │ ├── TestNnapiBackend.test_quantize │ ├── TestNnapiBackend.test_reshape │ ├── TestNnapiBackend.test_seblock_mul │ ├── TestNnapiBackend.test_slice │ ├── TestNnapiBackend.test_softmax │ ├── TestNnapiBackend.test_tensor_input │ ├── TestNnapiBackend.test_to │ ├── TestNnapiBackend.test_unsqueeze │ ├── TestNnapiBackend.test_upsample_nearest2d │ ├── TestOpenMP_ParallelFor.test_one_thread │ ├── TestOperatorsCPU.test_extremal_numerics_l1_loss_cpu │ ├── TestOperatorsCPU.test_extremal_numerics_nll_loss_cpu │ ├── TestPackageScript.test_load_shared_tensors_repackaged │ ├── TestParametrization.test_traceable │ ├── TestPeephole.test_peephole_int │ ├── TestPeephole.test_peephole_optional_refine │ ├── TestProfiler.test_profiler_metadata │ ├── TestProfiler.test_tensorboard_trace_handler │ ├── TestProfilerTree.test_profiler_experimental_tree │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_H_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_MapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NestedMapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpyCatCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpyCubeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpyMulCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpyNMSCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpyNonzeroCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpySortCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpySplitCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpySplitCopyWithIntCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpyTakeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_NumpyViewCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_TripleNestedMapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___getitem___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___radd___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___rdiv___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___rmatmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___rmod___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___rmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___rpow___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive___rsub___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive__native_batch_norm_legit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive__segment_reduce_lengths_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive__segment_reduce_offsets_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive__softmax_backward_data_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive__upsample_bilinear2d_aa_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_abs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_acos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_acosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_addbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_addcdiv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_addcmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_addmm_decomposed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_addmv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_addr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_all_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_allclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_aminmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_angle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_any_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_arange_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_argsort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_argwhere_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_as_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_as_strided_partial_views_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_as_strided_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_asin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_asinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_atan2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_atan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_atanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_atleast_1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_atleast_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_atleast_3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_baddbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_batch_norm_with_update_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_bernoulli_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_bfloat16_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_block_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_bmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_bool_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_broadcast_shapes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_broadcast_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_broadcast_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_bucketize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_byte_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cartesian_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cauchy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cdouble_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ceil_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cfloat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_chalf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_char_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cholesky_inverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cholesky_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_clamp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_clamp_max_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_clamp_min_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_clone_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_column_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_combinations_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_conj_physical_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_constant_pad_nd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_contiguous_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_copysign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_corrcoef_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_count_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cov_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cummax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cummin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_cumulative_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_deg2rad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_diag_embed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_diagflat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_diagonal_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_diagonal_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_diff_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_digamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_div_floor_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_div_no_rounding_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_div_trunc_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_double_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_dsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_dstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_einsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_empty_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_empty_permuted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_eq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_equal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_erf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_erfc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_erfinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_exp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_expand_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_expand_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_expm1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_eye_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_fft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_fft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_fftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_fftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_hfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_hfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_hfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_ifft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_ifft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_ifftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_ifftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_ihfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_ihfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_ihfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_irfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_irfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_irfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_rfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_rfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fft_rfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_flatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_flip_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fliplr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_flipud_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_float_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_float_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_floor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_floor_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_fmod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_frac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_frexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_full_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_gather_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ge_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_geometric_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_geqrf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_gradient_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_grid_sampler_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_gt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_half_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_heaviside_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_histc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_histogram_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_histogramdd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_hsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_hstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_hypot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_igamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_igammac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_index_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_index_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_index_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_index_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_index_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_index_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_inner_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_int_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isfinite_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isnan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isneginf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isposinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_isreal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_item_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_jiterator_2inputs_2outputs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_jiterator_4inputs_with_extra_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_jiterator_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_jiterator_binary_return_by_ref_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_jiterator_unary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_kron_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_kthvalue_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ldexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_le_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_lerp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_lgamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_cholesky_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_cond_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_det_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_det_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_eig_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_eigh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_eigvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_eigvalsh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_householder_product_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_inv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_inv_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_ldl_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_ldl_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_ldl_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_lstsq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_lstsq_grad_oriented_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_lu_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_lu_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_matrix_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_matrix_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_matrix_rank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_matrix_rank_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_multi_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_norm_subgradients_at_zero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_pinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_pinv_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_pinv_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_slogdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_solve_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_solve_triangular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_svdvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_tensorinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_tensorsolve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_vander_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_vecdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linalg_vector_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_linspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_log10_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_log1p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_log2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_log_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_log_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_log_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logaddexp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logcumsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logical_and_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logical_not_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logical_or_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logical_xor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_long_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_lt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_lu_unpack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mH_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mT_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_masked_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_matmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_matrix_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_max_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_max_pool2d_with_indices_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_max_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_max_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_maximum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_meshgrid_list_of_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_meshgrid_variadic_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_min_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_min_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_min_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_minimum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_movedim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_msort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_multinomial_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mvlgamma_mvlgamma_p_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mvlgamma_mvlgamma_p_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_mvlgamma_mvlgamma_p_5_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nan_to_num_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nanmean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nanmedian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nanquantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nansum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_narrow_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_narrow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_native_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_native_dropout_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_native_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ne_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_new_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_new_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_new_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_new_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_new_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nextafter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_adaptive_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_adaptive_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_adaptive_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_adaptive_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_adaptive_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_adaptive_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_alpha_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_binary_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_binary_cross_entropy_with_logits_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_celu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_conv1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_conv2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_conv3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_conv_transpose1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_conv_transpose2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_conv_transpose3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_cosine_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_cosine_similarity_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_ctc_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_dropout2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_dropout3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_elu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_embedding_bag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_embedding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_feature_alpha_dropout_with_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_feature_alpha_dropout_without_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_fractional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_fractional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_gaussian_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_gelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_glu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_grid_sample_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_group_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_hardshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_hardsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_hardswish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_hardtanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_hinge_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_huber_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_instance_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_interpolate_area_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_interpolate_bicubic_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_interpolate_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_interpolate_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_interpolate_nearest-exact_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_interpolate_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_interpolate_trilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_kl_div_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_leaky_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_local_response_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_logsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_margin_ranking_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_unpool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_unpool1d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_unpool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_unpool2d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_unpool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_max_unpool3d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_mish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_mse_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_multi_head_attention_forward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_multi_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_multilabel_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_multilabel_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pad_circular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pad_constant_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pad_reflect_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pad_replicate_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pad_replicate_negative_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pairwise_distance_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pixel_shuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_pixel_unshuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_poisson_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_prelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_relu6_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_rrelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_scaled_dot_product_attention_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_selu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_silu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_smooth_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_softmin_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_softplus_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_softshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_softsign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_tanhshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_threshold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_triplet_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_triplet_margin_with_distance_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_upsample_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nn_functional_upsample_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_nonzero_static_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_norm_fro_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_norm_inf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_norm_nuc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_normal_in_place_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_normal_number_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ones_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ormqr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_outer_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_pca_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_permute_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_pinverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_polar_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_polygamma_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_polygamma_polygamma_n_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_polygamma_polygamma_n_2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_polygamma_polygamma_n_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_polygamma_polygamma_n_4_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_positive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_pow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_quantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_rad2deg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_rand_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_randint_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_randint_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_randn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_randn_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_ravel_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_real_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_reciprocal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_remainder_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_renorm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_repeat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_repeat_interleave_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_reshape_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_reshape_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_resize__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_resize_as__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_resolve_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_resolve_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_roll_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_rot90_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_round_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_round_decimals_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_round_decimals_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_round_decimals_neg_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_rsqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_rsub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scalar_tensor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scatter_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scatter_reduce_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scatter_reduce_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scatter_reduce_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scatter_reduce_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_scatter_reduce_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_searchsorted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_select_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sgn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_short_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_bartlett_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_blackman_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_gaussian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_general_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_general_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_hann_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_kaiser_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signal_windows_nuttall_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_signbit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sinc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_slice_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_slice_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sparse_mm_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sparse_sampled_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_airy_ai_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_bessel_j1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_bessel_y0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_bessel_y1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_entr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_erfcx_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_hermite_polynomial_h_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_hermite_polynomial_he_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_i0e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_i1e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_laguerre_polynomial_l_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_legendre_polynomial_p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_log_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_modified_bessel_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_modified_bessel_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_ndtri_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_polygamma_special_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_scaled_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_scaled_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_shifted_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_shifted_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_shifted_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_shifted_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_spherical_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_xlog1py_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_special_zeta_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_split_list_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_split_with_sizes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_square_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_squeeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_squeeze_multiple_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_std_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_std_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_std_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_stft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_sum_to_size_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_svd_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_take_along_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_take_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_tan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_tanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_tensor_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_tensordot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_tile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_to_sparse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_topk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_trace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_transpose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_trapz_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_triangular_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_tril_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_triu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_true_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_trunc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unbind_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unflatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unfold_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_uniform_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unique_consecutive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unique_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unsafe_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unsafe_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_unsqueeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_var_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_var_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_var_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_vdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_view_as_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_view_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_view_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_view_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_vsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_vstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_where_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_xlogy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_zero__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_exhaustive_zeros_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_H_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_MapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NestedMapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpyCatCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpyCubeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpyMulCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpyNMSCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpyNonzeroCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpySortCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpySplitCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpySplitCopyWithIntCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpyTakeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_NumpyViewCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_TripleNestedMapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___getitem___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___radd___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___rdiv___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___rmatmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___rmod___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___rmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___rpow___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive___rsub___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive__native_batch_norm_legit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive__segment_reduce_lengths_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive__segment_reduce_offsets_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive__softmax_backward_data_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive__upsample_bilinear2d_aa_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_abs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_acos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_acosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_addbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_addcdiv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_addcmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_addmm_decomposed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_addmv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_addr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_all_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_allclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_aminmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_angle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_any_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_arange_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_argsort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_argwhere_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_as_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_as_strided_partial_views_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_as_strided_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_asin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_asinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_atan2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_atan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_atanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_atleast_1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_atleast_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_atleast_3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_baddbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_batch_norm_with_update_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_bernoulli_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_bfloat16_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_block_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_bmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_bool_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_broadcast_shapes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_broadcast_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_broadcast_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_bucketize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_byte_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cartesian_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cauchy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cdouble_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ceil_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cfloat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_chalf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_char_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cholesky_inverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cholesky_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_clamp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_clamp_max_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_clamp_min_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_clone_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_column_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_combinations_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_conj_physical_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_constant_pad_nd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_contiguous_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_copysign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_corrcoef_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_count_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cov_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cummax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cummin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_cumulative_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_deg2rad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_diag_embed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_diagflat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_diagonal_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_diagonal_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_diff_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_digamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_dist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_div_floor_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_div_no_rounding_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_div_trunc_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_double_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_dsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_dstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_einsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_empty_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_empty_permuted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_eq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_equal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_erf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_erfc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_erfinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_exp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_expand_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_expand_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_expm1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_eye_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_fft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_fft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_fftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_fftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_hfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_hfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_hfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_ifft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_ifft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_ifftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_ifftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_ihfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_ihfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_ihfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_irfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_irfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_irfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_rfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_rfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fft_rfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_flatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_flip_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fliplr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_flipud_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_float_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_float_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_floor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_floor_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_fmod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_frac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_frexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_full_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_gather_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ge_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_geometric_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_geqrf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_gradient_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_grid_sampler_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_gt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_half_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_heaviside_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_histc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_histogram_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_histogramdd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_hsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_hstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_hypot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_igamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_igammac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_index_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_index_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_index_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_index_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_index_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_index_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_inner_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_int_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isfinite_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isnan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isneginf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isposinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_isreal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_item_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_jiterator_2inputs_2outputs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_jiterator_4inputs_with_extra_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_jiterator_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_jiterator_binary_return_by_ref_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_jiterator_unary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_kron_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_kthvalue_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ldexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_le_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_lerp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_lgamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_cholesky_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_cond_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_det_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_det_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_eig_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_eigh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_eigvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_eigvalsh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_householder_product_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_inv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_inv_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_ldl_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_ldl_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_ldl_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_lstsq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_lstsq_grad_oriented_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_lu_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_lu_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_matrix_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_matrix_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_matrix_rank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_matrix_rank_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_multi_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_norm_subgradients_at_zero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_pinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_pinv_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_pinv_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_slogdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_solve_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_solve_triangular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_svdvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_tensorinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_tensorsolve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_vander_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_vecdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linalg_vector_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_linspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_log10_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_log1p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_log2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_log_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_log_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_log_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logaddexp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logcumsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logical_and_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logical_not_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logical_or_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logical_xor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_long_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_lt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_lu_unpack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mH_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mT_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_masked_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_matmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_matrix_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_max_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_max_pool2d_with_indices_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_max_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_max_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_maximum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_meshgrid_list_of_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_meshgrid_variadic_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_min_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_min_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_min_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_minimum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_movedim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_msort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_multinomial_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mvlgamma_mvlgamma_p_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mvlgamma_mvlgamma_p_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_mvlgamma_mvlgamma_p_5_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nan_to_num_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nanmean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nanmedian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nanquantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nansum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_narrow_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_narrow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_native_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_native_dropout_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_native_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ne_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_new_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_new_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_new_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_new_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_new_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nextafter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_adaptive_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_adaptive_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_adaptive_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_adaptive_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_adaptive_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_adaptive_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_alpha_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_binary_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_binary_cross_entropy_with_logits_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_celu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_conv1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_conv2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_conv3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_conv_transpose1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_conv_transpose2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_conv_transpose3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_cosine_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_cosine_similarity_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_ctc_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_dropout2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_dropout3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_elu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_embedding_bag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_embedding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_feature_alpha_dropout_with_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_feature_alpha_dropout_without_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_fractional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_fractional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_gaussian_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_gelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_glu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_grid_sample_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_group_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_hardshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_hardsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_hardswish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_hardtanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_hinge_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_huber_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_instance_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_interpolate_area_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_interpolate_bicubic_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_interpolate_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_interpolate_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_interpolate_nearest-exact_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_interpolate_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_interpolate_trilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_kl_div_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_leaky_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_local_response_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_logsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_margin_ranking_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_unpool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_unpool1d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_unpool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_unpool2d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_unpool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_max_unpool3d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_mish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_mse_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_multi_head_attention_forward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_multi_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_multilabel_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_multilabel_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pad_circular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pad_constant_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pad_reflect_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pad_replicate_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pad_replicate_negative_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pairwise_distance_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pixel_shuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_pixel_unshuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_poisson_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_prelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_relu6_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_rrelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_scaled_dot_product_attention_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_selu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_silu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_smooth_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_softmin_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_softplus_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_softshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_softsign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_tanhshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_threshold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_triplet_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_triplet_margin_with_distance_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_upsample_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nn_functional_upsample_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_nonzero_static_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_norm_fro_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_norm_inf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_norm_nuc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_normal_in_place_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_normal_number_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ones_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ormqr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_outer_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_pca_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_permute_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_pinverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_polar_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_polygamma_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_polygamma_polygamma_n_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_polygamma_polygamma_n_2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_polygamma_polygamma_n_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_polygamma_polygamma_n_4_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_positive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_pow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_quantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_rad2deg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_rand_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_randint_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_randint_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_randn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_randn_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_ravel_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_real_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_reciprocal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_remainder_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_renorm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_repeat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_repeat_interleave_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_reshape_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_reshape_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_resize__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_resize_as__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_resolve_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_resolve_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_roll_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_rot90_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_round_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_round_decimals_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_round_decimals_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_round_decimals_neg_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_rsqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_rsub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scalar_tensor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scatter_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scatter_reduce_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scatter_reduce_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scatter_reduce_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scatter_reduce_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_scatter_reduce_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_searchsorted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_select_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sgn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_short_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_bartlett_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_blackman_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_gaussian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_general_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_general_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_hann_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_kaiser_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signal_windows_nuttall_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_signbit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sinc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_slice_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_slice_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sparse_mm_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sparse_sampled_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_airy_ai_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_bessel_j1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_bessel_y0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_bessel_y1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_entr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_erfcx_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_hermite_polynomial_h_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_hermite_polynomial_he_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_i0e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_i1e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_laguerre_polynomial_l_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_legendre_polynomial_p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_log_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_modified_bessel_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_modified_bessel_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_ndtri_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_polygamma_special_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_scaled_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_scaled_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_shifted_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_shifted_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_shifted_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_shifted_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_spherical_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_xlog1py_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_special_zeta_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_split_list_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_split_with_sizes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_square_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_squeeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_squeeze_multiple_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_std_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_std_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_std_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_stft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_sum_to_size_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_svd_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_take_along_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_take_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_tan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_tanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_tensor_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_tensordot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_tile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_to_sparse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_topk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_trace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_transpose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_trapz_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_triangular_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_tril_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_triu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_true_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_trunc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unbind_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unflatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unfold_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_uniform_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unique_consecutive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unique_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unsafe_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unsafe_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_unsqueeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_var_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_var_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_var_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_vdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_view_as_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_view_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_view_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_view_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_vsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_vstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_where_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_xlogy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_zero__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_fake_exhaustive_zeros_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_H_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_MapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NestedMapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpyCatCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpyCubeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpyMulCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpyNMSCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpyNonzeroCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpySortCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpySplitCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpySplitCopyWithIntCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpyTakeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_NumpyViewCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_TripleNestedMapControlflowOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___getitem___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___radd___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___rdiv___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___rmatmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___rmod___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___rmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___rpow___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive___rsub___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive__native_batch_norm_legit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive__segment_reduce_lengths_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive__segment_reduce_offsets_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive__softmax_backward_data_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive__upsample_bilinear2d_aa_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_abs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_acos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_acosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_addbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_addcdiv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_addcmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_addmm_decomposed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_addmv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_addr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_all_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_allclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_aminmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_angle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_any_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_arange_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_argsort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_argwhere_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_as_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_as_strided_partial_views_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_as_strided_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_asin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_asinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_atan2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_atan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_atanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_atleast_1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_atleast_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_atleast_3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_baddbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_batch_norm_with_update_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_bernoulli_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_bfloat16_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_block_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_bmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_bool_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_broadcast_shapes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_broadcast_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_broadcast_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_bucketize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_byte_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cartesian_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cauchy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cdouble_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ceil_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cfloat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_chalf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_char_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cholesky_inverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cholesky_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_clamp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_clamp_max_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_clamp_min_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_clone_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_column_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_combinations_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_conj_physical_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_constant_pad_nd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_contiguous_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_copysign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_corrcoef_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_count_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cov_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cummax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cummin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_cumulative_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_deg2rad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_diag_embed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_diagflat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_diagonal_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_diagonal_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_diff_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_digamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_div_floor_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_div_no_rounding_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_div_trunc_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_double_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_dsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_dstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_einsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_empty_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_empty_permuted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_eq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_equal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_erf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_erfc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_erfinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_exp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_expand_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_expand_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_expm1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_eye_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_fft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_fft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_fftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_fftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_hfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_hfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_hfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_ifft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_ifft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_ifftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_ifftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_ihfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_ihfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_ihfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_irfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_irfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_irfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_rfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_rfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fft_rfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_flatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_flip_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fliplr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_flipud_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_float_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_float_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_floor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_floor_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_fmod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_frac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_frexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_full_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_gather_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ge_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_geometric_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_geqrf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_gradient_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_grid_sampler_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_gt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_half_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_heaviside_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_histc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_histogram_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_histogramdd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_hsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_hstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_hypot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_igamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_igammac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_index_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_index_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_index_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_index_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_index_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_index_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inner_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_H_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpyCatCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpyCubeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpyMulCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpyNMSCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpyNonzeroCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpySortCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpySplitCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpySplitCopyWithIntCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpyTakeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_NumpyViewCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___getitem___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___radd___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___rdiv___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___rmatmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___rmod___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___rmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___rpow___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace___rsub___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace__native_batch_norm_legit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace__segment_reduce_lengths_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace__segment_reduce_offsets_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace__softmax_backward_data_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace__upsample_bilinear2d_aa_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_abs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_acos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_acosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_addbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_addcdiv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_addcmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_addmm_decomposed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_addmv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_addr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_all_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_allclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_aminmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_angle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_any_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_arange_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_argsort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_argwhere_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_as_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_as_strided_partial_views_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_as_strided_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_asin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_asinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_atan2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_atan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_atanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_atleast_1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_atleast_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_atleast_3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_baddbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_batch_norm_with_update_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_bernoulli_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_bfloat16_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_block_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_bmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_bool_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_broadcast_shapes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_broadcast_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_broadcast_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_bucketize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_byte_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cartesian_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cauchy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cdouble_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ceil_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cfloat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_chalf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_char_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cholesky_inverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cholesky_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_clamp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_clamp_max_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_clamp_min_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_clone_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_column_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_combinations_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_conj_physical_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_constant_pad_nd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_contiguous_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_copysign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_corrcoef_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_count_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cov_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cummax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cummin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_cumulative_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_deg2rad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_diag_embed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_diagflat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_diagonal_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_diagonal_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_diff_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_digamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_dist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_div_floor_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_div_no_rounding_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_div_trunc_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_double_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_dsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_dstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_einsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_empty_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_empty_permuted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_eq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_equal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_erf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_erfc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_erfinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_exp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_expand_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_expand_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_expm1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_eye_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_fft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_fft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_fftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_fftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_hfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_hfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_hfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_ifft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_ifft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_ifftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_ifftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_ihfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_ihfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_ihfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_irfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_irfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_irfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_rfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_rfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fft_rfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_flatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_flip_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fliplr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_flipud_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_float_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_float_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_floor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_floor_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_fmod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_frac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_frexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_full_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_gather_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ge_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_geometric_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_geqrf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_gradient_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_grid_sampler_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_gt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_half_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_heaviside_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_histc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_histogram_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_histogramdd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_hsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_hstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_hypot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_igamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_igammac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_index_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_index_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_index_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_index_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_index_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_index_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_inner_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_int_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isfinite_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isnan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isneginf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isposinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_isreal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_item_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_jiterator_2inputs_2outputs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_jiterator_4inputs_with_extra_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_jiterator_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_jiterator_binary_return_by_ref_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_jiterator_unary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_kron_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_kthvalue_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ldexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_le_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_lerp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_lgamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_cholesky_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_cond_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_det_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_det_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_eig_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_eigh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_eigvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_eigvalsh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_householder_product_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_inv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_inv_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_ldl_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_ldl_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_ldl_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_lstsq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_lstsq_grad_oriented_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_lu_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_lu_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_matrix_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_matrix_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_matrix_rank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_matrix_rank_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_multi_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_norm_subgradients_at_zero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_pinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_pinv_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_pinv_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_slogdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_solve_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_solve_triangular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_svdvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_tensorinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_tensorsolve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_vander_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_vecdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linalg_vector_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_linspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_log10_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_log1p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_log2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_log_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_log_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_log_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logaddexp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logcumsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logical_and_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logical_not_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logical_or_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logical_xor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_long_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_lt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_lu_unpack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mH_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mT_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_masked_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_matmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_matrix_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_max_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_max_pool2d_with_indices_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_max_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_max_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_maximum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_meshgrid_list_of_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_meshgrid_variadic_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_min_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_min_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_min_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_minimum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_movedim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_msort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_multinomial_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mvlgamma_mvlgamma_p_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mvlgamma_mvlgamma_p_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_mvlgamma_mvlgamma_p_5_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nan_to_num_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nanmean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nanmedian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nanquantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nansum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_narrow_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_narrow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_native_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_native_dropout_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_native_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ne_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_new_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_new_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_new_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_new_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_new_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nextafter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_adaptive_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_adaptive_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_adaptive_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_adaptive_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_adaptive_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_adaptive_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_alpha_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_binary_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_binary_cross_entropy_with_logits_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_celu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_conv1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_conv2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_conv3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_conv_transpose1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_conv_transpose2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_conv_transpose3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_cosine_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_cosine_similarity_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_ctc_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_dropout2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_dropout3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_elu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_embedding_bag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_embedding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_feature_alpha_dropout_with_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_feature_alpha_dropout_without_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_fractional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_fractional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_gaussian_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_gelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_glu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_grid_sample_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_group_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_hardshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_hardsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_hardswish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_hardtanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_hinge_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_huber_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_instance_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_interpolate_area_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_interpolate_bicubic_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_interpolate_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_interpolate_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_interpolate_nearest-exact_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_interpolate_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_interpolate_trilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_kl_div_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_leaky_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_local_response_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_logsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_margin_ranking_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_unpool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_unpool1d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_unpool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_unpool2d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_unpool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_max_unpool3d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_mish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_mse_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_multi_head_attention_forward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_multi_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_multilabel_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_multilabel_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pad_circular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pad_constant_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pad_reflect_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pad_replicate_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pad_replicate_negative_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pairwise_distance_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pixel_shuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_pixel_unshuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_poisson_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_prelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_relu6_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_rrelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_scaled_dot_product_attention_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_selu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_silu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_smooth_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_softmin_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_softplus_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_softshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_softsign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_tanhshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_threshold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_triplet_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_triplet_margin_with_distance_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_upsample_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nn_functional_upsample_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_nonzero_static_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_norm_fro_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_norm_inf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_norm_nuc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_normal_in_place_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_normal_number_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ones_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ormqr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_outer_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_pca_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_permute_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_pinverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_polar_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_polygamma_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_polygamma_polygamma_n_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_polygamma_polygamma_n_2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_polygamma_polygamma_n_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_polygamma_polygamma_n_4_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_positive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_pow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_quantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_rad2deg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_rand_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_randint_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_randint_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_randn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_randn_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_ravel_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_real_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_reciprocal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_remainder_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_renorm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_repeat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_repeat_interleave_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_reshape_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_reshape_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_resize__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_resize_as__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_resolve_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_resolve_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_roll_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_rot90_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_round_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_round_decimals_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_round_decimals_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_round_decimals_neg_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_rsqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_rsub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scalar_tensor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scatter_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scatter_reduce_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scatter_reduce_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scatter_reduce_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scatter_reduce_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_scatter_reduce_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_searchsorted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_select_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sgn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_short_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_bartlett_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_blackman_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_gaussian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_general_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_general_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_hann_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_kaiser_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signal_windows_nuttall_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_signbit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sinc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_slice_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_slice_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sparse_mm_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sparse_sampled_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_airy_ai_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_bessel_j1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_bessel_y0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_bessel_y1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_entr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_erfcx_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_hermite_polynomial_h_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_hermite_polynomial_he_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_i0e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_i1e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_laguerre_polynomial_l_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_legendre_polynomial_p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_log_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_modified_bessel_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_modified_bessel_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_ndtri_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_polygamma_special_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_scaled_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_scaled_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_shifted_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_shifted_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_shifted_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_shifted_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_spherical_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_xlog1py_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_special_zeta_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_split_list_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_split_with_sizes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_square_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_squeeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_squeeze_multiple_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_std_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_std_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_std_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_stft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_sum_to_size_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_svd_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_take_along_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_take_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_tan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_tanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_tensor_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_tensordot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_tile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_to_sparse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_topk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_trace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_transpose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_trapz_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_triangular_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_tril_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_triu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_true_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_trunc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unbind_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unflatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unfold_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_uniform_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unique_consecutive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unique_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unsafe_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unsafe_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_unsqueeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_var_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_var_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_var_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_vdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_view_as_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_view_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_view_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_view_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_vsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_vstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_where_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_xlogy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_zero__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_inplace_zeros_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_int_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isfinite_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isnan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isneginf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isposinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_isreal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_item_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_jiterator_2inputs_2outputs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_jiterator_4inputs_with_extra_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_jiterator_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_jiterator_binary_return_by_ref_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_jiterator_unary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_kron_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_kthvalue_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ldexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_le_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_lerp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_lgamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_cholesky_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_cond_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_det_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_det_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_eig_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_eigh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_eigvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_eigvalsh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_householder_product_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_inv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_inv_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_ldl_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_ldl_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_ldl_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_lstsq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_lstsq_grad_oriented_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_lu_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_lu_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_matrix_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_matrix_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_matrix_rank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_matrix_rank_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_multi_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_norm_subgradients_at_zero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_pinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_pinv_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_pinv_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_slogdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_solve_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_solve_triangular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_svdvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_tensorinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_tensorsolve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_vander_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_vecdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linalg_vector_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_linspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_log10_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_log1p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_log2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_log_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_log_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_log_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logaddexp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logcumsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logical_and_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logical_not_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logical_or_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logical_xor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_long_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_lt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_lu_unpack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mH_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mT_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_masked_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_matmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_matrix_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_max_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_max_pool2d_with_indices_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_max_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_max_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_maximum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_meshgrid_list_of_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_meshgrid_variadic_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_min_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_min_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_min_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_minimum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_movedim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_msort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_multinomial_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mvlgamma_mvlgamma_p_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mvlgamma_mvlgamma_p_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_mvlgamma_mvlgamma_p_5_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nan_to_num_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nanmean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nanmedian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nanquantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nansum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_narrow_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_narrow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_native_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_native_dropout_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_native_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ne_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_new_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_new_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_new_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_new_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_new_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nextafter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_adaptive_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_adaptive_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_adaptive_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_adaptive_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_adaptive_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_adaptive_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_alpha_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_binary_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_binary_cross_entropy_with_logits_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_celu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_conv1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_conv2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_conv3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_conv_transpose1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_conv_transpose2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_conv_transpose3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_cosine_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_cosine_similarity_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_ctc_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_dropout2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_dropout3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_elu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_embedding_bag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_embedding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_feature_alpha_dropout_with_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_feature_alpha_dropout_without_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_fractional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_fractional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_gaussian_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_gelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_glu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_grid_sample_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_group_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_hardshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_hardsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_hardswish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_hardtanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_hinge_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_huber_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_instance_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_interpolate_area_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_interpolate_bicubic_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_interpolate_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_interpolate_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_interpolate_nearest-exact_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_interpolate_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_interpolate_trilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_kl_div_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_leaky_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_local_response_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_logsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_margin_ranking_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_unpool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_unpool1d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_unpool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_unpool2d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_unpool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_max_unpool3d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_mish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_mse_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_multi_head_attention_forward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_multi_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_multilabel_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_multilabel_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pad_circular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pad_constant_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pad_reflect_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pad_replicate_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pad_replicate_negative_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pairwise_distance_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pixel_shuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_pixel_unshuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_poisson_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_prelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_relu6_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_rrelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_scaled_dot_product_attention_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_selu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_silu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_smooth_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_softmin_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_softplus_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_softshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_softsign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_tanhshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_threshold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_triplet_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_triplet_margin_with_distance_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_upsample_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nn_functional_upsample_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_nonzero_static_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_norm_fro_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_norm_inf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_norm_nuc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_normal_in_place_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_normal_number_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ones_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ormqr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_H_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpyCatCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpyCubeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpyMulCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpyNMSCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpyNonzeroCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpySortCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpySplitCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpySplitCopyWithIntCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpyTakeCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_NumpyViewCopyCustomOp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___getitem___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___radd___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___rdiv___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___rmatmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___rmod___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___rmul___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___rpow___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out___rsub___cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out__native_batch_norm_legit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out__segment_reduce_lengths_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out__segment_reduce_offsets_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out__softmax_backward_data_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out__upsample_bilinear2d_aa_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_abs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_acos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_acosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_addbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_addcdiv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_addcmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_addmm_decomposed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_addmv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_addr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_all_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_allclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_aminmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_angle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_any_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_arange_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_argsort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_argwhere_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_as_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_as_strided_partial_views_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_as_strided_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_asin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_asinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_atan2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_atan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_atanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_atleast_1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_atleast_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_atleast_3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_baddbmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_batch_norm_with_update_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_bernoulli_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_bfloat16_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_block_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_bmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_bool_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_broadcast_shapes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_broadcast_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_broadcast_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_bucketize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_byte_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cartesian_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cauchy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cdouble_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ceil_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cfloat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_chalf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_char_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cholesky_inverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cholesky_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_clamp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_clamp_max_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_clamp_min_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_clone_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_column_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_combinations_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_conj_physical_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_constant_pad_nd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_contiguous_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_copysign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_corrcoef_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cos_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cosh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_count_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cov_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cummax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cummin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_cumulative_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_deg2rad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_diag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_diag_embed_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_diagflat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_diagonal_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_diagonal_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_diff_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_digamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_dist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_div_floor_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_div_no_rounding_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_div_trunc_rounding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_double_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_dsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_dstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_einsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_empty_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_empty_permuted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_eq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_equal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_erf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_erfc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_erfinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_exp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_expand_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_expand_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_expm1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_eye_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_fft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_fft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_fftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_fftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_hfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_hfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_hfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_ifft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_ifft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_ifftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_ifftshift_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_ihfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_ihfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_ihfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_irfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_irfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_irfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_rfft2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_rfft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fft_rfftn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_flatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_flip_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fliplr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_flipud_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_float_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_float_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_floor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_floor_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_fmod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_frac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_frexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_full_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_gather_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ge_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_geometric_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_geqrf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_gradient_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_grid_sampler_2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_gt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_half_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_heaviside_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_histc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_histogram_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_histogramdd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_hsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_hstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_hypot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_igamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_igammac_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_index_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_index_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_index_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_index_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_index_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_index_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_inner_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_int_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isclose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isfinite_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isnan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isneginf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isposinf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_isreal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_item_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_jiterator_2inputs_2outputs_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_jiterator_4inputs_with_extra_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_jiterator_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_jiterator_binary_return_by_ref_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_jiterator_unary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_kron_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_kthvalue_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ldexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_le_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_lerp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_lgamma_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_cholesky_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_cholesky_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_cond_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_cross_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_det_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_det_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_diagonal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_eig_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_eigh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_eigvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_eigvalsh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_householder_product_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_inv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_inv_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_ldl_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_ldl_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_ldl_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_lstsq_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_lstsq_grad_oriented_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_lu_factor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_lu_factor_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_matrix_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_matrix_power_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_matrix_rank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_matrix_rank_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_multi_dot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_norm_subgradients_at_zero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_pinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_pinv_hermitian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_pinv_singular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_slogdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_solve_ex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_solve_triangular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_svdvals_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_tensorinv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_tensorsolve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_vander_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_vecdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linalg_vector_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_linspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_log10_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_log1p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_log2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_log_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_log_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_log_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logaddexp2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logcumsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logdet_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logical_and_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logical_not_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logical_or_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logical_xor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logspace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logspace_tensor_overload_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_long_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_lt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_lu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_lu_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_lu_unpack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mH_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mT_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_argmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_argmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_cumprod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_cumsum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_fill_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_log_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_logaddexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_logsumexp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_masked_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_matmul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_matrix_exp_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_max_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_max_pool2d_with_indices_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_max_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_max_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_maximum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_median_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_meshgrid_list_of_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_meshgrid_variadic_tensors_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_min_binary_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_min_reduction_no_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_min_reduction_with_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_minimum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mode_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_movedim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_msort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mul_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_multinomial_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mv_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mvlgamma_mvlgamma_p_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mvlgamma_mvlgamma_p_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_mvlgamma_mvlgamma_p_5_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nan_to_num_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nanmean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nanmedian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nanquantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nansum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_narrow_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_narrow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_native_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_native_dropout_backward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_native_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ne_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_new_empty_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_new_empty_strided_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_new_full_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_new_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_new_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nextafter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_adaptive_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_adaptive_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_adaptive_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_adaptive_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_adaptive_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_adaptive_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_alpha_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_avg_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_avg_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_avg_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_batch_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_binary_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_binary_cross_entropy_with_logits_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_celu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_conv1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_conv2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_conv3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_conv_transpose1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_conv_transpose2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_conv_transpose3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_cosine_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_cosine_similarity_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_cross_entropy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_ctc_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_dropout2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_dropout3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_dropout_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_elu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_embedding_bag_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_embedding_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_feature_alpha_dropout_with_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_feature_alpha_dropout_without_train_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_fractional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_fractional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_gaussian_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_gelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_glu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_grid_sample_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_group_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_hardshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_hardsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_hardswish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_hardtanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_hinge_embedding_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_huber_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_instance_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_interpolate_area_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_interpolate_bicubic_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_interpolate_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_interpolate_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_interpolate_nearest-exact_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_interpolate_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_interpolate_trilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_kl_div_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_layer_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_leaky_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_linear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_local_response_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_logsigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_margin_ranking_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_pool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_pool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_pool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_unpool1d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_unpool1d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_unpool2d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_unpool2d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_unpool3d_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_max_unpool3d_grad_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_mish_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_mse_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_multi_head_attention_forward_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_multi_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_multilabel_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_multilabel_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_normalize_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pad_circular_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pad_constant_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pad_reflect_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pad_replicate_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pad_replicate_negative_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pairwise_distance_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pdist_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pixel_shuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_pixel_unshuffle_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_poisson_nll_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_prelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_relu6_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_relu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_rrelu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_scaled_dot_product_attention_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_selu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_silu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_smooth_l1_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_soft_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_softmin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_softmin_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_softplus_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_softshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_softsign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_tanhshrink_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_threshold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_triplet_margin_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_triplet_margin_with_distance_loss_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_upsample_bilinear_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nn_functional_upsample_nearest_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nonzero_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_nonzero_static_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_norm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_norm_fro_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_norm_inf_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_norm_nuc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_normal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_normal_in_place_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_normal_number_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ones_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ones_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ormqr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_outer_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_pca_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_permute_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_pinverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_polar_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_polygamma_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_polygamma_polygamma_n_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_polygamma_polygamma_n_2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_polygamma_polygamma_n_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_polygamma_polygamma_n_4_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_positive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_pow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_quantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_rad2deg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_rand_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_randint_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_randint_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_randn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_randn_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_ravel_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_real_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_reciprocal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_remainder_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_renorm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_repeat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_repeat_interleave_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_reshape_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_reshape_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_resize__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_resize_as__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_resolve_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_resolve_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_roll_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_rot90_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_round_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_round_decimals_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_round_decimals_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_round_decimals_neg_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_rsqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_rsub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scalar_tensor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scatter_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scatter_reduce_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scatter_reduce_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scatter_reduce_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scatter_reduce_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_scatter_reduce_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_searchsorted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_select_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sgn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_short_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_bartlett_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_blackman_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_gaussian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_general_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_general_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_hann_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_kaiser_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signal_windows_nuttall_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_signbit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sinc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_slice_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_slice_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sparse_mm_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sparse_sampled_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_airy_ai_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_bessel_j1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_bessel_y0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_bessel_y1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_entr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_erfcx_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_hermite_polynomial_h_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_hermite_polynomial_he_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_i0e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_i1e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_laguerre_polynomial_l_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_legendre_polynomial_p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_log_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_modified_bessel_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_modified_bessel_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_ndtri_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_polygamma_special_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_scaled_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_scaled_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_shifted_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_shifted_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_shifted_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_shifted_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_spherical_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_xlog1py_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_special_zeta_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_split_list_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_split_with_sizes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_square_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_squeeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_squeeze_multiple_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_std_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_std_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_std_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_stft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_sum_to_size_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_svd_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_take_along_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_take_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_tan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_tanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_tensor_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_tensordot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_tile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_to_sparse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_topk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_trace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_transpose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_trapz_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_triangular_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_tril_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_triu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_true_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_trunc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unbind_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unflatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unfold_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_uniform_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unique_consecutive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unique_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unsafe_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unsafe_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_unsqueeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_var_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_var_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_var_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_vdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_view_as_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_view_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_view_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_view_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_vsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_vstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_where_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_xlogy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_zero__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_out_zeros_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_outer_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_pca_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_permute_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_pinverse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_polar_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_polygamma_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_polygamma_polygamma_n_1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_polygamma_polygamma_n_2_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_polygamma_polygamma_n_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_polygamma_polygamma_n_4_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_positive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_pow_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_put_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_qr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_quantile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_rad2deg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_rand_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_randint_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_randint_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_randn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_randn_like_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_ravel_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_real_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_reciprocal_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_remainder_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_renorm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_repeat_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_repeat_interleave_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_reshape_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_reshape_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_resize__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_resize_as__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_resolve_conj_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_resolve_neg_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_roll_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_rot90_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_round_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_round_decimals_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_round_decimals_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_round_decimals_neg_3_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_rsqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_rsub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scalar_tensor_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scatter_add_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scatter_reduce_amax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scatter_reduce_amin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scatter_reduce_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scatter_reduce_prod_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_scatter_reduce_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_searchsorted_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_select_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_select_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sgn_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_short_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sigmoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sign_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_bartlett_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_blackman_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_exponential_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_gaussian_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_general_cosine_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_general_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_hamming_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_hann_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_kaiser_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signal_windows_nuttall_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_signbit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sin_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sinc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sinh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_slice_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_slice_scatter_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_softmax_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_softmax_with_dtype_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sort_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sparse_mm_reduce_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sparse_sampled_addmm_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_airy_ai_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_bessel_j1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_bessel_y0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_bessel_y1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_entr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_erfcx_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_hermite_polynomial_h_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_hermite_polynomial_he_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_i0e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_i1e_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_laguerre_polynomial_l_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_legendre_polynomial_p_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_log_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_modified_bessel_i0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_modified_bessel_i1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_ndtr_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_ndtri_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_polygamma_special_polygamma_n_0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_scaled_modified_bessel_k0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_scaled_modified_bessel_k1_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_shifted_chebyshev_polynomial_t_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_shifted_chebyshev_polynomial_u_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_shifted_chebyshev_polynomial_v_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_shifted_chebyshev_polynomial_w_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_spherical_bessel_j0_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_xlog1py_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_special_zeta_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_split_list_args_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_split_with_sizes_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sqrt_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_square_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_squeeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_squeeze_multiple_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_stack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_std_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_std_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_std_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_std_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_stft_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sub_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sum_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_sum_to_size_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_svd_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_svd_lowrank_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_take_along_dim_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_take_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_tan_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_tanh_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_tensor_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_tensordot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_tile_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_to_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_to_sparse_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_topk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_trace_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_transpose_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_trapezoid_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_trapz_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_triangular_solve_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_tril_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_triu_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_true_divide_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_trunc_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unbind_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unflatten_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unfold_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unfold_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_uniform_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unique_consecutive_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unique_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unsafe_chunk_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unsafe_split_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_unsqueeze_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_var_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_var_mean_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_var_mean_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_var_unbiased_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_vdot_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_view_as_complex_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_view_as_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_view_copy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_view_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_vsplit_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_vstack_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_where_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_xlogy_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_zero__cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_zeros_cpu_float32 │ ├── TestProxyTensorOpInfoCPU.test_make_fx_symbolic_exhaustive_zeros_like_cpu_float32 │ ├── TestPythonAutograd.test_backwards1 │ ├── TestPythonBindings.test_cu_create_function │ ├── TestPythonBuiltinOP.test_stepped_tuple_slicing │ ├── TestPythonDispatch.test_list_ret │ ├── TestPythonRegistration.test_fallback │ ├── TestPythonRegistration.test_fallback_keyset │ ├── TestQuantizedEmbeddingOps.test_embedding_bag_byte │ ├── TestRecursiveScript.test_inner_traced_module │ ├── TestReductionsCPU.test_all_any_vs_numpy_cpu_bool │ ├── TestReductionsCPU.test_all_any_vs_numpy_cpu_uint8 │ ├── TestReductionsCPU.test_histogram_cpu_float32 │ ├── TestReductionsCPU.test_histogramdd_cpu_float32 │ ├── TestReductionsCPU.test_logcumsumexp_complex_cpu_complex128 │ ├── TestReductionsCPU.test_logcumsumexp_complex_cpu_complex64 │ ├── TestReductionsCPU.test_tensor_compare_ops_argmax_argmix_kthvalue_dim_empty_cpu │ ├── TestReductionsCPU.test_tensor_compare_ops_empty_cpu │ ├── TestReductionsCPU.test_tensor_reduce_ops_empty_cpu │ ├── TestRepackage.test_repackage_import_indirectly_via_parent_module │ ├── TestSWAUtils.test_averaged_model_all_devices_ema_True │ ├── TestSWAUtils.test_averaged_model_exponential_use_multi_avg_fn_True_use_buffers_False │ ├── TestSWAUtils.test_averaged_model_exponential_use_multi_avg_fn_True_use_buffers_True │ ├── TestSaveLoadForOpVersion.test_versioned_div_tensor_inplace │ ├── TestSaveLoadForOpVersion.test_versioned_div_tensor_out │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_H_operation0 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_H_operation1 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_I_operation0 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_I_operation1 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_L_operation0 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_L_operation1 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_P_operation0 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_P_operation1 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_Q_operation0 │ ├── TestScalarOpsMisc.test_scalar_integer_operation_divbyzero_dtype_Q_operation1 │ ├── TestScript.test_function_overloading_isinstance │ ├── TestScript.test_function_overloads │ ├── TestScript.test_ignored_as_value │ ├── TestScript.test_namedtuple_python │ ├── TestScript.test_no_self_arg_ignore_function │ ├── TestScript.test_python_call_non_tensor_wrong │ ├── TestScript.test_python_op_builtins │ ├── TestScript.test_type_annotation_module │ ├── TestScript.test_unused_decorator │ ├── TestScript.test_wrong_return_type │ ├── TestScriptProfile.test_script │ ├── TestScriptProfile.test_section │ ├── TestSerializeCustomClass.test_custom_class │ ├── TestSetOps.test_in1d_table_timedelta_fails │ ├── TestSetOps.test_in1d_timedelta_kind0 │ ├── TestSetOps.test_in1d_timedelta_kind_sort │ ├── TestSetOps.test_setdiff1d │ ├── TestSlice.test_tuple_slicing │ ├── TestSparseAnyCPU.test_check_sparse_tensor_invariants_SparseBSC_cpu │ ├── TestSparseAnyCPU.test_check_sparse_tensor_invariants_SparseBSR_cpu │ ├── TestSparseAnyCPU.test_check_sparse_tensor_invariants_SparseCOO_cpu │ ├── TestSparseAnyCPU.test_check_sparse_tensor_invariants_SparseCSC_cpu │ ├── TestSparseAnyCPU.test_check_sparse_tensor_invariants_SparseCSR_cpu │ ├── TestSparseAnyCPU.test_constructor_autograd_SparseBSC_cpu │ ├── TestSparseAnyCPU.test_constructor_autograd_SparseBSR_cpu │ ├── TestSparseAnyCPU.test_constructor_autograd_SparseCOO_cpu │ ├── TestSparseAnyCPU.test_constructor_autograd_SparseCSC_cpu │ ├── TestSparseAnyCPU.test_constructor_autograd_SparseCSR_cpu │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_masked_fast_cpu_complex128 │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_masked_fast_cpu_float64 │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_masked_slow_cpu_complex128 │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_masked_slow_cpu_float64 │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_sparse_fast_cpu_complex128 │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_sparse_fast_cpu_float64 │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_sparse_slow_cpu_complex128 │ ├── TestSparseAnyCPU.test_gradcheck_mm_SparseCOO_sparse_slow_cpu_float64 │ ├── TestSparseCPU.test_add_dense_sparse_mismatch_cpu_complex128 │ ├── TestSparseCPU.test_add_dense_sparse_mismatch_cpu_float64 │ ├── TestSparseCPU.test_add_zeros_cpu_complex128 │ ├── TestSparseCPU.test_add_zeros_cpu_float64 │ ├── TestSparseCPU.test_any_cpu │ ├── TestSparseCPU.test_asin_arcsin_cpu_float32 │ ├── TestSparseCPU.test_asin_arcsin_cpu_float64 │ ├── TestSparseCPU.test_asin_arcsin_cpu_int16 │ ├── TestSparseCPU.test_asin_arcsin_cpu_int32 │ ├── TestSparseCPU.test_asin_arcsin_cpu_int64 │ ├── TestSparseCPU.test_asin_arcsin_cpu_int8 │ ├── TestSparseCPU.test_asin_arcsin_cpu_uint8 │ ├── TestSparseCPU.test_assign_cpu_float64 │ ├── TestSparseCPU.test_basic_cpu_complex128 │ ├── TestSparseCPU.test_basic_cpu_float64 │ ├── TestSparseCPU.test_basic_ops_cpu_float64 │ ├── TestSparseCPU.test_bmm_cpu_float64 │ ├── TestSparseCPU.test_cat_cpu_complex128 │ ├── TestSparseCPU.test_cat_cpu_float64 │ ├── TestSparseCPU.test_change_tensor_metadata_cpu_complex128 │ ├── TestSparseCPU.test_change_tensor_metadata_cpu_float64 │ ├── TestSparseCPU.test_clone_cpu_complex128 │ ├── TestSparseCPU.test_clone_cpu_float64 │ ├── TestSparseCPU.test_coalesce_transpose_mm_cpu_float64 │ ├── TestSparseCPU.test_contig_cpu_complex128 │ ├── TestSparseCPU.test_contig_cpu_float64 │ ├── TestSparseCPU.test_contig_hybrid_cpu_complex128 │ ├── TestSparseCPU.test_contig_hybrid_cpu_float64 │ ├── TestSparseCPU.test_ctor_is_coalesced_with_gradcheck_cpu_float64 │ ├── TestSparseCPU.test_div_rounding_mode_cpu_float32 │ ├── TestSparseCPU.test_div_rounding_mode_cpu_float64 │ ├── TestSparseCPU.test_dsmm_cpu_float64 │ ├── TestSparseCPU.test_empty_like_cpu_complex128 │ ├── TestSparseCPU.test_empty_like_cpu_float64 │ ├── TestSparseCPU.test_factory_copy_cpu │ ├── TestSparseCPU.test_factory_cpu_complex128 │ ├── TestSparseCPU.test_factory_cpu_complex64 │ ├── TestSparseCPU.test_factory_cpu_float16 │ ├── TestSparseCPU.test_factory_cpu_float32 │ ├── TestSparseCPU.test_factory_cpu_float64 │ ├── TestSparseCPU.test_factory_type_inference_cpu_complex128 │ ├── TestSparseCPU.test_factory_type_inference_cpu_complex64 │ ├── TestSparseCPU.test_factory_type_inference_cpu_float16 │ ├── TestSparseCPU.test_factory_type_inference_cpu_float32 │ ├── TestSparseCPU.test_factory_type_inference_cpu_float64 │ ├── TestSparseCPU.test_factory_type_inference_cpu_int64 │ ├── TestSparseCPU.test_hsmm_cpu_float64 │ ├── TestSparseCPU.test_index_select_cpu_complex128 │ ├── TestSparseCPU.test_index_select_cpu_float64 │ ├── TestSparseCPU.test_index_select_empty_and_non_contiguous_index_cpu_complex128 │ ├── TestSparseCPU.test_index_select_empty_and_non_contiguous_index_cpu_float64 │ ├── TestSparseCPU.test_index_select_exhaustive_index_large_cpu_complex128 │ ├── TestSparseCPU.test_index_select_exhaustive_index_large_cpu_float64 │ ├── TestSparseCPU.test_index_select_exhaustive_index_small_cpu_complex128 │ ├── TestSparseCPU.test_index_select_exhaustive_index_small_cpu_float64 │ ├── TestSparseCPU.test_index_select_parallelization_cpu_complex128 │ ├── TestSparseCPU.test_index_select_parallelization_cpu_float64 │ ├── TestSparseCPU.test_is_nonzero_cpu │ ├── TestSparseCPU.test_isnan_cpu │ ├── TestSparseCPU.test_legacy_new_cpu │ ├── TestSparseCPU.test_legacy_new_device_cpu │ ├── TestSparseCPU.test_log1p_cpu_float32 │ ├── TestSparseCPU.test_log1p_cpu_float64 │ ├── TestSparseCPU.test_log1p_cpu_int16 │ ├── TestSparseCPU.test_log1p_cpu_int32 │ ├── TestSparseCPU.test_log1p_cpu_int64 │ ├── TestSparseCPU.test_log1p_cpu_int8 │ ├── TestSparseCPU.test_log1p_cpu_uint8 │ ├── TestSparseCPU.test_mm_cpu_complex128 │ ├── TestSparseCPU.test_mm_cpu_float64 │ ├── TestSparseCPU.test_mv_cpu_float64 │ ├── TestSparseCPU.test_neg_negative_cpu_complex128 │ ├── TestSparseCPU.test_neg_negative_cpu_float64 │ ├── TestSparseCPU.test_new_cpu_complex128 │ ├── TestSparseCPU.test_new_cpu_float64 │ ├── TestSparseCPU.test_norm_cpu_complex128 │ ├── TestSparseCPU.test_norm_cpu_float64 │ ├── TestSparseCPU.test_permute_masked_cpu_complex128 │ ├── TestSparseCPU.test_permute_masked_cpu_float64 │ ├── TestSparseCPU.test_permute_sparse_cpu_complex128 │ ├── TestSparseCPU.test_permute_sparse_cpu_float64 │ ├── TestSparseCPU.test_pickle_cpu_float64 │ ├── TestSparseCPU.test_print_coalesced_cpu_float64 │ ├── TestSparseCPU.test_print_uncoalesced_cpu_float64 │ ├── TestSparseCPU.test_resize_cpu_complex128 │ ├── TestSparseCPU.test_resize_cpu_float64 │ ├── TestSparseCPU.test_saddmm_cpu_complex128 │ ├── TestSparseCPU.test_saddmm_cpu_float64 │ ├── TestSparseCPU.test_select_cpu_complex128 │ ├── TestSparseCPU.test_select_cpu_float64 │ ├── TestSparseCPU.test_select_no_type_promotion_cpu_int16 │ ├── TestSparseCPU.test_select_no_type_promotion_cpu_int32 │ ├── TestSparseCPU.test_select_no_type_promotion_cpu_int64 │ ├── TestSparseCPU.test_select_no_type_promotion_cpu_int8 │ ├── TestSparseCPU.test_select_no_type_promotion_cpu_uint8 │ ├── TestSparseCPU.test_shared_cpu_complex128 │ ├── TestSparseCPU.test_shared_cpu_float64 │ ├── TestSparseCPU.test_small_nnz_coalesced_cpu │ ├── TestSparseCPU.test_spadd_cpu_float64 │ ├── TestSparseCPU.test_sparse_add_coalesce_cpu_complex128 │ ├── TestSparseCPU.test_sparse_add_coalesce_cpu_complex64 │ ├── TestSparseCPU.test_sparse_add_coalesce_cpu_float32 │ ├── TestSparseCPU.test_sparse_add_coalesce_cpu_float64 │ ├── TestSparseCPU.test_sparse_add_out_bfloat16_cpu_float32 │ ├── TestSparseCPU.test_sparse_addmm_cpu_bfloat16 │ ├── TestSparseCPU.test_sparse_addmm_cpu_complex128 │ ├── TestSparseCPU.test_sparse_addmm_cpu_float64 │ ├── TestSparseCPU.test_sparse_broadcast_to_cpu_complex128 │ ├── TestSparseCPU.test_sparse_broadcast_to_cpu_float64 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_bool │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_complex128 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_complex64 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_float32 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_float64 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_int16 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_int32 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_int64 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_int8 │ ├── TestSparseCPU.test_sparse_dense_mul_cpu_uint8 │ ├── TestSparseCPU.test_sparse_mask_cpu_complex128 │ ├── TestSparseCPU.test_sparse_mask_cpu_float64 │ ├── TestSparseCPU.test_sparse_mask_hybrid_cpu_complex128 │ ├── TestSparseCPU.test_sparse_mask_hybrid_cpu_float64 │ ├── TestSparseCPU.test_sparse_mm_cpu_float64 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_complex128 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_complex64 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_float32 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_float64 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_int16 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_int32 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_int64 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_int8 │ ├── TestSparseCPU.test_sparse_sparse_mul_cpu_uint8 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_bool │ ├── TestSparseCPU.test_sparse_spdiags_cpu_complex128 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_complex64 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_float32 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_float64 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_int16 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_int32 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_int64 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_int8 │ ├── TestSparseCPU.test_sparse_spdiags_cpu_uint8 │ ├── TestSparseCPU.test_sparse_sum_cpu_float64 │ ├── TestSparseCPU.test_sparse_to_numpy_cpu │ ├── TestSparseCPU.test_sum_cpu_bool │ ├── TestSparseCPU.test_sum_cpu_complex128 │ ├── TestSparseCPU.test_sum_cpu_complex64 │ ├── TestSparseCPU.test_sum_cpu_float32 │ ├── TestSparseCPU.test_sum_cpu_float64 │ ├── TestSparseCPU.test_sum_cpu_int16 │ ├── TestSparseCPU.test_sum_cpu_int32 │ ├── TestSparseCPU.test_sum_cpu_int64 │ ├── TestSparseCPU.test_sum_cpu_int8 │ ├── TestSparseCPU.test_sum_cpu_uint8 │ ├── TestSparseCPU.test_to_dense_hybrid_masked_cpu_complex128 │ ├── TestSparseCPU.test_to_dense_hybrid_masked_cpu_float64 │ ├── TestSparseCPU.test_to_dense_hybrid_sparse_cpu_complex128 │ ├── TestSparseCPU.test_to_dense_hybrid_sparse_cpu_float64 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_masked_cpu_bfloat16 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_masked_cpu_complex128 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_masked_cpu_complex64 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_masked_cpu_float16 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_masked_cpu_float32 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_masked_cpu_float64 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_sparse_cpu_bfloat16 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_sparse_cpu_complex128 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_sparse_cpu_complex64 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_sparse_cpu_float16 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_sparse_cpu_float32 │ ├── TestSparseCPU.test_to_dense_with_gradcheck_sparse_cpu_float64 │ ├── TestSparseCPU.test_transpose_cpu_complex128 │ ├── TestSparseCPU.test_transpose_cpu_float64 │ ├── TestSparseCPU.test_unsqueeze_cpu_complex128 │ ├── TestSparseCPU.test_unsqueeze_cpu_float64 │ ├── TestSparseCPU.test_zeros_cpu_complex128 │ ├── TestSparseCPU.test_zeros_cpu_float64 │ ├── TestSparseCPU.test_zeros_like_cpu_complex128 │ ├── TestSparseCPU.test_zeros_like_cpu_float64 │ ├── TestSparseCSRCPU.test_addmm_errors_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_False_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_False_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_False_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_False_cpu_float64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_True_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_True_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_True_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int32_noncontiguous_True_cpu_float64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_False_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_False_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_False_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_False_cpu_float64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_True_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_True_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_True_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_2_int64_noncontiguous_True_cpu_float64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_False_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_False_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_False_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_False_cpu_float64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_True_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_True_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_True_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int32_noncontiguous_True_cpu_float64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_False_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_False_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_False_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_False_cpu_float64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_True_cpu_complex128 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_True_cpu_complex64 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_True_cpu_float32 │ ├── TestSparseCSRCPU.test_block_triangular_solve_block_size_3_int64_noncontiguous_True_cpu_float64 │ ├── TestSparseCSRCPU.test_csr_to_block_csr_blocksize_2_cpu_float64_int32 │ ├── TestSparseCSRCPU.test_csr_to_block_csr_blocksize_2_cpu_float64_int64 │ ├── TestSparseCSRCPU.test_csr_to_block_csr_blocksize_4_cpu_float64_int32 │ ├── TestSparseCSRCPU.test_csr_to_block_csr_blocksize_4_cpu_float64_int64 │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseBSC_Batched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseBSC_NonBatched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseBSR_Batched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseBSR_NonBatched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseCSC_Batched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseCSC_NonBatched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseCSR_Batched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_dense_to_from_sparse_compressed_SparseCSR_NonBatched_NonHybrid_cpu │ ├── TestSparseCSRCPU.test_mm_errors_cpu_float32 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_bfloat16 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_bool │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_complex128 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_complex64 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_float16 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_float32 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_float64 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_int16 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_int32 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_int64 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_int8 │ ├── TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_uint8 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_bfloat16 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_bool │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_complex128 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_complex64 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_float16 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_float32 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_float64 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_int16 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_int32 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_int64 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_int8 │ ├── TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int32_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSC_int64_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int32_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseBSR_int64_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int32_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSC_int64_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int32_cpu_uint8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_bfloat16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_bool │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_complex128 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_complex64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_float16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_float32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_float64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_int16 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_int32 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_int64 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_int8 │ ├── TestSparseCompressedCPU.test_select_copy_SparseCSR_int64_cpu_uint8 │ ├── TestSparseMeta.test_basic │ ├── TestSqueeze.test_squeeze_type │ ├── TestSubscripting.test_test_zero_rank │ ├── TestSymbolicShapeAnalysis.test_convolution_backward │ ├── TestSymbolicTracing.test_cpu_scalar_cuda │ ├── TestTEFuserDynamic.test_matmul │ ├── TestTEFuserDynamic.test_unary_ops │ ├── TestTEFuserStatic.test_skip_grad_in_check │ ├── TestTEFuserStatic.test_unary_ops │ ├── TestTensorBuiltins.test_scalar_to_num_conversions │ ├── TestTensorExprPyBind.test_kernel_with_custom_lowering │ ├── TestTorchTidyProfiler.test_optimizer │ ├── TestTorchTidyProfiler.test_optimizer_parameters_adam │ ├── TestTorchTidyProfiler.test_optimizer_parameters_sgd │ ├── TestTorchbind.test_default_args │ ├── TestTorchbind.test_lambda_as_constructor │ ├── TestTorchbind.test_profiler_custom_op │ ├── TestTorchbind.test_staticmethod │ ├── TestTorchbind.test_torchbind │ ├── TestTorchbind.test_torchbind_attr_exception │ ├── TestTorchbind.test_torchbind_class_attr_recursive │ ├── TestTorchbind.test_torchbind_class_attribute │ ├── TestTorchbind.test_torchbind_deepcopy │ ├── TestTorchbind.test_torchbind_def_property_getter_setter │ ├── TestTorchbind.test_torchbind_def_property_just_getter │ ├── TestTorchbind.test_torchbind_def_property_readwrite │ ├── TestTorchbind.test_torchbind_getattr │ ├── TestTorchbind.test_torchbind_getstate │ ├── TestTorchbind.test_torchbind_instantiate_missing_class │ ├── TestTorchbind.test_torchbind_lambda_method │ ├── TestTorchbind.test_torchbind_no_init │ ├── TestTorchbind.test_torchbind_optional_explicit_attr │ ├── TestTorchbind.test_torchbind_pass_wrong_type │ ├── TestTorchbind.test_torchbind_pickle_serialization │ ├── TestTorchbind.test_torchbind_python_deepcopy │ ├── TestTorchbind.test_torchbind_return_instance │ ├── TestTorchbind.test_torchbind_return_instance_from_method │ ├── TestTorchbind.test_torchbind_return_tuple │ ├── TestTorchbind.test_torchbind_save_load │ ├── TestTorchbind.test_torchbind_take_as_arg │ ├── TestTorchbind.test_torchbind_take_instance_as_method_arg │ ├── TestTorchbind.test_torchbind_tracing │ ├── TestTorchbind.test_torchbind_tracing_nested │ ├── TestTransformersCPU.test_decoder_padding_and_src_mask_bool_cpu │ ├── TestTypesAndAnnotation.test_pep585_type │ ├── TestTyping.test_optional_conversion │ ├── TestUnaryUfuncsCPU.test_sinc_cpu_float64 │ ├── TestUnaryUfuncsCPU.test_special_i0_i1_vs_scipy_cpu_bfloat16 │ ├── TestUnaryUfuncsCPU.test_special_i0_i1_vs_scipy_cpu_float32 │ ├── TestUnaryUfuncsCPU.test_special_i0_i1_vs_scipy_cpu_float64 │ ├── TestVerifyCorrectness.test_example_inputs │ ├── TestVmapOperators.test_conv2d │ ├── TestVstack.test_generator │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_cat_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpyCatCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpyCubeCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpyMulCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpyNMSCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpyNonzeroCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpySortCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpySplitCopyCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpySplitCopyWithIntCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpyTakeCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_custom_NumpyViewCopyCustomOp_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_mul_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_native_batch_norm_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_split_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_split_list_args_cpu_float32 │ ├── TestWrapperSubclassAliasingCPU.test_wrapper_subclass_aliasing_view_cpu_float32 │ └── TraceRuleTests.test_torch_name_rule_map_updated ├── edge │ ├── CMakeLists.txt │ ├── Evalue.h │ ├── custom_ops.cpp │ ├── custom_ops.yaml │ ├── event_tracer.h │ ├── event_tracer_hooks.h │ ├── kernel_runtime_context.h │ ├── operator_registry.cpp │ ├── operator_registry.h │ ├── selected_operators.yaml │ ├── templates │ │ ├── Functions.h │ │ ├── NativeFunctions.h │ │ ├── RegisterCodegenUnboxedKernels.cpp │ │ ├── RegisterDispatchKeyCustomOps.cpp │ │ ├── RegisterKernels.h │ │ └── RegisterSchema.cpp │ ├── test_main.cpp │ └── test_operator_registration.cpp ├── error_messages │ └── storage.py ├── expect │ ├── HasDecompTest.test_aten_core_operators.expect │ ├── HasDecompTest.test_has_decomposition.expect │ ├── TestAutograd.test_function-x_grad_desc.expect │ ├── TestAutograd.test_function-y_grad_desc.expect │ ├── TestFXAPIBackwardCompatibility.test_class_member_back_compat-fx_backcompat_class_members.expect │ ├── TestFXAPIBackwardCompatibility.test_function_back_compat-fx_backcompat_function_signatures.expect │ ├── TestJit.test_cu_escaped_number.expect │ ├── TestJit.test_import_method.expect │ ├── TestJit.test_non_ascii_string.expect │ ├── TestJit.test_pretty_printer-empty_float_list_test.expect │ ├── TestJit.test_pretty_printer-empty_int_list_test.expect │ ├── TestJit.test_pretty_printer-if_one.expect │ ├── TestJit.test_pretty_printer-if_test.expect │ ├── TestJit.test_pretty_printer-loop_use_test.expect │ ├── TestJit.test_pretty_printer-print_weird_test.expect │ ├── TestJit.test_pretty_printer-python_op_name_test.expect │ ├── TestJit.test_pretty_printer-while_if_test.expect │ ├── TestJit.test_pretty_printer-while_test.expect │ ├── TestPytorchExportModes.test_aten_fallback.expect │ ├── TestPytorchExportModes.test_onnx_aten.expect │ ├── TestScript.test_annot_ast_mypy_fn.expect │ ├── TestScript.test_annot_ast_mypy_method.expect │ ├── TestScript.test_annot_ast_py3_fn.expect │ ├── TestScript.test_annot_ast_py3_method.expect │ ├── TestScript.test_annot_string_mypy_fn.expect │ ├── TestScript.test_annot_string_mypy_method.expect │ ├── TestScript.test_annot_string_py3_fn.expect │ ├── TestScript.test_annot_string_py3_method.expect │ ├── TestScript.test_annotated_script_fn.expect │ ├── TestScript.test_annotated_script_method.expect │ ├── TestScript.test_format-stdout.expect │ ├── TestScript.test_listconstruct_erasure.expect │ ├── TestScript.test_parser_type_annotations.expect │ ├── TestScript.test_parser_type_annotations_comment.expect │ ├── TestScript.test_print-stdout.expect │ ├── TestScript.test_python_frontend.expect │ ├── TestScript.test_python_frontend_py2.expect │ ├── TestScript.test_python_frontend_py3.expect │ ├── TestScript.test_string_print-stdout.expect │ ├── TestScript.test_torch_dot_tensor_annotation.expect │ ├── TestSparseCPU.test_print_coalesced_cpu_float64.expect │ ├── TestSparseCPU.test_print_uncoalesced_cpu_float64.expect │ ├── TestSparseCUDA.test_print_coalesced_cuda_float64.expect │ ├── TestSparseCUDA.test_print_uncoalesced_cuda_float64.expect │ ├── TestSparseCompressedCPU.test_print_SparseBSC_cpu.expect │ ├── TestSparseCompressedCPU.test_print_SparseBSR_cpu.expect │ ├── TestSparseCompressedCPU.test_print_SparseCSC_cpu.expect │ ├── TestSparseCompressedCPU.test_print_SparseCSR_cpu.expect │ ├── TestSparseCompressedCUDA.test_print_SparseBSC_cuda.expect │ ├── TestSparseCompressedCUDA.test_print_SparseBSR_cuda.expect │ ├── TestSparseCompressedCUDA.test_print_SparseCSC_cuda.expect │ ├── TestSparseCompressedCUDA.test_print_SparseCSR_cuda.expect │ ├── TestSparseMeta.test_print_meta_SparseBSC_float64.expect │ ├── TestSparseMeta.test_print_meta_SparseBSR_float64.expect │ ├── TestSparseMeta.test_print_meta_SparseCOO_float64.expect │ ├── TestSparseMeta.test_print_meta_SparseCSC_float64.expect │ ├── TestSparseMeta.test_print_meta_SparseCSR_float64.expect │ ├── TestTensorBoard.test_audio.expect │ ├── TestTensorBoard.test_caffe2_simple_cnnmodel.expect │ ├── TestTensorBoard.test_caffe2_simple_model.expect │ ├── TestTensorBoard.test_histogram_auto.expect │ ├── TestTensorBoard.test_histogram_doane.expect │ ├── TestTensorBoard.test_histogram_fd.expect │ ├── TestTensorBoard.test_hparams_bool.expect │ ├── TestTensorBoard.test_hparams_number.expect │ ├── TestTensorBoard.test_hparams_string.expect │ ├── TestTensorBoard.test_image_with_3_channel_batched.expect │ ├── TestTensorBoard.test_image_with_boxes.expect │ ├── TestTensorBoard.test_image_with_one_channel.expect │ ├── TestTensorBoard.test_image_with_one_channel_batched.expect │ ├── TestTensorBoard.test_image_without_channel.expect │ ├── TestTensorBoard.test_mesh.expect │ ├── TestTensorBoard.test_nested_nn_squential.expect │ ├── TestTensorBoard.test_pr_curve.expect │ ├── TestTensorBoard.test_pr_curve_raw.expect │ ├── TestTensorBoard.test_pytorch_graph.expect │ ├── TestTensorBoard.test_scalar_new_style.expect │ ├── TestTensorBoard.test_text.expect │ ├── TestTensorBoard.test_video.expect │ ├── TestTorch.test_is_nonzero-empty.expect │ ├── TestTorch.test_is_nonzero-multiple.expect │ ├── TestTorch.test_print-non_contiguous.expect │ └── __init__.py ├── export │ ├── __init__.py │ ├── opinfo_schema.py │ ├── random_dag.py │ ├── test_converter.py │ ├── test_cpp_serdes.py │ ├── test_db.py │ ├── test_draft_export.py │ ├── test_experimental.py │ ├── test_export.py │ ├── test_export_legacy.py │ ├── test_export_nonstrict.py │ ├── test_export_training_ir_to_run_decomp.py │ ├── test_functionalized_assertions.py │ ├── test_hop.py │ ├── test_lift_unlift.py │ ├── test_pass_infra.py │ ├── test_passes.py │ ├── test_retraceability.py │ ├── test_schema.py │ ├── test_serdes.py │ ├── test_serialize.py │ ├── test_sparse.py │ ├── test_swap.py │ ├── test_tools.py │ ├── test_torchbind.py │ ├── test_tree_utils.py │ ├── test_unflatten.py │ ├── test_unflatten_training_ir.py │ ├── test_verifier.py │ └── testing.py ├── forward_backward_compatibility │ ├── check_forward_backward_compatibility.py │ └── dump_all_function_schemas.py ├── functorch │ ├── attn_ft.py │ ├── attn_positional.py │ ├── common_utils.py │ ├── discover_coverage.py │ ├── functorch_additional_op_db.py │ ├── test_ac.py │ ├── test_ac_knapsack.py │ ├── test_ac_logging.py │ ├── test_aotdispatch.py │ ├── test_control_flow.py │ ├── test_dims.py │ ├── test_eager_transforms.py │ ├── test_logging.py │ ├── test_memory_efficient_fusion.py │ ├── test_minifier.py │ ├── test_ops.py │ ├── test_parsing.py │ ├── test_rearrange.py │ ├── test_vmap.py │ ├── test_vmap_registrations.py │ └── xfail_suggester.py ├── fx │ ├── named_tup.py │ ├── quantization.py │ ├── test_common_passes.py │ ├── test_cse_pass.py │ ├── test_dce_pass.py │ ├── test_dynamism.py │ ├── test_future.py │ ├── test_fx_const_fold.py │ ├── test_fx_node_hook.py │ ├── test_fx_param_shape_control_flow.py │ ├── test_fx_split.py │ ├── test_fx_traceback.py │ ├── test_fx_xform_observer.py │ ├── test_gradual_type.py │ ├── test_graph_pickler.py │ ├── test_lazy_graph_module.py │ ├── test_matcher_utils.py │ ├── test_partitioner_order.py │ ├── test_pass_infra.py │ ├── test_shape_inference.py │ ├── test_source_matcher_utils.py │ ├── test_subgraph_rewriter.py │ └── test_z3_gradual_types.py ├── hi.py ├── higher_order_ops │ ├── test_invoke_quant.py │ ├── test_invoke_subgraph.py │ └── test_with_effects.py ├── inductor │ ├── CMakeLists.txt │ ├── __init__.py │ ├── cpp │ │ ├── .gitignore │ │ ├── CMakeLists.txt │ │ ├── test.sh │ │ └── test_cpp_prefix.cpp │ ├── custom_ops.cpp │ ├── extension_backends │ │ ├── cpp │ │ │ ├── extension_codegen_backend.py │ │ │ └── extension_device.cpp │ │ └── triton │ │ │ ├── device_interface.py │ │ │ └── extension_codegen_backend.py │ ├── indirect_assert_helper.py │ ├── minifier_smoke.py │ ├── opinfo_harness.py │ ├── s429861_repro.py │ ├── test_aot_inductor.py │ ├── test_aot_inductor_arrayref.py │ ├── test_aot_inductor_custom_ops.py │ ├── test_aot_inductor_package.py │ ├── test_aot_inductor_utils.py │ ├── test_async_compile.py │ ├── test_auto_functionalize.py │ ├── test_autoheuristic.py │ ├── test_b2b_gemm.py │ ├── test_benchmark_fusion.py │ ├── test_benchmarking.py │ ├── test_binary_folding.py │ ├── test_block_analysis.py │ ├── test_ck_backend.py │ ├── test_codecache.py │ ├── test_codegen_triton.py │ ├── test_combo_kernels.py │ ├── test_compile_worker.py │ ├── test_compiled_autograd.py │ ├── test_compiled_optimizers.py │ ├── test_config.py │ ├── test_control_flow.py │ ├── test_cooperative_reductions.py │ ├── test_coordinate_descent_tuner.py │ ├── test_cpp_wrapper_hipify.py │ ├── test_cpu_cpp_wrapper.py │ ├── test_cpu_repro.py │ ├── test_cpu_select_algorithm.py │ ├── test_cuda_repro.py │ ├── test_cudacodecache.py │ ├── test_cudagraph_trees.py │ ├── test_cudagraph_trees_expandable_segments.py │ ├── test_custom_lowering.py │ ├── test_custom_post_grad_passes.py │ ├── test_cutlass_backend.py │ ├── test_debug_trace.py │ ├── test_decompose_mem_bound_mm.py │ ├── test_dependencies.py │ ├── test_distributed_patterns.py │ ├── test_efficient_conv_bn_eval.py │ ├── test_extension_backend.py │ ├── test_external_callables.py │ ├── test_flex_attention.py │ ├── test_flex_decoding.py │ ├── test_foreach.py │ ├── test_fp8.py │ ├── test_fused_attention.py │ ├── test_fuzzer.py │ ├── test_fx_fusion.py │ ├── test_gpu_cpp_wrapper.py │ ├── test_graph_transform_observer.py │ ├── test_group_batch_fusion.py │ ├── test_halide.py │ ├── test_indexing.py │ ├── test_inductor_annotations.py │ ├── test_inductor_freezing.py │ ├── test_inductor_utils.py │ ├── test_inplace_padding.py │ ├── test_inplacing_pass.py │ ├── test_kernel_benchmark.py │ ├── test_layout_optim.py │ ├── test_loop_ordering.py │ ├── test_max_autotune.py │ ├── test_memory.py │ ├── test_memory_planning.py │ ├── test_metrics.py │ ├── test_minifier.py │ ├── test_minifier_isolate.py │ ├── test_minifier_utils.py │ ├── test_mkldnn_pattern_matcher.py │ ├── test_mmdecomp.py │ ├── test_move_constructors_to_cuda.py │ ├── test_mps_basic.py │ ├── test_multi_kernel.py │ ├── test_op_completeness.py │ ├── test_op_dtype_prop.py │ ├── test_ordered_set.py │ ├── test_pad_mm.py │ ├── test_padding.py │ ├── test_pattern_matcher.py │ ├── test_perf.py │ ├── test_profiler.py │ ├── test_provenance_tracing.py │ ├── test_scatter_optimization.py │ ├── test_select_algorithm.py │ ├── test_smoke.py │ ├── test_snode_runtime.py │ ├── test_split_cat_fx_aten_passes.py │ ├── test_split_cat_fx_passes.py │ ├── test_standalone_compile.py │ ├── test_torchbind.py │ ├── test_torchinductor.py │ ├── test_torchinductor_codegen_config_overrides.py │ ├── test_torchinductor_codegen_dynamic_shapes.py │ ├── test_torchinductor_dynamic_shapes.py │ ├── test_torchinductor_opinfo.py │ ├── test_torchinductor_strided_blocks.py │ ├── test_triton_cpu_backend.py │ ├── test_triton_extension_backend.py │ ├── test_triton_heuristics.py │ ├── test_triton_kernels.py │ ├── test_triton_syntax.py │ ├── test_triton_wrapper.py │ ├── test_unbacked_symints.py │ ├── test_utils.py │ └── test_xpu_basic.py ├── inductor_expected_failures │ ├── TestCommonCPU.test_out_linalg_matrix_rank_cpu_float32 │ ├── TestCommonCPU.test_out_linalg_matrix_rank_hermitian_cpu_float32 │ ├── TestCommonCUDA.test_out_linalg_matrix_rank_cuda_float32 │ ├── TestCommonCUDA.test_out_linalg_matrix_rank_hermitian_cuda_float32 │ ├── TestMathBitsCPU.test_conj_view_add_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_addbmm_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_addcdiv_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_addcmul_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_addmm_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_addmm_decomposed_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_addmv_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_addr_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_baddbmm_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_div_no_rounding_mode_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_index_add_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_index_copy_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_index_fill_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_index_put_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_lerp_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_masked_fill_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_masked_scatter_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_mul_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_pow_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_put_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_scatter_add_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_scatter_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_sub_cpu_complex64 │ ├── TestMathBitsCPU.test_conj_view_true_divide_cpu_complex64 │ ├── TestMathBitsCPU.test_neg_conj_view_add_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_addbmm_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_addcdiv_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_addcmul_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_addmm_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_addmm_decomposed_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_addmv_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_addr_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_baddbmm_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_div_no_rounding_mode_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_float_power_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_index_add_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_index_copy_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_lerp_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_masked_scatter_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_mul_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_pow_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_put_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_scatter_add_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_scatter_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_sub_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_conj_view_true_divide_cpu_complex128 │ ├── TestMathBitsCPU.test_neg_view_abs_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_acos_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_acosh_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_add_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_addbmm_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_addcdiv_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_addcmul_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_addmm_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_addmm_decomposed_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_addmv_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_addr_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_asin_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_asinh_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_atan2_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_atan_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_atanh_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_baddbmm_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_ceil_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_clamp_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_clamp_max_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_clamp_min_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_conj_physical_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_copysign_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_cos_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_cosh_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_cumprod_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_cumsum_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_deg2rad_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_digamma_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_div_floor_rounding_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_div_no_rounding_mode_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_div_trunc_rounding_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_erf_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_erfc_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_erfinv_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_exp2_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_exp_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_expm1_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_fill_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_float_power_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_floor_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_floor_divide_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_fmod_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_frac_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_heaviside_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_hypot_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_i0_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_igamma_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_igammac_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_index_add_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_index_copy_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_index_fill_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_index_reduce_amax_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_index_reduce_amin_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_index_reduce_mean_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_index_reduce_prod_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_lerp_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_lgamma_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_log10_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_log1p_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_log2_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_log_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_logit_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_masked_fill_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_masked_scatter_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_mul_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_mvlgamma_mvlgamma_p_1_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_mvlgamma_mvlgamma_p_3_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_mvlgamma_mvlgamma_p_5_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nan_to_num_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_neg_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nextafter_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_alpha_dropout_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_celu_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_dropout2d_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_dropout3d_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_dropout_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_elu_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_feature_alpha_dropout_with_train_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_feature_alpha_dropout_without_train_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_hardsigmoid_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_leaky_relu_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_mish_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_rrelu_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_selu_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_silu_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_nn_functional_threshold_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_polygamma_polygamma_n_0_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_polygamma_polygamma_n_1_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_polygamma_polygamma_n_2_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_polygamma_polygamma_n_3_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_polygamma_polygamma_n_4_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_pow_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_put_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_rad2deg_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_reciprocal_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_remainder_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_renorm_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_resize_as__cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_round_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_round_decimals_0_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_rsqrt_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_scatter_add_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_scatter_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_scatter_reduce_amax_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_scatter_reduce_amin_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_scatter_reduce_mean_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_scatter_reduce_prod_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_scatter_reduce_sum_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sgn_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sigmoid_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sign_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sin_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sinc_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sinh_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sqrt_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_square_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_squeeze_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_squeeze_multiple_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_sub_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_t_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_tan_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_tanh_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_transpose_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_tril_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_triu_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_true_divide_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_trunc_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_unsqueeze_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_xlogy_cpu_float64 │ ├── TestMathBitsCPU.test_neg_view_zero__cpu_float64 │ ├── TestMathBitsCUDA.test_conj_view_add_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_addbmm_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_addcdiv_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_addcmul_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_addmm_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_addmm_decomposed_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_addmv_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_addr_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_baddbmm_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_div_no_rounding_mode_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_index_add_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_index_copy_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_index_fill_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_index_put_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_lerp_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_masked_fill_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_masked_scatter_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_mul_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_pow_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_put_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_scatter_add_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_scatter_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_sub_cuda_complex64 │ ├── TestMathBitsCUDA.test_conj_view_true_divide_cuda_complex64 │ ├── TestMathBitsCUDA.test_neg_conj_view_add_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_addbmm_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_addcdiv_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_addcmul_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_addmm_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_addmm_decomposed_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_addmv_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_addr_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_baddbmm_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_div_no_rounding_mode_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_float_power_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_index_add_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_index_copy_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_lerp_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_masked_scatter_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_mul_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_pow_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_put_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_scatter_add_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_scatter_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_sub_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_conj_view_true_divide_cuda_complex128 │ ├── TestMathBitsCUDA.test_neg_view_abs_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_acos_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_acosh_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_add_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_addbmm_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_addcdiv_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_addcmul_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_addmm_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_addmm_decomposed_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_addmv_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_addr_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_asin_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_asinh_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_atan2_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_atan_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_atanh_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_baddbmm_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_ceil_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_clamp_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_clamp_max_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_clamp_min_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_conj_physical_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_copysign_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_cos_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_cosh_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_cumprod_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_cumsum_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_deg2rad_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_digamma_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_div_floor_rounding_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_div_no_rounding_mode_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_div_trunc_rounding_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_erf_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_erfc_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_erfinv_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_exp2_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_exp_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_expm1_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_fill_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_float_power_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_floor_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_floor_divide_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_fmod_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_frac_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_heaviside_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_hypot_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_i0_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_igamma_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_igammac_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_index_add_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_index_copy_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_index_fill_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_index_reduce_amax_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_index_reduce_amin_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_index_reduce_mean_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_index_reduce_prod_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_lerp_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_lgamma_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_log10_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_log1p_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_log2_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_log_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_logit_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_masked_fill_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_masked_scatter_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_mul_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_mvlgamma_mvlgamma_p_1_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_mvlgamma_mvlgamma_p_3_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_mvlgamma_mvlgamma_p_5_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nan_to_num_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_neg_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nextafter_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_alpha_dropout_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_celu_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_dropout2d_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_dropout3d_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_elu_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_feature_alpha_dropout_with_train_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_feature_alpha_dropout_without_train_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_hardsigmoid_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_leaky_relu_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_mish_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_rrelu_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_selu_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_silu_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_nn_functional_threshold_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_polygamma_polygamma_n_0_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_polygamma_polygamma_n_1_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_polygamma_polygamma_n_2_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_polygamma_polygamma_n_3_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_polygamma_polygamma_n_4_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_pow_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_put_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_rad2deg_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_reciprocal_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_remainder_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_renorm_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_resize_as__cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_round_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_round_decimals_0_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_rsqrt_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_scatter_add_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_scatter_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_scatter_reduce_amax_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_scatter_reduce_amin_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_scatter_reduce_mean_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_scatter_reduce_prod_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_scatter_reduce_sum_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sgn_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sigmoid_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sign_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sin_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sinc_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sinh_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sqrt_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_square_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_squeeze_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_squeeze_multiple_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_sub_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_t_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_tan_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_tanh_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_transpose_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_tril_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_triu_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_true_divide_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_trunc_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_unsqueeze_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_xlogy_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_zero__cuda_float64 │ ├── TestModuleCPU.test_to_nn_Embedding_swap_True_set_grad_True_cpu_float32 │ ├── TestModuleCPU.test_to_nn_PReLU_swap_True_set_grad_True_cpu_float32 │ ├── TestModuleCPU.test_to_nn_RMSNorm_swap_True_set_grad_True_cpu_float32 │ ├── TestModuleCUDA.test_to_nn_Embedding_swap_True_set_grad_True_cuda_float32 │ ├── TestModuleCUDA.test_to_nn_PReLU_swap_True_set_grad_True_cuda_float32 │ └── TestModuleCUDA.test_to_nn_RMSNorm_swap_True_set_grad_True_cuda_float32 ├── inductor_skips │ ├── .gitkeep │ ├── TestMathBitsCUDA.test_neg_view_jiterator_2inputs_2outputs_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_jiterator_4inputs_with_extra_args_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_jiterator_binary_cuda_float64 │ ├── TestMathBitsCUDA.test_neg_view_jiterator_binary_return_by_ref_cuda_float64 │ └── TestMathBitsCUDA.test_neg_view_jiterator_unary_cuda_float64 ├── jit │ ├── __init__.py │ ├── _imported_class_test │ │ ├── __init__.py │ │ ├── bar.py │ │ ├── foo.py │ │ └── very │ │ │ ├── __init__.py │ │ │ └── very │ │ │ ├── __init__.py │ │ │ └── nested.py │ ├── fixtures │ │ ├── _test_serialization_subcmul_v2.pt │ │ ├── test_versioned_div_scalar_float_v3.pt │ │ ├── test_versioned_div_scalar_inplace_float_v3.pt │ │ ├── test_versioned_div_scalar_inplace_int_v3.pt │ │ ├── test_versioned_div_scalar_int_v3.pt │ │ ├── test_versioned_div_scalar_reciprocal_float_v3.pt │ │ ├── test_versioned_div_scalar_reciprocal_int_v3.pt │ │ ├── test_versioned_div_scalar_scalar_v3.pt │ │ ├── test_versioned_div_tensor_example_v7.ptl │ │ ├── test_versioned_div_tensor_inplace_v3.pt │ │ ├── test_versioned_div_tensor_out_v3.pt │ │ ├── test_versioned_div_tensor_v3.pt │ │ ├── test_versioned_full_integer_value_v4.pt │ │ ├── test_versioned_full_preserved_v4.pt │ │ ├── test_versioned_gelu_out_v9.ptl │ │ ├── test_versioned_gelu_v9.ptl │ │ ├── test_versioned_linspace_out_v7.ptl │ │ ├── test_versioned_linspace_v7.ptl │ │ ├── test_versioned_logspace_out_v8.ptl │ │ ├── test_versioned_logspace_v8.ptl │ │ ├── test_versioned_random_func_v10.ptl │ │ ├── test_versioned_random_out_v10.ptl │ │ └── test_versioned_random_v10.ptl │ ├── fixtures_srcs │ │ ├── __init__.py │ │ ├── fixtures_src.py │ │ ├── generate_models.py │ │ └── test_upgrader_models_generation.py │ ├── mydecorator.py │ ├── myexception.py │ ├── myfunction_a.py │ ├── myfunction_b.py │ ├── test_alias_analysis.py │ ├── test_async.py │ ├── test_aten_pow.py │ ├── test_attr.py │ ├── test_autodiff.py │ ├── test_autodiff_subgraph_slicing.py │ ├── test_await.py │ ├── test_backend_nnapi.py │ ├── test_backends.py │ ├── test_batch_mm.py │ ├── test_builtins.py │ ├── test_class_type.py │ ├── test_complex.py │ ├── test_complexity.py │ ├── test_convert_activation.py │ ├── test_cuda.py │ ├── test_custom_operators.py │ ├── test_data_parallel.py │ ├── test_dataclasses.py │ ├── test_dce.py │ ├── test_decorator.py │ ├── test_device_analysis.py │ ├── test_dtype_analysis.py │ ├── test_enum.py │ ├── test_exception.py │ ├── test_freezing.py │ ├── test_functional_blocks.py │ ├── test_fuser_common.py │ ├── test_generator.py │ ├── test_graph_rewrite_passes.py │ ├── test_hash.py │ ├── test_hooks.py │ ├── test_hooks_modules.py │ ├── test_ignorable_args.py │ ├── test_ignore_context_manager.py │ ├── test_isinstance.py │ ├── test_jit_utils.py │ ├── test_list_dict.py │ ├── test_logging.py │ ├── test_misc.py │ ├── test_models.py │ ├── test_module_apis.py │ ├── test_module_containers.py │ ├── test_module_interface.py │ ├── test_modules.py │ ├── test_op_decompositions.py │ ├── test_optimize_for_mobile_preserve_debug_info.py │ ├── test_parametrization.py │ ├── test_pdt.py │ ├── test_peephole.py │ ├── test_profiler.py │ ├── test_python_bindings.py │ ├── test_python_builtins.py │ ├── test_python_ir.py │ ├── test_recursive_script.py │ ├── test_remove_mutation.py │ ├── test_save_load.py │ ├── test_save_load_for_op_version.py │ ├── test_script_profile.py │ ├── test_scriptmod_ann.py │ ├── test_slice.py │ ├── test_sparse.py │ ├── test_string_formatting.py │ ├── test_symbolic_shape_analysis.py │ ├── test_tensor_creation_ops.py │ ├── test_tensor_methods.py │ ├── test_torchbind.py │ ├── test_tracer.py │ ├── test_type_sharing.py │ ├── test_types.py │ ├── test_typing.py │ ├── test_union.py │ ├── test_union_pep604.py │ ├── test_unsupported_ops.py │ ├── test_upgraders.py │ ├── test_warn.py │ ├── test_with.py │ └── xnnpack │ │ └── test_xnnpack_delegate.py ├── jit_hooks │ ├── CMakeLists.txt │ ├── model.py │ └── test_jit_hooks.cpp ├── lazy │ ├── __init__.py │ ├── test_bindings.py │ ├── test_debug_util.py │ ├── test_extract_compiled_graph.py │ ├── test_functionalization.py │ ├── test_generator.py │ ├── test_meta_kernel.py │ ├── test_reuse_ir.py │ ├── test_step_closures.py │ └── test_ts_opinfo.py ├── linear.py ├── load_torchscript_model.py ├── minioptest_failures_dict.json ├── mkl_verbose.py ├── mkldnn_verbose.py ├── mobile │ ├── custom_build │ │ ├── CMakeLists.txt │ │ ├── build.sh │ │ ├── expected_output.txt │ │ ├── predictor.cpp │ │ └── prepare_model.py │ ├── lightweight_dispatch │ │ ├── CMakeLists.txt │ │ ├── build.sh │ │ ├── lightweight_dispatch_ops.yaml │ │ ├── test_codegen_unboxing.cpp │ │ ├── test_lightweight_dispatch.cpp │ │ └── tests_setup.py │ ├── model_test │ │ ├── README.md │ │ ├── android_api_module.py │ │ ├── builtin_ops.py │ │ ├── coverage.yaml │ │ ├── gen_test_model.py │ │ ├── math_ops.py │ │ ├── model_ops.yaml │ │ ├── nn_ops.py │ │ ├── quantization_ops.py │ │ ├── sampling_ops.py │ │ ├── tensor_ops.py │ │ ├── torchvision_models.py │ │ └── update_production_ops.py │ ├── nnc │ │ ├── CMakeLists.txt │ │ ├── aot_test_model.py │ │ ├── test_aot_compile.sh │ │ ├── test_context.cpp │ │ ├── test_nnc_backend.cpp │ │ └── test_registry.cpp │ ├── test_bytecode.py │ ├── test_lite_script_module.py │ ├── test_lite_script_type.py │ ├── test_quantize_fx_lite_script_module.py │ ├── test_upgrader_bytecode_table_example.cpp │ ├── test_upgrader_codegen.py │ └── test_upgraders.py ├── nn │ ├── test_convolution.py │ ├── test_dropout.py │ ├── test_embedding.py │ ├── test_init.py │ ├── test_lazy_modules.py │ ├── test_load_state_dict.py │ ├── test_module_hooks.py │ ├── test_multihead_attention.py │ ├── test_packed_sequence.py │ ├── test_parametrization.py │ ├── test_pooling.py │ └── test_pruning.py ├── onnx │ ├── assets │ │ ├── grace_hopper_517x606.jpg │ │ └── rgb_pytorch.png │ ├── autograd_helper.py │ ├── dynamo │ │ └── test_dynamo_with_onnxruntime_backend.py │ ├── expect │ │ ├── TestFindMismatch_ONNX.test_pretty_print_tree_visualizes_mismatch.expect │ │ ├── TestFindMismatch_ONNX_RUNTIME_CPU.test_find_mismatch_prints_correct_info_when_no_mismatch.expect │ │ └── TestFindMismatch_ONNX_RUNTIME_CPU.test_pretty_print_tree_visualizes_mismatch.expect │ ├── exporter │ │ ├── README.md │ │ ├── test_api.py │ │ ├── test_building.py │ │ ├── test_capture_strategies.py │ │ ├── test_core.py │ │ ├── test_dynamic_shapes.py │ │ ├── test_hf_models_e2e.py │ │ ├── test_ir_passes.py │ │ ├── test_small_models_e2e.py │ │ └── test_tensors.py │ ├── internal │ │ ├── test_diagnostics.py │ │ └── test_registraion.py │ ├── model_defs │ │ ├── __init__.py │ │ ├── dcgan.py │ │ ├── emb_seq.py │ │ ├── lstm_flattening_result.py │ │ ├── mnist.py │ │ ├── op_test.py │ │ ├── rnn_model_with_packed_sequence.py │ │ ├── squeezenet.py │ │ ├── srresnet.py │ │ ├── super_resolution.py │ │ └── word_language_model.py │ ├── onnx_test_common.py │ ├── pytorch_test_common.py │ ├── test_autograd_funs.py │ ├── test_custom_ops.py │ ├── test_fx_passes.py │ ├── test_fx_type_promotion.py │ ├── test_lazy_import.py │ ├── test_models.py │ ├── test_models_onnxruntime.py │ ├── test_models_quantized_onnxruntime.py │ ├── test_onnx_opset.py │ ├── test_onnxscript_no_runtime.py │ ├── test_onnxscript_runtime.py │ ├── test_op_consistency.py │ ├── test_pytorch_jit_onnx.py │ ├── test_pytorch_onnx_no_runtime.py │ ├── test_pytorch_onnx_onnxruntime.py │ ├── test_pytorch_onnx_onnxruntime_cuda.py │ ├── test_pytorch_onnx_shape_inference.py │ ├── test_symbolic_helper.py │ ├── test_utility_funs.py │ ├── test_verification.py │ ├── torchlib │ │ ├── README.md │ │ ├── error_reproduction.py │ │ ├── ops_test_common.py │ │ ├── ops_test_data.py │ │ └── test_ops.py │ └── verify.py ├── optim │ ├── test_lrscheduler.py │ ├── test_optim.py │ └── test_swa_utils.py ├── package │ ├── __init__.py │ ├── common.py │ ├── generate_bc_packages.py │ ├── module_a.py │ ├── module_a_remapped_path.py │ ├── package_a │ │ ├── __init__.py │ │ ├── fake_interface.py │ │ ├── fake_script_class.py │ │ ├── long_name.py │ │ ├── std_sys_module_hacks.py │ │ ├── std_sys_module_hacks_3_13.py │ │ ├── subpackage.py │ │ ├── test_all_leaf_modules_tracer.py │ │ ├── test_module.py │ │ ├── test_nn_module.py │ │ ├── use_dunder_package.py │ │ └── use_torch_package_importer.py │ ├── package_b │ │ ├── __init__.py │ │ ├── subpackage_0 │ │ │ ├── __init__.py │ │ │ └── subsubpackage_0 │ │ │ │ └── __init__.py │ │ ├── subpackage_1.py │ │ └── subpackage_2.py │ ├── package_bc │ │ ├── test_fx_module.pt │ │ ├── test_nn_module.pt │ │ └── test_torchscript_module.pt │ ├── package_c │ │ ├── __init__.py │ │ └── test_module.py │ ├── package_d │ │ ├── __init__.py │ │ ├── imports_directly.py │ │ ├── imports_indirectly.py │ │ └── subpackage_0 │ │ │ ├── __init__.py │ │ │ └── subsubpackage_0 │ │ │ └── __init__.py │ ├── package_e │ │ └── test_nn_module.pt │ ├── test_analyze.py │ ├── test_dependency_api.py │ ├── test_dependency_hooks.py │ ├── test_digraph.py │ ├── test_directory_reader.py │ ├── test_glob_group.py │ ├── test_importer.py │ ├── test_load_bc_packages.py │ ├── test_mangling.py │ ├── test_misc.py │ ├── test_model.py │ ├── test_package_fx.py │ ├── test_package_script.py │ ├── test_repackage.py │ ├── test_resources.py │ ├── test_save_load.py │ └── test_trace_dep │ │ └── __init__.py ├── profiler │ ├── profiler_utils_mock_events.json │ ├── test_cpp_thread.cpp │ ├── test_cpp_thread.py │ ├── test_cpp_thread_lib.pyi │ ├── test_execution_trace.py │ ├── test_kineto.py │ ├── test_memory_profiler.py │ ├── test_profiler.py │ ├── test_profiler_tree.py │ ├── test_record_function.py │ └── test_torch_tidy.py ├── pytest_shard_custom.py ├── quantization │ ├── __init__.py │ ├── ao_migration │ │ ├── __init__.py │ │ ├── common.py │ │ ├── test_ao_migration.py │ │ ├── test_quantization.py │ │ └── test_quantization_fx.py │ ├── bc │ │ ├── __init__.py │ │ └── test_backward_compatibility.py │ ├── core │ │ ├── __init__.py │ │ ├── experimental │ │ │ ├── apot_fx_graph_mode_ptq.py │ │ │ ├── apot_fx_graph_mode_qat.py │ │ │ ├── quantization_util.py │ │ │ ├── test_adaround_eager.py │ │ │ ├── test_bits.py │ │ │ ├── test_fake_quantize.py │ │ │ ├── test_float8.py │ │ │ ├── test_linear.py │ │ │ ├── test_nonuniform_observer.py │ │ │ ├── test_quantized_tensor.py │ │ │ └── test_quantizer.py │ │ ├── test_backend_config.py │ │ ├── test_docs.py │ │ ├── test_quantized_functional.py │ │ ├── test_quantized_module.py │ │ ├── test_quantized_op.py │ │ ├── test_quantized_tensor.py │ │ ├── test_top_level_apis.py │ │ ├── test_utils.py │ │ ├── test_workflow_module.py │ │ └── test_workflow_ops.py │ ├── eager │ │ ├── __init__.py │ │ ├── test_bias_correction_eager.py │ │ ├── test_equalize_eager.py │ │ ├── test_fuse_eager.py │ │ ├── test_model_numerics.py │ │ ├── test_numeric_suite_eager.py │ │ ├── test_quantize_eager_ptq.py │ │ └── test_quantize_eager_qat.py │ ├── fx │ │ ├── __init__.py │ │ ├── test_equalize_fx.py │ │ ├── test_model_report_fx.py │ │ ├── test_numeric_suite_fx.py │ │ ├── test_quantize_fx.py │ │ └── test_subgraph_rewriter.py │ ├── jit │ │ ├── __init__.py │ │ ├── test_deprecated_jit_quant.py │ │ ├── test_fusion_passes.py │ │ ├── test_ondevice_quantization.py │ │ └── test_quantize_jit.py │ ├── pt2e │ │ ├── test_duplicate_dq.py │ │ ├── test_graph_utils.py │ │ ├── test_metadata_porting.py │ │ ├── test_numeric_debugger.py │ │ ├── test_quantize_pt2e.py │ │ ├── test_quantize_pt2e_qat.py │ │ ├── test_representation.py │ │ ├── test_x86inductor_quantizer.py │ │ └── test_xnnpack_quantizer.py │ └── serialized │ │ ├── TestSerialization.test_conv2d.expected.pt │ │ ├── TestSerialization.test_conv2d.input.pt │ │ ├── TestSerialization.test_conv2d.scripted.pt │ │ ├── TestSerialization.test_conv2d.state_dict.pt │ │ ├── TestSerialization.test_conv2d.traced.pt │ │ ├── TestSerialization.test_conv2d_graph.expected.pt │ │ ├── TestSerialization.test_conv2d_graph.input.pt │ │ ├── TestSerialization.test_conv2d_graph.scripted.pt │ │ ├── TestSerialization.test_conv2d_graph.traced.pt │ │ ├── TestSerialization.test_conv2d_graph_v2.expected.pt │ │ ├── TestSerialization.test_conv2d_graph_v2.input.pt │ │ ├── TestSerialization.test_conv2d_graph_v2.scripted.pt │ │ ├── TestSerialization.test_conv2d_graph_v2.traced.pt │ │ ├── TestSerialization.test_conv2d_graph_v3.expected.pt │ │ ├── TestSerialization.test_conv2d_graph_v3.input.pt │ │ ├── TestSerialization.test_conv2d_graph_v3.scripted.pt │ │ ├── TestSerialization.test_conv2d_graph_v3.traced.pt │ │ ├── TestSerialization.test_conv2d_nobias.expected.pt │ │ ├── TestSerialization.test_conv2d_nobias.input.pt │ │ ├── TestSerialization.test_conv2d_nobias.scripted.pt │ │ ├── TestSerialization.test_conv2d_nobias.state_dict.pt │ │ ├── TestSerialization.test_conv2d_nobias.traced.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph.expected.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph.input.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph.scripted.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph.traced.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v2.expected.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v2.input.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v2.scripted.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v2.traced.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v3.expected.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v3.input.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v3.scripted.pt │ │ ├── TestSerialization.test_conv2d_nobias_graph_v3.traced.pt │ │ ├── TestSerialization.test_conv2d_relu.expected.pt │ │ ├── TestSerialization.test_conv2d_relu.input.pt │ │ ├── TestSerialization.test_conv2d_relu.scripted.pt │ │ ├── TestSerialization.test_conv2d_relu.state_dict.pt │ │ ├── TestSerialization.test_conv2d_relu.traced.pt │ │ ├── TestSerialization.test_conv3d.expected.pt │ │ ├── TestSerialization.test_conv3d.input.pt │ │ ├── TestSerialization.test_conv3d.scripted.pt │ │ ├── TestSerialization.test_conv3d.state_dict.pt │ │ ├── TestSerialization.test_conv3d.traced.pt │ │ ├── TestSerialization.test_conv3d_relu.expected.pt │ │ ├── TestSerialization.test_conv3d_relu.input.pt │ │ ├── TestSerialization.test_conv3d_relu.scripted.pt │ │ ├── TestSerialization.test_conv3d_relu.state_dict.pt │ │ ├── TestSerialization.test_conv3d_relu.traced.pt │ │ ├── TestSerialization.test_default_qat_qconfig.expected.pt │ │ ├── TestSerialization.test_default_qat_qconfig.input.pt │ │ ├── TestSerialization.test_default_qat_qconfig.state_dict.pt │ │ ├── TestSerialization.test_linear.expected.pt │ │ ├── TestSerialization.test_linear.input.pt │ │ ├── TestSerialization.test_linear.scripted.pt │ │ ├── TestSerialization.test_linear.state_dict.pt │ │ ├── TestSerialization.test_linear.traced.pt │ │ ├── TestSerialization.test_linear_dynamic_float16.expected.pt │ │ ├── TestSerialization.test_linear_dynamic_float16.input.pt │ │ ├── TestSerialization.test_linear_dynamic_float16.scripted.pt │ │ ├── TestSerialization.test_linear_dynamic_float16.state_dict.pt │ │ ├── TestSerialization.test_linear_dynamic_float16.traced.pt │ │ ├── TestSerialization.test_linear_dynamic_qint8.expected.pt │ │ ├── TestSerialization.test_linear_dynamic_qint8.input.pt │ │ ├── TestSerialization.test_linear_dynamic_qint8.scripted.pt │ │ ├── TestSerialization.test_linear_dynamic_qint8.state_dict.pt │ │ ├── TestSerialization.test_linear_dynamic_qint8.traced.pt │ │ ├── TestSerialization.test_linear_relu.expected.pt │ │ ├── TestSerialization.test_linear_relu.input.pt │ │ ├── TestSerialization.test_linear_relu.scripted.pt │ │ ├── TestSerialization.test_linear_relu.state_dict.pt │ │ ├── TestSerialization.test_linear_relu.traced.pt │ │ ├── TestSerialization.test_linear_relu_package_quantization_transforms.expected.pt │ │ ├── TestSerialization.test_linear_relu_package_quantization_transforms.get_attr_targets.pt │ │ ├── TestSerialization.test_linear_relu_package_quantization_transforms.input.pt │ │ ├── TestSerialization.test_linear_relu_package_quantization_transforms.package.pt │ │ ├── TestSerialization.test_lstm.expected.pt │ │ ├── TestSerialization.test_lstm.input.pt │ │ ├── TestSerialization.test_lstm.scripted.pt │ │ ├── TestSerialization.test_lstm.state_dict.pt │ │ ├── TestSerialization.test_lstm.traced.pt │ │ ├── TestSerialization.test_per_channel_observer.expected.pt │ │ ├── TestSerialization.test_per_channel_observer.input.pt │ │ ├── TestSerialization.test_per_channel_observer.state_dict.pt │ │ ├── TestSerialization.test_per_tensor_observer.expected.pt │ │ ├── TestSerialization.test_per_tensor_observer.input.pt │ │ └── TestSerialization.test_per_tensor_observer.state_dict.pt ├── run_doctests.sh ├── run_test.py ├── scripts │ ├── cuda_memcheck_common.py │ └── run_cuda_memcheck.py ├── simulate_nccl_errors.py ├── slow_tests.json ├── strobelight │ └── examples │ │ ├── cli_function_profiler_example.py │ │ └── compile_time_profile_example.py ├── 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_img │ └── p1.jpg ├── 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 ├── torch_np │ ├── __init__.py │ ├── check_tests_conform.py │ ├── conftest.py │ ├── numpy_tests │ │ ├── core │ │ │ ├── test_dlpack.py │ │ │ ├── test_dtype.py │ │ │ ├── test_einsum.py │ │ │ ├── test_getlimits.py │ │ │ ├── test_indexing.py │ │ │ ├── test_multiarray.py │ │ │ ├── test_numeric.py │ │ │ ├── test_numerictypes.py │ │ │ ├── test_scalar_ctors.py │ │ │ ├── test_scalar_methods.py │ │ │ ├── test_scalarinherit.py │ │ │ ├── test_scalarmath.py │ │ │ └── test_shape_base.py │ │ ├── fft │ │ │ ├── test_helper.py │ │ │ └── test_pocketfft.py │ │ ├── lib │ │ │ ├── test_arraypad.py │ │ │ ├── test_arraysetops.py │ │ │ ├── test_function_base.py │ │ │ ├── test_histograms.py │ │ │ ├── test_index_tricks.py │ │ │ ├── test_shape_base_.py │ │ │ ├── test_twodim_base.py │ │ │ └── test_type_check.py │ │ └── linalg │ │ │ └── test_linalg.py │ ├── test_basic.py │ ├── test_binary_ufuncs.py │ ├── test_dtype.py │ ├── test_function_base.py │ ├── test_ndarray_methods.py │ ├── test_nep50_examples.py │ ├── test_random.py │ ├── test_reductions.py │ ├── test_scalars_0D_arrays.py │ ├── test_ufuncs_basic.py │ └── test_unary_ufuncs.py ├── typing │ ├── fail │ │ ├── arithmetic_ops.py │ │ ├── creation_ops.py │ │ ├── disabled_bitwise_ops.py │ │ ├── random.py │ │ └── torch_size.py │ ├── pass │ │ ├── arithmetic_ops.py │ │ ├── creation_ops.py │ │ ├── cuda_steam.py │ │ ├── disabled_jit.py │ │ ├── distributions.py │ │ ├── math_ops.py │ │ └── torch_size.py │ └── reveal │ │ ├── module_list.py │ │ ├── namedtuple.py │ │ ├── opt_size.py │ │ ├── size.py │ │ ├── tensor_constructors.py │ │ ├── tensor_copy.py │ │ ├── tensor_sampling.py │ │ └── torch_optim.py └── xpu │ ├── test_conv.py │ └── test_gemm.py ├── third_party ├── BUCK.oss ├── BUILD ├── LICENSES_BUNDLED.txt ├── METADATA.bzl ├── README.md ├── build_bundled.py ├── cpp-httplib.BUILD ├── cuda.BUILD ├── cudnn.BUILD ├── cudnn_frontend.BUILD ├── cutlass.BUILD ├── eigen.BUILD ├── fmt.BUILD ├── generate-cpuinfo-wrappers.py ├── generate-xnnpack-wrappers.py ├── glog.buck.bzl ├── gloo.BUILD ├── ideep.BUILD ├── kineto.BUILD ├── kineto.buck.bzl ├── miniz-3.0.2 │ ├── BUILD.bazel │ ├── ChangeLog.md │ ├── LICENSE │ ├── examples │ │ ├── example1.c │ │ ├── example2.c │ │ ├── example3.c │ │ ├── example4.c │ │ ├── example5.c │ │ └── example6.c │ ├── miniz.c │ ├── miniz.h │ └── readme.md ├── mkl-dnn.BUILD ├── mkl.BUILD ├── mkl_headers.BUILD ├── nlohmann.BUILD ├── onnx.BUILD ├── opentelemetry-cpp.BUILD ├── sleef.BUILD ├── sleef.bzl ├── substitution.bzl ├── tensorflow_cuda_bazel_build │ └── cuda │ │ └── build_defs.bzl ├── tensorpipe.BUILD ├── valgrind-headers │ ├── README.md │ ├── callgrind.h │ └── valgrind.h ├── xnnpack.buck.bzl ├── xnnpack_buck_shim.bzl ├── xnnpack_src_defs.bzl ├── xnnpack_wrapper_defs.bzl └── xpu.txt ├── tools ├── BUCK.bzl ├── BUCK.oss ├── README.md ├── __init__.py ├── alerts │ ├── __init__.py │ └── create_alerts.py ├── amd_build │ └── build_amd.py ├── autograd │ ├── BUILD.bazel │ ├── README.md │ ├── __init__.py │ ├── build.bzl │ ├── context.py │ ├── deprecated.yaml │ ├── derivatives.yaml │ ├── gen_annotated_fn_args.py │ ├── gen_autograd.py │ ├── gen_autograd_functions.py │ ├── gen_inplace_or_view_type.py │ ├── gen_python_functions.py │ ├── gen_trace_type.py │ ├── gen_variable_factories.py │ ├── gen_variable_type.py │ ├── gen_view_funcs.py │ ├── load_derivatives.py │ └── templates │ │ ├── ADInplaceOrViewType.cpp │ │ ├── Functions.cpp │ │ ├── Functions.h │ │ ├── TraceType.cpp │ │ ├── VariableType.cpp │ │ ├── VariableType.h │ │ ├── ViewFuncs.cpp │ │ ├── ViewFuncs.h │ │ ├── annotated_fn_args.py.in │ │ ├── python_enum_tag.cpp │ │ ├── python_fft_functions.cpp │ │ ├── python_functions.cpp │ │ ├── python_functions.h │ │ ├── python_linalg_functions.cpp │ │ ├── python_nested_functions.cpp │ │ ├── python_nn_functions.cpp │ │ ├── python_return_types.cpp │ │ ├── python_return_types.h │ │ ├── python_sparse_functions.cpp │ │ ├── python_special_functions.cpp │ │ ├── python_torch_functions.cpp │ │ ├── python_variable_methods.cpp │ │ └── variable_factories.h ├── bazel.bzl ├── bazel_tools │ ├── BUILD.bazel │ └── shellwrap.sh ├── build │ └── bazel │ │ ├── requirements.in │ │ └── requirements.txt ├── build_defs │ ├── android │ │ └── build_mode_defs.bzl │ ├── apple │ │ └── build_mode_defs.bzl │ ├── buck_helpers.bzl │ ├── default_platform_defs.bzl │ ├── expect.bzl │ ├── fb_native_wrapper.bzl │ ├── fb_xplat_cxx_library.bzl │ ├── fb_xplat_cxx_test.bzl │ ├── fb_xplat_genrule.bzl │ ├── fbsource_utils.bzl │ ├── glob_defs.bzl │ ├── platform_defs.bzl │ ├── select.bzl │ ├── type_defs.bzl │ └── windows │ │ └── windows_flag_map.bzl ├── build_libtorch.py ├── build_pytorch_libs.py ├── build_with_debinfo.py ├── code_analyzer │ ├── gen_op_registration_allowlist.py │ ├── gen_operators_yaml.py │ └── gen_oplist.py ├── code_coverage │ ├── README.md │ ├── oss_coverage.py │ └── package │ │ ├── __init__.py │ │ ├── oss │ │ ├── __init__.py │ │ ├── cov_json.py │ │ ├── init.py │ │ ├── run.py │ │ └── utils.py │ │ ├── tool │ │ ├── __init__.py │ │ ├── clang_coverage.py │ │ ├── gcc_coverage.py │ │ ├── parser │ │ │ ├── __init__.py │ │ │ ├── coverage_record.py │ │ │ ├── gcov_coverage_parser.py │ │ │ ├── llvm_coverage_parser.py │ │ │ └── llvm_coverage_segment.py │ │ ├── print_report.py │ │ ├── summarize_jsons.py │ │ └── utils.py │ │ └── util │ │ ├── __init__.py │ │ ├── setting.py │ │ ├── utils.py │ │ └── utils_init.py ├── config │ ├── BUILD │ └── defs.bzl ├── coverage_plugins_package │ ├── README.md │ ├── pyproject.toml │ ├── setup.py │ └── src │ │ └── coverage_plugins │ │ ├── __init__.py │ │ └── jit_plugin.py ├── download_mnist.py ├── dynamo │ └── verify_dynamo.py ├── extract_scripts.py ├── flight_recorder │ ├── components │ │ ├── builder.py │ │ ├── config_manager.py │ │ ├── fr_logger.py │ │ ├── loader.py │ │ ├── types.py │ │ └── utils.py │ └── fr_trace.py ├── gdb │ └── pytorch-gdb.py ├── gen_flatbuffers.sh ├── gen_vulkan_spv.py ├── generate_torch_version.py ├── generated_dirs.txt ├── git_add_generated_dirs.sh ├── git_reset_generated_dirs.sh ├── github │ ├── __init__.py │ └── github_utils.py ├── iwyu │ ├── all.imp │ ├── aten.imp │ ├── c10.imp │ ├── fixup.py │ ├── gtest.imp │ ├── run.sh │ ├── system.imp │ └── torch.imp ├── jit │ ├── __init__.py │ ├── gen_unboxing.py │ ├── templates │ │ ├── aten_schema_declarations.cpp │ │ └── external_functions_codegen_template.cpp │ └── test │ │ ├── __init__.py │ │ └── test_gen_unboxing.py ├── linter │ ├── __init__.py │ ├── adapters │ │ ├── README.md │ │ ├── _linter.py │ │ ├── actionlint_linter.py │ │ ├── bazel_linter.py │ │ ├── black_linter.py │ │ ├── clangformat_linter.py │ │ ├── clangtidy_linter.py │ │ ├── cmake_linter.py │ │ ├── docstring_linter.py │ │ ├── exec_linter.py │ │ ├── flake8_linter.py │ │ ├── gha_linter.py │ │ ├── grep_linter.py │ │ ├── import_linter.py │ │ ├── lintrunner_version_linter.py │ │ ├── mypy_linter.py │ │ ├── nativefunctions_linter.py │ │ ├── newlines_linter.py │ │ ├── no_merge_conflict_csv_linter.py │ │ ├── no_workflows_on_fork.py │ │ ├── pip_init.py │ │ ├── pyfmt_linter.py │ │ ├── ruff_linter.py │ │ ├── s3_init.py │ │ ├── s3_init_config.json │ │ ├── set_linter.py │ │ ├── shellcheck_linter.py │ │ ├── test_has_main_linter.py │ │ ├── testowners_linter.py │ │ ├── update_s3.py │ │ └── workflow_consistency_linter.py │ └── clang_tidy │ │ ├── __init__.py │ │ └── generate_build_files.py ├── lite_interpreter │ ├── __init__.py │ └── gen_selected_mobile_ops_header.py ├── lldb │ ├── deploy_debugger.py │ └── pytorch_lldb.py ├── nightly.py ├── nightly_hotpatch.py ├── nvcc_fix_deps.py ├── onnx │ ├── gen_diagnostics.py │ ├── gen_diagnostics.sh │ ├── sarif │ │ ├── code-gen-hints.json │ │ └── gen_sarif.sh │ ├── templates │ │ ├── rules.h.in │ │ └── rules.py.in │ └── update_default_opset_version.py ├── packaging │ ├── build_wheel.py │ └── split_wheel.py ├── pyi │ ├── __init__.py │ └── gen_pyi.py ├── render_junit.py ├── rules │ ├── BUILD │ ├── METADATA.bzl │ ├── cu.bzl │ └── workspace.bzl ├── rules_cc │ └── cuda_support.patch ├── setup_helpers │ ├── BUILD.bazel │ ├── __init__.py │ ├── build.bzl │ ├── cmake.py │ ├── cmake_utils.py │ ├── env.py │ ├── gen.py │ ├── gen_unboxing.py │ ├── gen_version_header.py │ ├── generate_code.py │ └── generate_linker_script.py ├── shared │ ├── __init__.py │ ├── logging_utils.py │ └── module_loader.py ├── stats │ ├── README.md │ ├── __init__.py │ ├── check_disabled_tests.py │ ├── export_test_times.py │ ├── import_test_stats.py │ ├── monitor.py │ ├── sccache_stats_to_benchmark_format.py │ ├── test_dashboard.py │ ├── upload_artifacts.py │ ├── upload_dynamo_perf_stats.py │ ├── upload_external_contrib_stats.py │ ├── upload_metrics.py │ ├── upload_sccache_stats.py │ ├── upload_stats_lib.py │ ├── upload_test_stats.py │ ├── upload_test_stats_intermediate.py │ ├── upload_test_stats_running_jobs.py │ ├── upload_utilization_stats │ │ ├── test_upload_utilization_stats.py │ │ └── upload_utilization_stats.py │ └── utilization_stats_lib.py ├── substitute.py ├── test │ ├── docstring_linter_testdata │ │ ├── python_code.py.txt │ │ ├── python_code.py.txt.json │ │ └── python_code.py.txt.lintrunner │ ├── gen_operators_yaml_test.py │ ├── gen_oplist_test.py │ ├── heuristics │ │ ├── __init__.py │ │ ├── test_heuristics.py │ │ ├── test_interface.py │ │ └── test_utils.py │ ├── linter_test_case.py │ ├── set_linter_testdata │ │ ├── includes.py.txt │ │ ├── includes.py.txt.json │ │ ├── includes.py.txt.lintrunner │ │ ├── includes.py.txt.python │ │ ├── includes_doesnt_change.py.txt │ │ ├── includes_doesnt_change.py.txt.json │ │ ├── includes_doesnt_change.py.txt.lintrunner │ │ ├── includes_doesnt_change.py.txt.python │ │ ├── python_code.py.txt │ │ ├── python_code.py.txt.json │ │ ├── python_code.py.txt.lintrunner │ │ └── python_code.py.txt.python │ ├── test_cmake.py │ ├── test_codegen.py │ ├── test_codegen_model.py │ ├── test_create_alerts.py │ ├── test_docstring_linter.py │ ├── test_executorch_custom_ops.py │ ├── test_executorch_gen.py │ ├── test_executorch_signatures.py │ ├── test_executorch_types.py │ ├── test_executorch_unboxing.py │ ├── test_gen_backend_stubs.py │ ├── test_selective_build.py │ ├── test_set_linter.py │ ├── test_test_run.py │ ├── test_test_selections.py │ ├── test_upload_stats_lib.py │ ├── test_upload_test_stats.py │ ├── test_utils.py │ └── test_vulkan_codegen.py ├── testing │ ├── __init__.py │ ├── clickhouse.py │ ├── discover_tests.py │ ├── do_target_determination_for_s3.py │ ├── explicit_ci_jobs.py │ ├── modulefinder_determinator.py │ ├── target_determination │ │ ├── determinator.py │ │ ├── gen_artifact.py │ │ └── heuristics │ │ │ ├── __init__.py │ │ │ ├── correlated_with_historical_failures.py │ │ │ ├── edited_by_pr.py │ │ │ ├── filepath.py │ │ │ ├── historical_class_failure_correlation.py │ │ │ ├── historical_edited_files.py │ │ │ ├── interface.py │ │ │ ├── llm.py │ │ │ ├── mentioned_in_pr.py │ │ │ ├── previously_failed_in_pr.py │ │ │ ├── profiling.py │ │ │ ├── public_bindings.py │ │ │ └── utils.py │ ├── test_run.py │ ├── test_selections.py │ ├── update_slow_tests.py │ └── upload_artifacts.py ├── update_masked_docs.py └── vscode_settings.py ├── torch ├── CMakeLists.txt ├── README.txt ├── _C │ ├── _VariableFunctions.pyi.in │ ├── __init__.pyi.in │ ├── _aoti.pyi │ ├── _autograd.pyi │ ├── _cpu.pyi │ ├── _cudnn.pyi │ ├── _cusparselt.pyi │ ├── _distributed_autograd.pyi │ ├── _distributed_c10d.pyi │ ├── _distributed_rpc.pyi │ ├── _distributed_rpc_testing.pyi │ ├── _dynamo │ │ ├── __init__.pyi │ │ ├── compiled_autograd.pyi │ │ ├── eval_frame.pyi │ │ └── guards.pyi │ ├── _export.pyi │ ├── _functions.pyi │ ├── _functorch.pyi │ ├── _instruction_counter.pyi │ ├── _itt.pyi │ ├── _lazy.pyi │ ├── _lazy_ts_backend.pyi │ ├── _monitor.pyi │ ├── _nn.pyi.in │ ├── _nvtx.pyi │ ├── _onnx.pyi │ ├── _profiler.pyi │ ├── _verbose.pyi │ ├── build.bzl │ └── return_types.pyi.in ├── _C_flatbuffer │ └── __init__.pyi ├── _VF.py ├── __config__.py ├── __future__.py ├── __init__.py ├── _appdirs.py ├── _awaits │ └── __init__.py ├── _classes.py ├── _compile.py ├── _custom_op │ ├── __init__.py │ ├── autograd.py │ └── impl.py ├── _custom_ops.py ├── _decomp │ ├── __init__.py │ ├── decompositions.py │ ├── decompositions_for_jvp.py │ └── decompositions_for_rng.py ├── _deploy.py ├── _dispatch │ ├── __init__.py │ └── python.py ├── _dynamo │ ├── __init__.py │ ├── _trace_wrapped_higher_order_op.py │ ├── backends │ │ ├── __init__.py │ │ ├── common.py │ │ ├── cudagraphs.py │ │ ├── debugging.py │ │ ├── distributed.py │ │ ├── inductor.py │ │ ├── onnxrt.py │ │ ├── registry.py │ │ ├── tensorrt.py │ │ ├── torchxla.py │ │ └── tvm.py │ ├── bytecode_analysis.py │ ├── bytecode_transformation.py │ ├── cache_size.py │ ├── callback.py │ ├── code_context.py │ ├── codegen.py │ ├── compiled_autograd.py │ ├── comptime.py │ ├── config.py │ ├── convert_frame.py │ ├── create_parameter_op.py │ ├── current_scope_id.py │ ├── debug_utils.py │ ├── decorators.py │ ├── device_interface.py │ ├── distributed.py │ ├── eval_frame.py │ ├── exc.py │ ├── external_utils.py │ ├── funcname_cache.py │ ├── graph_break_hints.py │ ├── graph_deduplication.py │ ├── graph_region_tracker.py │ ├── guards.py │ ├── hooks.py │ ├── logging.py │ ├── metrics_context.py │ ├── mutation_guard.py │ ├── output_graph.py │ ├── pgo.py │ ├── polyfills │ │ ├── __init__.py │ │ ├── builtins.py │ │ ├── functools.py │ │ ├── fx.py │ │ ├── itertools.py │ │ ├── loader.py │ │ ├── operator.py │ │ ├── os.py │ │ ├── pytree.py │ │ └── sys.py │ ├── profiler.py │ ├── replay_record.py │ ├── repro │ │ ├── __init__.py │ │ ├── after_aot.py │ │ ├── after_dynamo.py │ │ └── aoti.py │ ├── resume_execution.py │ ├── side_effects.py │ ├── source.py │ ├── symbolic_convert.py │ ├── tensor_version_op.py │ ├── test_case.py │ ├── test_minifier_common.py │ ├── testing.py │ ├── trace_rules.py │ ├── types.py │ ├── utils.py │ └── variables │ │ ├── __init__.py │ │ ├── base.py │ │ ├── builder.py │ │ ├── builtin.py │ │ ├── constant.py │ │ ├── ctx_manager.py │ │ ├── dicts.py │ │ ├── distributed.py │ │ ├── functions.py │ │ ├── higher_order_ops.py │ │ ├── iter.py │ │ ├── lazy.py │ │ ├── lists.py │ │ ├── misc.py │ │ ├── nn_module.py │ │ ├── optimizer.py │ │ ├── script_object.py │ │ ├── sdpa.py │ │ ├── tensor.py │ │ ├── torch.py │ │ ├── torch_function.py │ │ └── user_defined.py ├── _environment.py ├── _export │ ├── __init__.py │ ├── converter.py │ ├── db │ │ ├── __init__.py │ │ ├── case.py │ │ ├── examples │ │ │ ├── __init__.py │ │ │ ├── assume_constant_result.py │ │ │ ├── autograd_function.py │ │ │ ├── class_method.py │ │ │ ├── cond_branch_class_method.py │ │ │ ├── cond_branch_nested_function.py │ │ │ ├── cond_branch_nonlocal_variables.py │ │ │ ├── cond_closed_over_variable.py │ │ │ ├── cond_operands.py │ │ │ ├── cond_predicate.py │ │ │ ├── constrain_as_size_example.py │ │ │ ├── constrain_as_value_example.py │ │ │ ├── decorator.py │ │ │ ├── dictionary.py │ │ │ ├── dynamic_shape_assert.py │ │ │ ├── dynamic_shape_constructor.py │ │ │ ├── dynamic_shape_if_guard.py │ │ │ ├── dynamic_shape_map.py │ │ │ ├── dynamic_shape_round.py │ │ │ ├── dynamic_shape_slicing.py │ │ │ ├── dynamic_shape_view.py │ │ │ ├── fn_with_kwargs.py │ │ │ ├── list_contains.py │ │ │ ├── list_unpack.py │ │ │ ├── model_attr_mutation.py │ │ │ ├── nested_function.py │ │ │ ├── null_context_manager.py │ │ │ ├── optional_input.py │ │ │ ├── pytree_flatten.py │ │ │ ├── scalar_output.py │ │ │ ├── specialized_attribute.py │ │ │ ├── static_for_loop.py │ │ │ ├── static_if.py │ │ │ ├── tensor_setattr.py │ │ │ ├── type_reflection_method.py │ │ │ ├── unsupported_operator.py │ │ │ └── user_input_mutation.py │ │ ├── gen_example.py │ │ └── logging.py │ ├── error.py │ ├── non_strict_utils.py │ ├── pass_base.py │ ├── pass_infra │ │ ├── __init__.py │ │ ├── node_metadata.py │ │ └── proxy_value.py │ ├── passes │ │ ├── __init__.py │ │ ├── _node_metadata_hook.py │ │ ├── add_runtime_assertions_for_constraints_pass.py │ │ ├── collect_tracepoints_pass.py │ │ ├── constant_folding.py │ │ ├── functionalize_side_effectful_ops_pass.py │ │ ├── insert_custom_op_guards.py │ │ ├── lift_constants_pass.py │ │ ├── remove_runtime_assertions.py │ │ ├── replace_autocast_with_hop_pass.py │ │ ├── replace_quantized_ops_with_standard_ops_pass.py │ │ ├── replace_set_grad_with_hop_pass.py │ │ ├── replace_view_ops_with_view_copy_ops_pass.py │ │ └── replace_with_hop_pass_util.py │ ├── serde │ │ ├── __init__.py │ │ ├── aoti_schema.py │ │ ├── dynamic_shapes.py │ │ ├── export_schema.thrift │ │ ├── gen-cpp2 │ │ │ ├── export_schema_constants.h │ │ │ ├── export_schema_types.h │ │ │ └── export_schema_types_custom_protocol.h │ │ ├── schema.py │ │ ├── schema.yaml │ │ ├── schema_check.py │ │ ├── serialize.py │ │ └── union.py │ ├── tools.py │ ├── utils.py │ ├── verifier.py │ └── wrappers.py ├── _functorch │ ├── __init__.py │ ├── _activation_checkpointing │ │ ├── __init__.py │ │ ├── ac_logging_utils.py │ │ ├── graph_info_provider.py │ │ ├── knapsack.py │ │ └── knapsack_evaluator.py │ ├── _aot_autograd │ │ ├── __init__.py │ │ ├── autograd_cache.py │ │ ├── collect_metadata_analysis.py │ │ ├── dispatch_and_compile_graph.py │ │ ├── functional_utils.py │ │ ├── input_output_analysis.py │ │ ├── jit_compile_runtime_wrappers.py │ │ ├── logging_utils.py │ │ ├── runtime_wrappers.py │ │ ├── schemas.py │ │ ├── subclass_parametrization.py │ │ ├── subclass_utils.py │ │ ├── traced_function_transforms.py │ │ └── utils.py │ ├── aot_autograd.py │ ├── apis.py │ ├── autograd_function.py │ ├── batch_norm_replacement.py │ ├── benchmark_utils.py │ ├── compile_utils.py │ ├── compilers.py │ ├── config.py │ ├── deprecated.py │ ├── eager_transforms.py │ ├── functional_call.py │ ├── fx_minifier.py │ ├── make_functional.py │ ├── partitioners.py │ ├── pyfunctorch.py │ ├── python_key.py │ ├── pytree_hacks.py │ ├── top_operators_github_usage.py │ ├── utils.py │ └── vmap.py ├── _guards.py ├── _higher_order_ops │ ├── __init__.py │ ├── _invoke_quant.py │ ├── aoti_call_delegate.py │ ├── associative_scan.py │ ├── auto_functionalize.py │ ├── base_hop.py │ ├── cond.py │ ├── effects.py │ ├── executorch_call_delegate.py │ ├── flat_apply.py │ ├── flex_attention.py │ ├── foreach_map.py │ ├── hints_wrap.py │ ├── invoke_subgraph.py │ ├── map.py │ ├── out_dtype.py │ ├── run_const_graph.py │ ├── scan.py │ ├── strict_mode.py │ ├── torchbind.py │ ├── triton_kernel_wrap.py │ ├── utils.py │ ├── while_loop.py │ └── wrap.py ├── _inductor │ ├── __init__.py │ ├── analyze_preserves_zero_mask.py │ ├── aoti_eager.py │ ├── async_compile.py │ ├── autoheuristic │ │ ├── __init__.py │ │ ├── artifacts │ │ │ ├── _MMRankingA100.py │ │ │ ├── _MMRankingH100.py │ │ │ ├── _MixedMMA100.py │ │ │ ├── _MixedMMH100.py │ │ │ ├── _PadMMA100.py │ │ │ └── __init__.py │ │ ├── autoheuristic.py │ │ ├── autoheuristic_utils.py │ │ ├── learned_heuristic_controller.py │ │ └── learnedheuristic_interface.py │ ├── autotune_process.py │ ├── bounds.py │ ├── choices.py │ ├── codecache.py │ ├── codegen │ │ ├── __init__.py │ │ ├── aoti_hipify_utils.py │ │ ├── aoti_runtime │ │ │ └── interface.cpp │ │ ├── block_analysis.py │ │ ├── common.py │ │ ├── cpp.py │ │ ├── cpp_bmm_template.py │ │ ├── cpp_flex_attention_template.py │ │ ├── cpp_gemm_template.py │ │ ├── cpp_grouped_gemm_template.py │ │ ├── cpp_micro_gemm.py │ │ ├── cpp_prefix.h │ │ ├── cpp_template.py │ │ ├── cpp_template_kernel.py │ │ ├── cpp_utils.py │ │ ├── cpp_wrapper_cpu.py │ │ ├── cpp_wrapper_cpu_array_ref.py │ │ ├── cpp_wrapper_gpu.py │ │ ├── cpu_device_op_overrides.py │ │ ├── cuda │ │ │ ├── __init__.py │ │ │ ├── cuda_cpp_scheduling.py │ │ │ ├── cuda_env.py │ │ │ ├── cuda_kernel.py │ │ │ ├── cuda_template.py │ │ │ ├── cutlass_lib_extensions │ │ │ │ ├── __init__.py │ │ │ │ └── gemm_operation_extensions.py │ │ │ ├── cutlass_utils.py │ │ │ ├── device_op_overrides.py │ │ │ └── gemm_template.py │ │ ├── cuda_combined_scheduling.py │ │ ├── debug_utils.py │ │ ├── halide.py │ │ ├── memory_planning.py │ │ ├── mps.py │ │ ├── mps_device_op_overrides.py │ │ ├── multi_kernel.py │ │ ├── rocm │ │ │ ├── __init__.py │ │ │ ├── ck_conv_template.py │ │ │ ├── ck_template.py │ │ │ ├── ck_universal_gemm_template.py │ │ │ ├── compile_command.py │ │ │ ├── rocm_benchmark_request.py │ │ │ ├── rocm_cpp_scheduling.py │ │ │ ├── rocm_kernel.py │ │ │ ├── rocm_template.py │ │ │ └── rocm_template_buffer.py │ │ ├── simd.py │ │ ├── simd_kernel_features.py │ │ ├── triton.py │ │ ├── triton_combo_kernel.py │ │ ├── triton_split_scan.py │ │ ├── triton_utils.py │ │ ├── wrapper.py │ │ └── xpu │ │ │ ├── __init__.py │ │ │ └── device_op_overrides.py │ ├── comm_analysis.py │ ├── comm_lowering.py │ ├── comms.py │ ├── compile_fx.py │ ├── compile_worker │ │ ├── __init__.py │ │ ├── __main__.py │ │ ├── subproc_pool.py │ │ └── watchdog.py │ ├── compiler_bisector.py │ ├── config.py │ ├── constant_folding.py │ ├── cpp_builder.py │ ├── cpu_vec_isa.py │ ├── cudagraph_trees.py │ ├── cudagraph_utils.py │ ├── custom_graph_pass.py │ ├── debug.py │ ├── decomposition.py │ ├── dependencies.py │ ├── dtype_propagation.py │ ├── exc.py │ ├── extern_node_serializer.py │ ├── freezing.py │ ├── freezing_utils.py │ ├── fuzzer.py │ ├── fx_passes │ │ ├── README.md │ │ ├── __init__.py │ │ ├── b2b_gemm.py │ │ ├── binary_folding.py │ │ ├── ddp_fusion.py │ │ ├── decompose_mem_bound_mm.py │ │ ├── dedupe_symint_uses.py │ │ ├── efficient_conv_bn_eval.py │ │ ├── freezing_patterns.py │ │ ├── fuse_attention.py │ │ ├── group_batch_fusion.py │ │ ├── joint_graph.py │ │ ├── micro_pipeline_tp.py │ │ ├── misc_patterns.py │ │ ├── mkldnn_fusion.py │ │ ├── numeric_utils.py │ │ ├── pad_mm.py │ │ ├── post_grad.py │ │ ├── pre_grad.py │ │ ├── quantization.py │ │ ├── reinplace.py │ │ ├── replace_random.py │ │ ├── serialized_patterns │ │ │ ├── __init__.py │ │ │ ├── _sfdp_pattern_1.py │ │ │ ├── _sfdp_pattern_10.py │ │ │ ├── _sfdp_pattern_11.py │ │ │ ├── _sfdp_pattern_12.py │ │ │ ├── _sfdp_pattern_13.py │ │ │ ├── _sfdp_pattern_14.py │ │ │ ├── _sfdp_pattern_15.py │ │ │ ├── _sfdp_pattern_16.py │ │ │ ├── _sfdp_pattern_17.py │ │ │ ├── _sfdp_pattern_18.py │ │ │ ├── _sfdp_pattern_19.py │ │ │ ├── _sfdp_pattern_2.py │ │ │ ├── _sfdp_pattern_3.py │ │ │ ├── _sfdp_pattern_4.py │ │ │ ├── _sfdp_pattern_5.py │ │ │ ├── _sfdp_pattern_6.py │ │ │ ├── _sfdp_pattern_7.py │ │ │ ├── _sfdp_pattern_8.py │ │ │ ├── _sfdp_pattern_9.py │ │ │ ├── addmm_pattern.py │ │ │ ├── bmm_pattern.py │ │ │ └── mm_pattern.py │ │ └── split_cat.py │ ├── fx_utils.py │ ├── graph.py │ ├── hooks.py │ ├── index_propagation.py │ ├── inductor_prims.py │ ├── ir.py │ ├── jagged_lowerings.py │ ├── kernel │ │ ├── __init__.py │ │ ├── bmm.py │ │ ├── conv.py │ │ ├── flex_attention.py │ │ ├── flex_decoding.py │ │ ├── mm.py │ │ ├── mm_common.py │ │ ├── mm_plus_mm.py │ │ └── mm_scaled.py │ ├── loop_body.py │ ├── lowering.py │ ├── memory.py │ ├── metrics.py │ ├── mkldnn_ir.py │ ├── mkldnn_lowerings.py │ ├── mock_cache.py │ ├── ops_handler.py │ ├── optimize_indexing.py │ ├── output_code.py │ ├── package │ │ ├── __init__.py │ │ ├── build_package.py │ │ ├── package.py │ │ └── pt2_archive_constants.py │ ├── pattern_matcher.py │ ├── quantized_lowerings.py │ ├── remote_cache.py │ ├── runtime │ │ ├── README.md │ │ ├── __init__.py │ │ ├── autotune_cache.py │ │ ├── benchmarking.py │ │ ├── cache_dir_utils.py │ │ ├── compile_tasks.py │ │ ├── coordinate_descent_tuner.py │ │ ├── halide_helpers.py │ │ ├── hints.py │ │ ├── runtime_utils.py │ │ ├── triton_compat.py │ │ ├── triton_helpers.py │ │ └── triton_heuristics.py │ ├── scheduler.py │ ├── script.ld │ ├── select_algorithm.py │ ├── sizevars.py │ ├── subgraph_lowering.py │ ├── test_case.py │ ├── test_operators.py │ ├── triton_bundler.py │ ├── utils.py │ ├── virtualized.py │ └── wrapper_benchmark.py ├── _jit_internal.py ├── _lazy │ ├── __init__.py │ ├── closure.py │ ├── computation.py │ ├── config.py │ ├── debug.py │ ├── device_context.py │ ├── extract_compiled_graph.py │ ├── ir_cache.py │ ├── metrics.py │ ├── tensor_factory_functions.py │ └── ts_backend.py ├── _library │ ├── __init__.py │ ├── autograd.py │ ├── custom_ops.py │ ├── fake_class_registry.py │ ├── fake_impl.py │ ├── infer_schema.py │ ├── simple_registry.py │ ├── triton.py │ └── utils.py ├── _linalg_utils.py ├── _lobpcg.py ├── _logging │ ├── __init__.py │ ├── _internal.py │ ├── _registrations.py │ ├── scribe.py │ └── structured.py ├── _lowrank.py ├── _meta_registrations.py ├── _namedtensor_internals.py ├── _numpy │ ├── README.md │ ├── __init__.py │ ├── _binary_ufuncs_impl.py │ ├── _casting_dicts.py │ ├── _dtypes.py │ ├── _dtypes_impl.py │ ├── _funcs.py │ ├── _funcs_impl.py │ ├── _getlimits.py │ ├── _ndarray.py │ ├── _normalizations.py │ ├── _reductions_impl.py │ ├── _ufuncs.py │ ├── _unary_ufuncs_impl.py │ ├── _util.py │ ├── fft.py │ ├── linalg.py │ ├── random.py │ └── testing │ │ ├── __init__.py │ │ └── utils.py ├── _ops.py ├── _prims │ ├── __init__.py │ ├── context.py │ ├── debug_prims.py │ ├── executor.py │ └── rng_prims.py ├── _prims_common │ ├── __init__.py │ └── wrappers.py ├── _python_dispatcher.py ├── _refs │ ├── __init__.py │ ├── _conversions.py │ ├── fft.py │ ├── linalg │ │ └── __init__.py │ ├── nn │ │ ├── __init__.py │ │ └── functional │ │ │ └── __init__.py │ └── special │ │ └── __init__.py ├── _size_docs.py ├── _sources.py ├── _storage_docs.py ├── _streambase.py ├── _strobelight │ ├── __init__.py │ ├── cli_function_profiler.py │ └── compile_time_profiler.py ├── _subclasses │ ├── __init__.py │ ├── _fake_tensor_utils.py │ ├── fake_impls.py │ ├── fake_tensor.py │ ├── fake_utils.py │ ├── functional_tensor.py │ ├── meta_utils.py │ └── schema_check_mode.py ├── _tensor.py ├── _tensor_docs.py ├── _tensor_str.py ├── _thread_safe_fork.py ├── _torch_docs.py ├── _utils.py ├── _utils_internal.py ├── _vendor │ ├── README.md │ ├── __init__.py │ └── packaging │ │ ├── LICENSE │ │ ├── LICENSE.APACHE │ │ ├── LICENSE.BSD │ │ ├── __init__.py │ │ ├── _structures.py │ │ └── version.py ├── _vmap_internals.py ├── _weights_only_unpickler.py ├── abi-check.cpp ├── accelerator │ ├── __init__.py │ └── _utils.py ├── amp │ ├── __init__.py │ ├── autocast_mode.py │ └── grad_scaler.py ├── ao │ ├── __init__.py │ ├── nn │ │ ├── __init__.py │ │ ├── intrinsic │ │ │ ├── __init__.py │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ └── fused.py │ │ │ ├── qat │ │ │ │ ├── __init__.py │ │ │ │ └── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── conv_fused.py │ │ │ │ │ ├── linear_fused.py │ │ │ │ │ └── linear_relu.py │ │ │ └── quantized │ │ │ │ ├── __init__.py │ │ │ │ ├── dynamic │ │ │ │ ├── __init__.py │ │ │ │ └── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── linear_relu.py │ │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── bn_relu.py │ │ │ │ ├── conv_add.py │ │ │ │ ├── conv_relu.py │ │ │ │ └── linear_relu.py │ │ ├── qat │ │ │ ├── __init__.py │ │ │ ├── dynamic │ │ │ │ ├── __init__.py │ │ │ │ └── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── linear.py │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── conv.py │ │ │ │ ├── embedding_ops.py │ │ │ │ └── linear.py │ │ ├── quantizable │ │ │ ├── __init__.py │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── activation.py │ │ │ │ └── rnn.py │ │ ├── quantized │ │ │ ├── __init__.py │ │ │ ├── dynamic │ │ │ │ ├── __init__.py │ │ │ │ └── modules │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── conv.py │ │ │ │ │ ├── linear.py │ │ │ │ │ └── rnn.py │ │ │ ├── functional.py │ │ │ ├── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── activation.py │ │ │ │ ├── batchnorm.py │ │ │ │ ├── conv.py │ │ │ │ ├── dropout.py │ │ │ │ ├── embedding_ops.py │ │ │ │ ├── functional_modules.py │ │ │ │ ├── linear.py │ │ │ │ ├── normalization.py │ │ │ │ ├── rnn.py │ │ │ │ └── utils.py │ │ │ └── reference │ │ │ │ ├── __init__.py │ │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── conv.py │ │ │ │ ├── linear.py │ │ │ │ ├── rnn.py │ │ │ │ ├── sparse.py │ │ │ │ └── utils.py │ │ └── sparse │ │ │ ├── __init__.py │ │ │ └── quantized │ │ │ ├── __init__.py │ │ │ ├── dynamic │ │ │ ├── __init__.py │ │ │ └── linear.py │ │ │ ├── linear.py │ │ │ └── utils.py │ ├── ns │ │ ├── __init__.py │ │ ├── _numeric_suite.py │ │ ├── _numeric_suite_fx.py │ │ └── fx │ │ │ ├── __init__.py │ │ │ ├── graph_matcher.py │ │ │ ├── graph_passes.py │ │ │ ├── mappings.py │ │ │ ├── n_shadows_utils.py │ │ │ ├── ns_types.py │ │ │ ├── pattern_utils.py │ │ │ ├── qconfig_multi_mapping.py │ │ │ ├── utils.py │ │ │ └── weight_utils.py │ ├── pruning │ │ ├── __init__.py │ │ ├── _experimental │ │ │ ├── __init__.py │ │ │ ├── activation_sparsifier │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ └── activation_sparsifier.py │ │ │ ├── data_scheduler │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ └── base_data_scheduler.py │ │ │ ├── data_sparsifier │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── base_data_sparsifier.py │ │ │ │ ├── benchmarks │ │ │ │ │ ├── README.md │ │ │ │ │ ├── dlrm_utils.py │ │ │ │ │ ├── evaluate_disk_savings.py │ │ │ │ │ ├── evaluate_forward_time.py │ │ │ │ │ ├── evaluate_model_metrics.py │ │ │ │ │ └── images │ │ │ │ │ │ ├── accuracy.png │ │ │ │ │ │ ├── disk_savings.png │ │ │ │ │ │ └── forward_time.png │ │ │ │ ├── data_norm_sparsifier.py │ │ │ │ ├── lightning │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── callbacks │ │ │ │ │ │ ├── README.md │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── _data_sparstity_utils.py │ │ │ │ │ │ └── data_sparsity.py │ │ │ │ │ └── tests │ │ │ │ │ │ └── test_callbacks.py │ │ │ │ └── quantization_utils.py │ │ │ └── pruner │ │ │ │ ├── FPGM_pruner.py │ │ │ │ ├── README.md │ │ │ │ ├── __init__.py │ │ │ │ ├── base_structured_sparsifier.py │ │ │ │ ├── images │ │ │ │ ├── prune_1.png │ │ │ │ ├── prune_2.png │ │ │ │ ├── prune_3.png │ │ │ │ ├── prune_4.png │ │ │ │ ├── prune_5.png │ │ │ │ └── prune_6.png │ │ │ │ ├── lstm_saliency_pruner.py │ │ │ │ ├── match_utils.py │ │ │ │ ├── parametrization.py │ │ │ │ ├── prune_functions.py │ │ │ │ └── saliency_pruner.py │ │ ├── _mappings.py │ │ ├── scheduler │ │ │ ├── __init__.py │ │ │ ├── base_scheduler.py │ │ │ ├── cubic_scheduler.py │ │ │ └── lambda_scheduler.py │ │ └── sparsifier │ │ │ ├── __init__.py │ │ │ ├── base_sparsifier.py │ │ │ ├── nearly_diagonal_sparsifier.py │ │ │ ├── utils.py │ │ │ └── weight_norm_sparsifier.py │ └── quantization │ │ ├── __init__.py │ │ ├── _correct_bias.py │ │ ├── _equalize.py │ │ ├── _learnable_fake_quantize.py │ │ ├── backend_config │ │ ├── README.md │ │ ├── __init__.py │ │ ├── _common_operator_config_utils.py │ │ ├── _qnnpack_pt2e.py │ │ ├── backend_config.py │ │ ├── executorch.py │ │ ├── fbgemm.py │ │ ├── native.py │ │ ├── observation_type.py │ │ ├── onednn.py │ │ ├── qnnpack.py │ │ ├── tensorrt.py │ │ ├── utils.py │ │ └── x86.py │ │ ├── experimental │ │ ├── APoT_tensor.py │ │ ├── adaround_fake_quantize.py │ │ ├── adaround_loss.py │ │ ├── adaround_optimization.py │ │ ├── apot_utils.py │ │ ├── fake_quantize.py │ │ ├── fake_quantize_function.py │ │ ├── linear.py │ │ ├── observer.py │ │ ├── qconfig.py │ │ └── quantizer.py │ │ ├── fake_quantize.py │ │ ├── fuse_modules.py │ │ ├── fuser_method_mappings.py │ │ ├── fx │ │ ├── README.md │ │ ├── __init__.py │ │ ├── _decomposed.py │ │ ├── _equalize.py │ │ ├── _lower_to_native_backend.py │ │ ├── _model_report │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── detector.py │ │ │ ├── model_report.py │ │ │ ├── model_report_observer.py │ │ │ └── model_report_visualizer.py │ │ ├── convert.py │ │ ├── custom_config.py │ │ ├── fuse.py │ │ ├── fuse_handler.py │ │ ├── graph_module.py │ │ ├── lower_to_fbgemm.py │ │ ├── lower_to_qnnpack.py │ │ ├── lstm_utils.py │ │ ├── match_utils.py │ │ ├── pattern_utils.py │ │ ├── prepare.py │ │ ├── qconfig_mapping_utils.py │ │ ├── quantize_handler.py │ │ ├── tracer.py │ │ └── utils.py │ │ ├── observer.py │ │ ├── pattern.md │ │ ├── pt2e │ │ ├── __init__.py │ │ ├── _affine_quantization.py │ │ ├── _numeric_debugger.py │ │ ├── duplicate_dq_pass.py │ │ ├── export_utils.py │ │ ├── graph_utils.py │ │ ├── port_metadata_pass.py │ │ ├── prepare.py │ │ ├── qat_utils.py │ │ ├── representation │ │ │ ├── __init__.py │ │ │ └── rewrite.py │ │ └── utils.py │ │ ├── qconfig.py │ │ ├── qconfig_mapping.py │ │ ├── quant_type.py │ │ ├── quantization_mappings.py │ │ ├── quantize.py │ │ ├── quantize_fx.py │ │ ├── quantize_jit.py │ │ ├── quantize_pt2e.py │ │ ├── quantizer │ │ ├── __init__.py │ │ ├── composable_quantizer.py │ │ ├── embedding_quantizer.py │ │ ├── quantizer.py │ │ ├── utils.py │ │ ├── x86_inductor_quantizer.py │ │ ├── xnnpack_quantizer.py │ │ ├── xnnpack_quantizer_utils.py │ │ └── xpu_inductor_quantizer.py │ │ ├── stubs.py │ │ └── utils.py ├── autograd │ ├── __init__.py │ ├── _functions │ │ ├── __init__.py │ │ ├── replace.vim │ │ ├── tensor.py │ │ └── utils.py │ ├── anomaly_mode.py │ ├── forward_ad.py │ ├── function.py │ ├── functional.py │ ├── grad_mode.py │ ├── gradcheck.py │ ├── graph.py │ ├── profiler.py │ ├── profiler_legacy.py │ ├── profiler_util.py │ └── variable.py ├── backends │ ├── __init__.py │ ├── _coreml │ │ ├── __init__.py │ │ └── preprocess.py │ ├── _nnapi │ │ ├── __init__.py │ │ ├── prepare.py │ │ └── serializer.py │ ├── cpu │ │ └── __init__.py │ ├── cuda │ │ └── __init__.py │ ├── cudnn │ │ ├── __init__.py │ │ └── rnn.py │ ├── cusparselt │ │ └── __init__.py │ ├── kleidiai │ │ └── __init__.py │ ├── mha │ │ └── __init__.py │ ├── mkl │ │ └── __init__.py │ ├── mkldnn │ │ └── __init__.py │ ├── mps │ │ └── __init__.py │ ├── nnpack │ │ └── __init__.py │ ├── openmp │ │ └── __init__.py │ ├── opt_einsum │ │ └── __init__.py │ ├── quantized │ │ └── __init__.py │ ├── xeon │ │ ├── __init__.py │ │ └── run_cpu.py │ └── xnnpack │ │ └── __init__.py ├── compiler │ ├── __init__.py │ ├── _cache.py │ └── config.py ├── contrib │ ├── __init__.py │ └── _tensorboard_vis.py ├── cpu │ ├── __init__.py │ └── amp │ │ ├── __init__.py │ │ ├── autocast_mode.py │ │ └── grad_scaler.py ├── csrc │ ├── CudaIPCTypes.cpp │ ├── CudaIPCTypes.h │ ├── DataLoader.cpp │ ├── DataLoader.h │ ├── Device.cpp │ ├── Device.h │ ├── DeviceAccelerator.cpp │ ├── DeviceAccelerator.h │ ├── Dtype.cpp │ ├── Dtype.h │ ├── DynamicTypes.cpp │ ├── DynamicTypes.h │ ├── Event.cpp │ ├── Event.h │ ├── Exceptions.cpp │ ├── Exceptions.h │ ├── Export.h │ ├── Generator.cpp │ ├── Generator.h │ ├── Layout.cpp │ ├── Layout.h │ ├── MemoryFormat.cpp │ ├── MemoryFormat.h │ ├── Module.cpp │ ├── Module.h │ ├── PyInterpreter.cpp │ ├── PyInterpreter.h │ ├── QScheme.cpp │ ├── QScheme.h │ ├── README.md │ ├── Size.cpp │ ├── Size.h │ ├── Storage.cpp │ ├── Storage.h │ ├── StorageMethods.cpp │ ├── StorageMethods.h │ ├── StorageSharing.cpp │ ├── StorageSharing.h │ ├── Stream.cpp │ ├── Stream.h │ ├── THConcat.h │ ├── THP.h │ ├── TypeInfo.cpp │ ├── TypeInfo.h │ ├── Types.h │ ├── api │ │ ├── include │ │ │ └── torch │ │ │ │ ├── all.h │ │ │ │ ├── arg.h │ │ │ │ ├── autograd.h │ │ │ │ ├── cuda.h │ │ │ │ ├── data.h │ │ │ │ ├── data │ │ │ │ ├── dataloader.h │ │ │ │ ├── dataloader │ │ │ │ │ ├── base.h │ │ │ │ │ ├── stateful.h │ │ │ │ │ └── stateless.h │ │ │ │ ├── dataloader_options.h │ │ │ │ ├── datasets.h │ │ │ │ ├── datasets │ │ │ │ │ ├── base.h │ │ │ │ │ ├── chunk.h │ │ │ │ │ ├── map.h │ │ │ │ │ ├── mnist.h │ │ │ │ │ ├── shared.h │ │ │ │ │ ├── stateful.h │ │ │ │ │ └── tensor.h │ │ │ │ ├── detail │ │ │ │ │ ├── data_shuttle.h │ │ │ │ │ ├── queue.h │ │ │ │ │ └── sequencers.h │ │ │ │ ├── example.h │ │ │ │ ├── iterator.h │ │ │ │ ├── samplers.h │ │ │ │ ├── samplers │ │ │ │ │ ├── base.h │ │ │ │ │ ├── custom_batch_request.h │ │ │ │ │ ├── distributed.h │ │ │ │ │ ├── random.h │ │ │ │ │ ├── sequential.h │ │ │ │ │ ├── serialize.h │ │ │ │ │ └── stream.h │ │ │ │ ├── transforms.h │ │ │ │ ├── transforms │ │ │ │ │ ├── base.h │ │ │ │ │ ├── collate.h │ │ │ │ │ ├── lambda.h │ │ │ │ │ ├── stack.h │ │ │ │ │ └── tensor.h │ │ │ │ └── worker_exception.h │ │ │ │ ├── detail │ │ │ │ ├── TensorDataContainer.h │ │ │ │ └── static.h │ │ │ │ ├── enum.h │ │ │ │ ├── expanding_array.h │ │ │ │ ├── fft.h │ │ │ │ ├── imethod.h │ │ │ │ ├── jit.h │ │ │ │ ├── mps.h │ │ │ │ ├── nested.h │ │ │ │ ├── nn.h │ │ │ │ ├── nn │ │ │ │ ├── cloneable.h │ │ │ │ ├── functional.h │ │ │ │ ├── functional │ │ │ │ │ ├── activation.h │ │ │ │ │ ├── batchnorm.h │ │ │ │ │ ├── conv.h │ │ │ │ │ ├── distance.h │ │ │ │ │ ├── dropout.h │ │ │ │ │ ├── embedding.h │ │ │ │ │ ├── fold.h │ │ │ │ │ ├── instancenorm.h │ │ │ │ │ ├── linear.h │ │ │ │ │ ├── loss.h │ │ │ │ │ ├── normalization.h │ │ │ │ │ ├── padding.h │ │ │ │ │ ├── pixelshuffle.h │ │ │ │ │ ├── pooling.h │ │ │ │ │ ├── upsampling.h │ │ │ │ │ └── vision.h │ │ │ │ ├── init.h │ │ │ │ ├── module.h │ │ │ │ ├── modules.h │ │ │ │ ├── modules │ │ │ │ │ ├── _functions.h │ │ │ │ │ ├── activation.h │ │ │ │ │ ├── adaptive.h │ │ │ │ │ ├── batchnorm.h │ │ │ │ │ ├── common.h │ │ │ │ │ ├── container │ │ │ │ │ │ ├── any.h │ │ │ │ │ │ ├── any_module_holder.h │ │ │ │ │ │ ├── any_value.h │ │ │ │ │ │ ├── functional.h │ │ │ │ │ │ ├── moduledict.h │ │ │ │ │ │ ├── modulelist.h │ │ │ │ │ │ ├── named_any.h │ │ │ │ │ │ ├── parameterdict.h │ │ │ │ │ │ ├── parameterlist.h │ │ │ │ │ │ └── sequential.h │ │ │ │ │ ├── conv.h │ │ │ │ │ ├── distance.h │ │ │ │ │ ├── dropout.h │ │ │ │ │ ├── embedding.h │ │ │ │ │ ├── fold.h │ │ │ │ │ ├── instancenorm.h │ │ │ │ │ ├── linear.h │ │ │ │ │ ├── loss.h │ │ │ │ │ ├── normalization.h │ │ │ │ │ ├── padding.h │ │ │ │ │ ├── pixelshuffle.h │ │ │ │ │ ├── pooling.h │ │ │ │ │ ├── rnn.h │ │ │ │ │ ├── transformer.h │ │ │ │ │ ├── transformercoder.h │ │ │ │ │ ├── transformerlayer.h │ │ │ │ │ ├── upsampling.h │ │ │ │ │ └── utils.h │ │ │ │ ├── options.h │ │ │ │ ├── options │ │ │ │ │ ├── activation.h │ │ │ │ │ ├── adaptive.h │ │ │ │ │ ├── batchnorm.h │ │ │ │ │ ├── conv.h │ │ │ │ │ ├── distance.h │ │ │ │ │ ├── dropout.h │ │ │ │ │ ├── embedding.h │ │ │ │ │ ├── fold.h │ │ │ │ │ ├── instancenorm.h │ │ │ │ │ ├── linear.h │ │ │ │ │ ├── loss.h │ │ │ │ │ ├── normalization.h │ │ │ │ │ ├── padding.h │ │ │ │ │ ├── pixelshuffle.h │ │ │ │ │ ├── pooling.h │ │ │ │ │ ├── rnn.h │ │ │ │ │ ├── transformer.h │ │ │ │ │ ├── transformercoder.h │ │ │ │ │ ├── transformerlayer.h │ │ │ │ │ ├── upsampling.h │ │ │ │ │ └── vision.h │ │ │ │ ├── parallel │ │ │ │ │ └── data_parallel.h │ │ │ │ ├── pimpl-inl.h │ │ │ │ ├── pimpl.h │ │ │ │ ├── utils.h │ │ │ │ └── utils │ │ │ │ │ ├── clip_grad.h │ │ │ │ │ ├── convert_parameters.h │ │ │ │ │ └── rnn.h │ │ │ │ ├── optim.h │ │ │ │ ├── optim │ │ │ │ ├── adagrad.h │ │ │ │ ├── adam.h │ │ │ │ ├── adamw.h │ │ │ │ ├── lbfgs.h │ │ │ │ ├── optimizer.h │ │ │ │ ├── rmsprop.h │ │ │ │ ├── schedulers │ │ │ │ │ ├── lr_scheduler.h │ │ │ │ │ ├── reduce_on_plateau_scheduler.h │ │ │ │ │ └── step_lr.h │ │ │ │ ├── serialize.h │ │ │ │ └── sgd.h │ │ │ │ ├── ordered_dict.h │ │ │ │ ├── python.h │ │ │ │ ├── python │ │ │ │ └── init.h │ │ │ │ ├── serialize.h │ │ │ │ ├── serialize │ │ │ │ ├── archive.h │ │ │ │ ├── input-archive.h │ │ │ │ ├── output-archive.h │ │ │ │ └── tensor.h │ │ │ │ ├── sparse.h │ │ │ │ ├── special.h │ │ │ │ ├── torch.h │ │ │ │ ├── types.h │ │ │ │ ├── utils.h │ │ │ │ ├── version.h.in │ │ │ │ └── xpu.h │ │ └── src │ │ │ ├── cuda.cpp │ │ │ ├── data │ │ │ ├── datasets │ │ │ │ └── mnist.cpp │ │ │ └── samplers │ │ │ │ ├── distributed.cpp │ │ │ │ ├── random.cpp │ │ │ │ ├── sequential.cpp │ │ │ │ └── stream.cpp │ │ │ ├── enum.cpp │ │ │ ├── imethod.cpp │ │ │ ├── jit.cpp │ │ │ ├── mps.cpp │ │ │ ├── nn │ │ │ ├── init.cpp │ │ │ ├── module.cpp │ │ │ ├── modules │ │ │ │ ├── _functions.cpp │ │ │ │ ├── activation.cpp │ │ │ │ ├── adaptive.cpp │ │ │ │ ├── batchnorm.cpp │ │ │ │ ├── container │ │ │ │ │ └── functional.cpp │ │ │ │ ├── conv.cpp │ │ │ │ ├── distance.cpp │ │ │ │ ├── dropout.cpp │ │ │ │ ├── embedding.cpp │ │ │ │ ├── fold.cpp │ │ │ │ ├── instancenorm.cpp │ │ │ │ ├── linear.cpp │ │ │ │ ├── loss.cpp │ │ │ │ ├── normalization.cpp │ │ │ │ ├── padding.cpp │ │ │ │ ├── pixelshuffle.cpp │ │ │ │ ├── pooling.cpp │ │ │ │ ├── rnn.cpp │ │ │ │ ├── transformer.cpp │ │ │ │ └── upsampling.cpp │ │ │ └── options │ │ │ │ ├── activation.cpp │ │ │ │ ├── adaptive.cpp │ │ │ │ ├── batchnorm.cpp │ │ │ │ ├── conv.cpp │ │ │ │ ├── dropout.cpp │ │ │ │ ├── embedding.cpp │ │ │ │ ├── instancenorm.cpp │ │ │ │ ├── linear.cpp │ │ │ │ ├── normalization.cpp │ │ │ │ ├── padding.cpp │ │ │ │ ├── pooling.cpp │ │ │ │ ├── rnn.cpp │ │ │ │ ├── transformer.cpp │ │ │ │ └── vision.cpp │ │ │ ├── optim │ │ │ ├── adagrad.cpp │ │ │ ├── adam.cpp │ │ │ ├── adamw.cpp │ │ │ ├── lbfgs.cpp │ │ │ ├── optimizer.cpp │ │ │ ├── rmsprop.cpp │ │ │ ├── schedulers │ │ │ │ ├── lr_scheduler.cpp │ │ │ │ ├── reduce_on_plateau_scheduler.cpp │ │ │ │ └── step_lr.cpp │ │ │ ├── serialize.cpp │ │ │ └── sgd.cpp │ │ │ ├── python │ │ │ └── init.cpp │ │ │ ├── serialize.cpp │ │ │ ├── serialize │ │ │ ├── input-archive.cpp │ │ │ └── output-archive.cpp │ │ │ └── xpu.cpp │ ├── autograd │ │ ├── FunctionsManual.cpp │ │ ├── FunctionsManual.h │ │ ├── InferenceMode.h │ │ ├── README.md │ │ ├── TraceTypeManual.cpp │ │ ├── VariableTypeManual.cpp │ │ ├── VariableTypeUtils.h │ │ ├── anomaly_mode.cpp │ │ ├── anomaly_mode.h │ │ ├── autograd.cpp │ │ ├── autograd.h │ │ ├── autograd_meta.cpp │ │ ├── autograd_not_implemented_fallback.cpp │ │ ├── autograd_not_implemented_fallback.h │ │ ├── cpp_hook.cpp │ │ ├── cpp_hook.h │ │ ├── custom_function.cpp │ │ ├── custom_function.h │ │ ├── edge.h │ │ ├── engine.cpp │ │ ├── engine.h │ │ ├── forward_grad.cpp │ │ ├── forward_grad.h │ │ ├── function.cpp │ │ ├── function.h │ │ ├── function_hook.h │ │ ├── functions │ │ │ ├── accumulate_grad.cpp │ │ │ ├── accumulate_grad.h │ │ │ ├── basic_ops.cpp │ │ │ ├── basic_ops.h │ │ │ ├── comm.cpp │ │ │ ├── comm.h │ │ │ ├── init.cpp │ │ │ ├── pybind.h │ │ │ ├── tensor.cpp │ │ │ ├── tensor.h │ │ │ ├── utils.cpp │ │ │ └── utils.h │ │ ├── grad_mode.h │ │ ├── graph_task.h │ │ ├── init.cpp │ │ ├── input_buffer.cpp │ │ ├── input_buffer.h │ │ ├── input_metadata.cpp │ │ ├── input_metadata.h │ │ ├── jit_decomp_interface.cpp │ │ ├── jit_decomp_interface.h │ │ ├── profiler.h │ │ ├── profiler_kineto.cpp │ │ ├── profiler_kineto.h │ │ ├── profiler_legacy.cpp │ │ ├── profiler_legacy.h │ │ ├── profiler_python.cpp │ │ ├── profiler_python.h │ │ ├── python_anomaly_mode.cpp │ │ ├── python_anomaly_mode.h │ │ ├── python_autograd.h │ │ ├── python_cpp_function.cpp │ │ ├── python_cpp_function.h │ │ ├── python_engine.cpp │ │ ├── python_engine.h │ │ ├── python_enum_tag.h │ │ ├── python_fft_functions.h │ │ ├── python_function.cpp │ │ ├── python_function.h │ │ ├── python_hook.cpp │ │ ├── python_hook.h │ │ ├── python_legacy_variable.cpp │ │ ├── python_legacy_variable.h │ │ ├── python_linalg_functions.h │ │ ├── python_nested_functions.h │ │ ├── python_nested_functions_manual.cpp │ │ ├── python_nn_functions.h │ │ ├── python_saved_variable_hooks.cpp │ │ ├── python_saved_variable_hooks.h │ │ ├── python_sparse_functions.h │ │ ├── python_special_functions.h │ │ ├── python_torch_functions.h │ │ ├── python_torch_functions_manual.cpp │ │ ├── python_variable.cpp │ │ ├── python_variable.h │ │ ├── python_variable_indexing.cpp │ │ ├── python_variable_indexing.h │ │ ├── record_function_ops.cpp │ │ ├── record_function_ops.h │ │ ├── saved_variable.cpp │ │ ├── saved_variable.h │ │ ├── saved_variable_hooks.h │ │ ├── symbolic.h │ │ ├── utils │ │ │ ├── error_messages.h │ │ │ ├── grad_layout_contract.h │ │ │ ├── lambda_post_hook.h │ │ │ ├── python_arg_parsing.h │ │ │ ├── warnings.cpp │ │ │ ├── warnings.h │ │ │ └── wrap_outputs.h │ │ ├── variable.cpp │ │ ├── variable.h │ │ ├── variable_info.cpp │ │ └── variable_info.h │ ├── copy_utils.h │ ├── cpu │ │ ├── Module.cpp │ │ └── Module.h │ ├── cuda │ │ ├── CUDAPluggableAllocator.cpp │ │ ├── CUDAPluggableAllocator.h │ │ ├── Event.cpp │ │ ├── Event.h │ │ ├── GdsFile.cpp │ │ ├── GdsFile.h │ │ ├── Graph.cpp │ │ ├── MemPool.cpp │ │ ├── Module.cpp │ │ ├── Module.h │ │ ├── Stream.cpp │ │ ├── Stream.h │ │ ├── THCP.h │ │ ├── Tensor.cpp │ │ ├── comm.cpp │ │ ├── comm.h │ │ ├── device_set.h │ │ ├── memory_snapshot.cpp │ │ ├── memory_snapshot.h │ │ ├── nccl.cpp │ │ ├── nccl.h │ │ ├── python_comm.cpp │ │ ├── python_comm.h │ │ ├── python_nccl.cpp │ │ ├── python_nccl.h │ │ ├── shared │ │ │ ├── cudart.cpp │ │ │ ├── cudnn.cpp │ │ │ ├── cusparselt.cpp │ │ │ └── nvtx.cpp │ │ └── utils.cpp │ ├── deploy │ │ └── README.md │ ├── distributed │ │ ├── autograd │ │ │ ├── autograd.cpp │ │ │ ├── autograd.h │ │ │ ├── context │ │ │ │ ├── container.cpp │ │ │ │ ├── container.h │ │ │ │ ├── context.cpp │ │ │ │ └── context.h │ │ │ ├── engine │ │ │ │ ├── dist_engine.cpp │ │ │ │ └── dist_engine.h │ │ │ ├── functions │ │ │ │ ├── recvrpc_backward.cpp │ │ │ │ ├── recvrpc_backward.h │ │ │ │ ├── sendrpc_backward.cpp │ │ │ │ └── sendrpc_backward.h │ │ │ ├── init.cpp │ │ │ ├── python_autograd.h │ │ │ ├── rpc_messages │ │ │ │ ├── autograd_metadata.cpp │ │ │ │ ├── autograd_metadata.h │ │ │ │ ├── cleanup_autograd_context_req.cpp │ │ │ │ ├── cleanup_autograd_context_req.h │ │ │ │ ├── cleanup_autograd_context_resp.cpp │ │ │ │ ├── cleanup_autograd_context_resp.h │ │ │ │ ├── propagate_gradients_req.cpp │ │ │ │ ├── propagate_gradients_req.h │ │ │ │ ├── propagate_gradients_resp.cpp │ │ │ │ ├── propagate_gradients_resp.h │ │ │ │ ├── rpc_with_autograd.cpp │ │ │ │ ├── rpc_with_autograd.h │ │ │ │ ├── rpc_with_profiling_req.cpp │ │ │ │ ├── rpc_with_profiling_req.h │ │ │ │ ├── rpc_with_profiling_resp.cpp │ │ │ │ ├── rpc_with_profiling_resp.h │ │ │ │ ├── rref_backward_req.cpp │ │ │ │ ├── rref_backward_req.h │ │ │ │ ├── rref_backward_resp.cpp │ │ │ │ └── rref_backward_resp.h │ │ │ ├── utils.cpp │ │ │ └── utils.h │ │ ├── c10d │ │ │ ├── Backend.cpp │ │ │ ├── Backend.hpp │ │ │ ├── Backoff.cpp │ │ │ ├── Backoff.hpp │ │ │ ├── CUDASymmetricMemory-inl.h │ │ │ ├── CUDASymmetricMemory.cu │ │ │ ├── CUDASymmetricMemory.hpp │ │ │ ├── CUDASymmetricMemoryOps.cu │ │ │ ├── CudaDMAConnectivity.cpp │ │ │ ├── DMAConnectivity.cpp │ │ │ ├── DMAConnectivity.hpp │ │ │ ├── FakeProcessGroup.hpp │ │ │ ├── FileStore.cpp │ │ │ ├── FileStore.hpp │ │ │ ├── FlightRecorder.cpp │ │ │ ├── FlightRecorder.hpp │ │ │ ├── Functional.cpp │ │ │ ├── Functional.hpp │ │ │ ├── GlooDeviceFactory.cpp │ │ │ ├── GlooDeviceFactory.hpp │ │ │ ├── GroupRegistry.cpp │ │ │ ├── GroupRegistry.hpp │ │ │ ├── HashStore.cpp │ │ │ ├── HashStore.hpp │ │ │ ├── NCCLUtils.cpp │ │ │ ├── NCCLUtils.hpp │ │ │ ├── NanCheck.cu │ │ │ ├── NanCheck.hpp │ │ │ ├── Ops.cpp │ │ │ ├── ParamCommsUtils.cpp │ │ │ ├── ParamCommsUtils.hpp │ │ │ ├── PrefixStore.cpp │ │ │ ├── PrefixStore.hpp │ │ │ ├── ProcessGroup.cpp │ │ │ ├── ProcessGroup.hpp │ │ │ ├── ProcessGroupGloo.cpp │ │ │ ├── ProcessGroupGloo.hpp │ │ │ ├── ProcessGroupMPI.cpp │ │ │ ├── ProcessGroupMPI.hpp │ │ │ ├── ProcessGroupNCCL.cpp │ │ │ ├── ProcessGroupNCCL.hpp │ │ │ ├── ProcessGroupUCC.cpp │ │ │ ├── ProcessGroupUCC.hpp │ │ │ ├── ProcessGroupWrapper.cpp │ │ │ ├── ProcessGroupWrapper.hpp │ │ │ ├── PyProcessGroup.hpp │ │ │ ├── RankLocal.hpp │ │ │ ├── Store.cpp │ │ │ ├── Store.hpp │ │ │ ├── SymmetricMemory.cpp │ │ │ ├── SymmetricMemory.hpp │ │ │ ├── TCPStore.cpp │ │ │ ├── TCPStore.hpp │ │ │ ├── TCPStoreBackend.cpp │ │ │ ├── TCPStoreBackend.hpp │ │ │ ├── TCPStoreLibUvBackend.cpp │ │ │ ├── TraceUtils.h │ │ │ ├── Types.hpp │ │ │ ├── UCCTracing.cpp │ │ │ ├── UCCTracing.hpp │ │ │ ├── UCCUtils.cpp │ │ │ ├── UCCUtils.hpp │ │ │ ├── UnixSockUtils.hpp │ │ │ ├── Utils.cpp │ │ │ ├── Utils.hpp │ │ │ ├── WinSockUtils.hpp │ │ │ ├── Work.cpp │ │ │ ├── Work.hpp │ │ │ ├── c10d.h │ │ │ ├── comm.cpp │ │ │ ├── comm.hpp │ │ │ ├── control_collectives │ │ │ │ ├── ControlCollectives.hpp │ │ │ │ ├── StoreCollectives.cpp │ │ │ │ └── StoreCollectives.hpp │ │ │ ├── control_plane │ │ │ │ ├── Handlers.cpp │ │ │ │ ├── Handlers.hpp │ │ │ │ ├── PythonHandlers.cpp │ │ │ │ ├── WorkerServer.cpp │ │ │ │ └── WorkerServer.hpp │ │ │ ├── cuda │ │ │ │ ├── AsyncMM.cu │ │ │ │ ├── AsyncMM.cuh │ │ │ │ ├── cutlass │ │ │ │ │ └── gemm │ │ │ │ │ │ └── kernel │ │ │ │ │ │ └── persistent_async_input_scheduler.cuh │ │ │ │ ├── utils.cpp │ │ │ │ └── utils.hpp │ │ │ ├── debug.cpp │ │ │ ├── debug.h │ │ │ ├── default_comm_hooks.cpp │ │ │ ├── default_comm_hooks.hpp │ │ │ ├── error.h │ │ │ ├── exception.h │ │ │ ├── init.cpp │ │ │ ├── intra_node_comm.cpp │ │ │ ├── intra_node_comm.cu │ │ │ ├── intra_node_comm.hpp │ │ │ ├── logger.cpp │ │ │ ├── logger.hpp │ │ │ ├── logging.cpp │ │ │ ├── logging.h │ │ │ ├── python_comm_hook.cpp │ │ │ ├── python_comm_hook.h │ │ │ ├── quantization │ │ │ │ ├── quantization.cpp │ │ │ │ ├── quantization.h │ │ │ │ ├── quantization_gpu.cu │ │ │ │ ├── quantization_gpu.h │ │ │ │ └── quantization_utils.h │ │ │ ├── reducer.cpp │ │ │ ├── reducer.hpp │ │ │ ├── reducer_cuda.cpp │ │ │ ├── reducer_timer.hpp │ │ │ ├── sequence_num.cpp │ │ │ ├── sequence_num.hpp │ │ │ ├── socket.cpp │ │ │ ├── socket.h │ │ │ └── socket_fmt.h │ │ └── rpc │ │ │ ├── agent_utils.cpp │ │ │ ├── agent_utils.h │ │ │ ├── init.cpp │ │ │ ├── message.cpp │ │ │ ├── message.h │ │ │ ├── metrics │ │ │ ├── RpcMetricsHandler.h │ │ │ └── registry.cpp │ │ │ ├── profiler │ │ │ ├── remote_profiler_manager.cpp │ │ │ ├── remote_profiler_manager.h │ │ │ ├── server_process_global_profiler.cpp │ │ │ └── server_process_global_profiler.h │ │ │ ├── py_rref.cpp │ │ │ ├── py_rref.h │ │ │ ├── python_call.cpp │ │ │ ├── python_call.h │ │ │ ├── python_functions.cpp │ │ │ ├── python_functions.h │ │ │ ├── python_remote_call.cpp │ │ │ ├── python_remote_call.h │ │ │ ├── python_resp.cpp │ │ │ ├── python_resp.h │ │ │ ├── python_rpc_handler.cpp │ │ │ ├── python_rpc_handler.h │ │ │ ├── request_callback.cpp │ │ │ ├── request_callback.h │ │ │ ├── request_callback_impl.cpp │ │ │ ├── request_callback_impl.h │ │ │ ├── request_callback_no_python.cpp │ │ │ ├── request_callback_no_python.h │ │ │ ├── rpc.h │ │ │ ├── rpc_agent.cpp │ │ │ ├── rpc_agent.h │ │ │ ├── rpc_command_base.h │ │ │ ├── rref_context.cpp │ │ │ ├── rref_context.h │ │ │ ├── rref_impl.cpp │ │ │ ├── rref_impl.h │ │ │ ├── rref_proto.cpp │ │ │ ├── rref_proto.h │ │ │ ├── script_call.cpp │ │ │ ├── script_call.h │ │ │ ├── script_remote_call.cpp │ │ │ ├── script_remote_call.h │ │ │ ├── script_resp.cpp │ │ │ ├── script_resp.h │ │ │ ├── tensorpipe_agent.cpp │ │ │ ├── tensorpipe_agent.h │ │ │ ├── tensorpipe_cuda.cpp │ │ │ ├── tensorpipe_utils.cpp │ │ │ ├── tensorpipe_utils.h │ │ │ ├── testing │ │ │ ├── faulty_tensorpipe_agent.cpp │ │ │ ├── faulty_tensorpipe_agent.h │ │ │ ├── init.cpp │ │ │ └── testing.h │ │ │ ├── torchscript_functions.cpp │ │ │ ├── torchscript_functions.h │ │ │ ├── types.cpp │ │ │ ├── types.h │ │ │ ├── unpickled_python_call.cpp │ │ │ ├── unpickled_python_call.h │ │ │ ├── unpickled_python_remote_call.cpp │ │ │ ├── unpickled_python_remote_call.h │ │ │ ├── utils.cpp │ │ │ └── utils.h │ ├── dynamo │ │ ├── cache_entry.cpp │ │ ├── cache_entry.h │ │ ├── compiled_autograd.cpp │ │ ├── compiled_autograd.h │ │ ├── cpp_shim.cpp │ │ ├── cpp_shim.h │ │ ├── cpython_defs.c │ │ ├── cpython_defs.h │ │ ├── cpython_includes.h │ │ ├── debug_macros.h │ │ ├── eval_frame.c │ │ ├── eval_frame.h │ │ ├── eval_frame_cpp.cpp │ │ ├── eval_frame_cpp.h │ │ ├── extra_state.cpp │ │ ├── extra_state.h │ │ ├── framelocals_mapping.cpp │ │ ├── framelocals_mapping.h │ │ ├── guards.cpp │ │ ├── guards.h │ │ ├── init.cpp │ │ ├── init.h │ │ ├── python_compiled_autograd.cpp │ │ ├── python_compiled_autograd.h │ │ ├── utils.cpp │ │ └── utils.h │ ├── empty.c │ ├── export │ │ ├── pybind.cpp │ │ └── pybind.h │ ├── functorch │ │ ├── init.cpp │ │ └── init.h │ ├── fx │ │ ├── node.cpp │ │ └── node.h │ ├── inductor │ │ ├── aoti_eager │ │ │ ├── kernel_holder.cpp │ │ │ ├── kernel_holder.h │ │ │ ├── kernel_meta_info.cpp │ │ │ └── kernel_meta_info.h │ │ ├── aoti_include │ │ │ ├── array_ref.h │ │ │ ├── common.h │ │ │ ├── cpu.h │ │ │ ├── cuda.h │ │ │ └── xpu.h │ │ ├── aoti_package │ │ │ ├── model_package_loader.cpp │ │ │ ├── model_package_loader.h │ │ │ ├── pybind.cpp │ │ │ └── pybind.h │ │ ├── aoti_runner │ │ │ ├── model_container_runner.cpp │ │ │ ├── model_container_runner.h │ │ │ ├── model_container_runner_cpu.cpp │ │ │ ├── model_container_runner_cpu.h │ │ │ ├── model_container_runner_cuda.cpp │ │ │ ├── model_container_runner_cuda.h │ │ │ ├── model_container_runner_xpu.cpp │ │ │ ├── model_container_runner_xpu.h │ │ │ ├── pybind.cpp │ │ │ └── pybind.h │ │ ├── aoti_runtime │ │ │ ├── arrayref_tensor.h │ │ │ ├── device_utils.h │ │ │ ├── interface.h │ │ │ ├── model.h │ │ │ ├── model_container.h │ │ │ ├── scalar_to_tensor.h │ │ │ ├── sycl_runtime_wrappers.h │ │ │ ├── thread_local.h │ │ │ ├── utils.h │ │ │ ├── utils_cuda.h │ │ │ └── utils_xpu.h │ │ ├── aoti_torch │ │ │ ├── c │ │ │ │ ├── shim.h │ │ │ │ ├── shim_mkldnn.h │ │ │ │ └── shim_xpu.h │ │ │ ├── generated │ │ │ │ ├── c_shim_cpu.h │ │ │ │ ├── c_shim_cuda.h │ │ │ │ └── c_shim_xpu.h │ │ │ ├── mkldnn_tensor.cpp │ │ │ ├── mkldnn_tensor.h │ │ │ ├── oss_proxy_executor.cpp │ │ │ ├── oss_proxy_executor.h │ │ │ ├── proxy_executor.h │ │ │ ├── shim_common.cpp │ │ │ ├── shim_cuda.cpp │ │ │ ├── shim_mkldnn.cpp │ │ │ ├── shim_xpu.cpp │ │ │ ├── tensor_converter.cpp │ │ │ ├── tensor_converter.h │ │ │ └── utils.h │ │ ├── array_ref_impl.h │ │ ├── cpp_wrapper │ │ │ ├── array_ref.h │ │ │ ├── common.h │ │ │ ├── cpu.h │ │ │ ├── cuda.h │ │ │ ├── device_internal │ │ │ │ ├── cpu.h │ │ │ │ ├── cuda.h │ │ │ │ └── xpu.h │ │ │ └── xpu.h │ │ ├── inductor_ops.cpp │ │ ├── inductor_ops.h │ │ └── resize_storage_bytes.cpp │ ├── instruction_counter │ │ ├── Module.cpp │ │ └── Module.h │ ├── itt.cpp │ ├── itt_wrapper.cpp │ ├── itt_wrapper.h │ ├── jit │ │ ├── JIT-AUTOCAST.md │ │ ├── OVERVIEW.md │ │ ├── README.md │ │ ├── api │ │ │ ├── compilation_unit.h │ │ │ ├── function_impl.cpp │ │ │ ├── function_impl.h │ │ │ ├── method.h │ │ │ ├── module.cpp │ │ │ ├── module.h │ │ │ ├── module_save.cpp │ │ │ ├── object.cpp │ │ │ └── object.h │ │ ├── backends │ │ │ ├── backend.h │ │ │ ├── backend_debug_handler.cpp │ │ │ ├── backend_debug_handler.h │ │ │ ├── backend_debug_info.cpp │ │ │ ├── backend_debug_info.h │ │ │ ├── backend_detail.cpp │ │ │ ├── backend_detail.h │ │ │ ├── backend_exception.h │ │ │ ├── backend_init.cpp │ │ │ ├── backend_init.h │ │ │ ├── backend_interface.cpp │ │ │ ├── backend_interface.h │ │ │ ├── backend_preprocess.h │ │ │ ├── backend_resolver.cpp │ │ │ ├── backend_resolver.h │ │ │ ├── coreml │ │ │ │ ├── cpp │ │ │ │ │ ├── backend.cpp │ │ │ │ │ ├── context.cpp │ │ │ │ │ ├── context.h │ │ │ │ │ └── preprocess.cpp │ │ │ │ └── objc │ │ │ │ │ ├── PTMCoreMLBackend.mm │ │ │ │ │ ├── PTMCoreMLCompiler.h │ │ │ │ │ ├── PTMCoreMLCompiler.mm │ │ │ │ │ ├── PTMCoreMLExecutor.h │ │ │ │ │ ├── PTMCoreMLExecutor.mm │ │ │ │ │ ├── PTMCoreMLFeatureProvider.h │ │ │ │ │ ├── PTMCoreMLFeatureProvider.mm │ │ │ │ │ ├── PTMCoreMLModelWrapper.h │ │ │ │ │ └── PTMCoreMLTensorSpec.h │ │ │ ├── nnapi │ │ │ │ ├── nnapi_backend_lib.cpp │ │ │ │ └── nnapi_backend_preprocess.cpp │ │ │ └── xnnpack │ │ │ │ ├── compiler │ │ │ │ ├── xnn_compiler.cpp │ │ │ │ └── xnn_compiler.h │ │ │ │ ├── executor │ │ │ │ └── xnn_executor.h │ │ │ │ ├── serialization │ │ │ │ ├── schema.fbs │ │ │ │ ├── serializer.cpp │ │ │ │ └── serializer.h │ │ │ │ ├── xnnpack_backend_lib.cpp │ │ │ │ ├── xnnpack_backend_preprocess.cpp │ │ │ │ ├── xnnpack_graph_builder.cpp │ │ │ │ └── xnnpack_graph_builder.h │ │ ├── codegen │ │ │ ├── cuda │ │ │ │ ├── README.md │ │ │ │ ├── interface.cpp │ │ │ │ └── interface.h │ │ │ ├── fuser │ │ │ │ ├── README.md │ │ │ │ ├── arg_spec.h │ │ │ │ ├── codegen.cpp │ │ │ │ ├── codegen.h │ │ │ │ ├── compiler.cpp │ │ │ │ ├── compiler.h │ │ │ │ ├── cpu │ │ │ │ │ ├── fused_kernel.cpp │ │ │ │ │ ├── fused_kernel.h │ │ │ │ │ ├── resource_strings.h │ │ │ │ │ └── temp_file.h │ │ │ │ ├── cuda │ │ │ │ │ ├── fused_kernel.cpp │ │ │ │ │ ├── fused_kernel.h │ │ │ │ │ └── resource_strings.h │ │ │ │ ├── executor.cpp │ │ │ │ ├── executor.h │ │ │ │ ├── fallback.cpp │ │ │ │ ├── fallback.h │ │ │ │ ├── fused_kernel.h │ │ │ │ ├── interface.cpp │ │ │ │ ├── interface.h │ │ │ │ ├── kernel_cache.cpp │ │ │ │ ├── kernel_cache.h │ │ │ │ ├── kernel_spec.h │ │ │ │ ├── partition_desc.h │ │ │ │ ├── tensor_desc.h │ │ │ │ └── tensor_info.h │ │ │ └── onednn │ │ │ │ ├── LlgaTensorImpl.cpp │ │ │ │ ├── LlgaTensorImpl.h │ │ │ │ ├── README.md │ │ │ │ ├── decompose_silu.cpp │ │ │ │ ├── decompose_silu.h │ │ │ │ ├── defer_size_check.cpp │ │ │ │ ├── defer_size_check.h │ │ │ │ ├── graph_fuser.cpp │ │ │ │ ├── graph_fuser.h │ │ │ │ ├── graph_helper.cpp │ │ │ │ ├── graph_helper.h │ │ │ │ ├── graph_rewriter.cpp │ │ │ │ ├── guard_shape.cpp │ │ │ │ ├── guard_shape.h │ │ │ │ ├── interface.cpp │ │ │ │ ├── interface.h │ │ │ │ ├── kernel.cpp │ │ │ │ ├── kernel.h │ │ │ │ ├── layout_propagation.cpp │ │ │ │ ├── layout_propagation.h │ │ │ │ ├── operator.h │ │ │ │ ├── prepare_binary.cpp │ │ │ │ ├── prepare_binary.h │ │ │ │ └── register_interface.cpp │ │ ├── cuda │ │ │ └── cuda.h │ │ ├── docs │ │ │ └── serialization.md │ │ ├── frontend │ │ │ ├── builtin_functions.cpp │ │ │ ├── builtin_functions.h │ │ │ ├── canonicalize_modified_loop.cpp │ │ │ ├── canonicalize_modified_loop.h │ │ │ ├── concrete_module_type.cpp │ │ │ ├── concrete_module_type.h │ │ │ ├── convert_to_ssa.cpp │ │ │ ├── convert_to_ssa.h │ │ │ ├── edit_distance.cpp │ │ │ ├── edit_distance.h │ │ │ ├── error_report.cpp │ │ │ ├── error_report.h │ │ │ ├── exit_transforms.cpp │ │ │ ├── exit_transforms.h │ │ │ ├── function_schema_parser.cpp │ │ │ ├── function_schema_parser.h │ │ │ ├── inline_loop_condition.cpp │ │ │ ├── inline_loop_condition.h │ │ │ ├── ir_emitter.cpp │ │ │ ├── ir_emitter.h │ │ │ ├── lexer.cpp │ │ │ ├── lexer.h │ │ │ ├── mini_environment.h │ │ │ ├── name_mangler.cpp │ │ │ ├── name_mangler.h │ │ │ ├── parse_string_literal.h │ │ │ ├── parser.cpp │ │ │ ├── parser.h │ │ │ ├── parser_constants.h │ │ │ ├── resolver.h │ │ │ ├── schema_matching.cpp │ │ │ ├── schema_matching.h │ │ │ ├── schema_type_parser.cpp │ │ │ ├── schema_type_parser.h │ │ │ ├── script_type_parser.cpp │ │ │ ├── script_type_parser.h │ │ │ ├── source_range.cpp │ │ │ ├── source_range.h │ │ │ ├── source_ref.h │ │ │ ├── strtod.cpp │ │ │ ├── strtod.h │ │ │ ├── sugared_value.cpp │ │ │ ├── sugared_value.h │ │ │ ├── tracer.cpp │ │ │ ├── tracer.h │ │ │ ├── tree.h │ │ │ ├── tree_views.cpp │ │ │ ├── tree_views.h │ │ │ ├── versioned_symbols.cpp │ │ │ └── versioned_symbols.h │ │ ├── ir │ │ │ ├── alias_analysis.cpp │ │ │ ├── alias_analysis.h │ │ │ ├── attributes.cpp │ │ │ ├── attributes.h │ │ │ ├── constants.cpp │ │ │ ├── constants.h │ │ │ ├── graph_node_list.h │ │ │ ├── graph_utils.cpp │ │ │ ├── graph_utils.h │ │ │ ├── ir.cpp │ │ │ ├── ir.h │ │ │ ├── ir_views.h │ │ │ ├── irparser.cpp │ │ │ ├── irparser.h │ │ │ ├── named_value.h │ │ │ ├── node_hashing.cpp │ │ │ ├── node_hashing.h │ │ │ ├── scope.cpp │ │ │ ├── scope.h │ │ │ ├── subgraph_matcher.cpp │ │ │ ├── subgraph_matcher.h │ │ │ ├── type_hashing.cpp │ │ │ └── type_hashing.h │ │ ├── jit_log.cpp │ │ ├── jit_log.h │ │ ├── jit_opt_limit.cpp │ │ ├── jit_opt_limit.h │ │ ├── mobile │ │ │ ├── code.h │ │ │ ├── compatibility │ │ │ │ ├── backport.cpp │ │ │ │ ├── backport.h │ │ │ │ ├── backport_manager.cpp │ │ │ │ ├── backport_manager.h │ │ │ │ ├── model_compatibility.cpp │ │ │ │ ├── model_compatibility.h │ │ │ │ ├── runtime_compatibility.cpp │ │ │ │ └── runtime_compatibility.h │ │ │ ├── debug_info.cpp │ │ │ ├── debug_info.h │ │ │ ├── file_format.h │ │ │ ├── flatbuffer_loader.cpp │ │ │ ├── flatbuffer_loader.h │ │ │ ├── frame.h │ │ │ ├── function.cpp │ │ │ ├── function.h │ │ │ ├── import.cpp │ │ │ ├── import.h │ │ │ ├── import_data.cpp │ │ │ ├── import_data.h │ │ │ ├── import_export_common.h │ │ │ ├── interpreter.cpp │ │ │ ├── interpreter.h │ │ │ ├── method.h │ │ │ ├── model_tracer │ │ │ │ ├── BuildFeatureTracer.cpp │ │ │ │ ├── BuildFeatureTracer.h │ │ │ │ ├── CMakeLists.txt │ │ │ │ ├── CustomClassTracer.cpp │ │ │ │ ├── CustomClassTracer.h │ │ │ │ ├── KernelDTypeTracer.cpp │ │ │ │ ├── KernelDTypeTracer.h │ │ │ │ ├── MobileModelRunner.cpp │ │ │ │ ├── MobileModelRunner.h │ │ │ │ ├── OperatorCallTracer.cpp │ │ │ │ ├── OperatorCallTracer.h │ │ │ │ ├── TensorUtils.cpp │ │ │ │ ├── TensorUtils.h │ │ │ │ ├── TracerRunner.cpp │ │ │ │ ├── TracerRunner.h │ │ │ │ └── tracer.cpp │ │ │ ├── module.cpp │ │ │ ├── module.h │ │ │ ├── nnc │ │ │ │ ├── aot_compiler.cpp │ │ │ │ ├── aot_compiler.h │ │ │ │ ├── backend.cpp │ │ │ │ ├── context.cpp │ │ │ │ ├── context.h │ │ │ │ ├── registry.cpp │ │ │ │ └── registry.h │ │ │ ├── observer.cpp │ │ │ ├── observer.h │ │ │ ├── parse_bytecode.cpp │ │ │ ├── parse_bytecode.h │ │ │ ├── parse_operators.cpp │ │ │ ├── parse_operators.h │ │ │ ├── prim_ops_registery.cpp │ │ │ ├── prim_ops_registery.h │ │ │ ├── profiler_edge.cpp │ │ │ ├── profiler_edge.h │ │ │ ├── promoted_prim_ops.cpp │ │ │ ├── promoted_prim_ops.h │ │ │ ├── quantization.cpp │ │ │ ├── quantization.h │ │ │ ├── register_ops_common_utils.cpp │ │ │ ├── register_ops_common_utils.h │ │ │ ├── train │ │ │ │ ├── export_data.cpp │ │ │ │ ├── export_data.h │ │ │ │ ├── optim │ │ │ │ │ ├── sgd.cpp │ │ │ │ │ └── sgd.h │ │ │ │ ├── random.cpp │ │ │ │ ├── random.h │ │ │ │ ├── sequential.cpp │ │ │ │ └── sequential.h │ │ │ ├── type_parser.cpp │ │ │ ├── type_parser.h │ │ │ ├── upgrader_mobile.cpp │ │ │ └── upgrader_mobile.h │ │ ├── operator_upgraders │ │ │ ├── README.md │ │ │ ├── upgraders.cpp │ │ │ ├── upgraders.h │ │ │ ├── upgraders_entry.cpp │ │ │ ├── upgraders_entry.h │ │ │ ├── utils.cpp │ │ │ ├── utils.h │ │ │ ├── version_map.cpp │ │ │ └── version_map.h │ │ ├── passes │ │ │ ├── add_if_then_else.cpp │ │ │ ├── add_if_then_else.h │ │ │ ├── annotate_warns.cpp │ │ │ ├── annotate_warns.h │ │ │ ├── autocast.cpp │ │ │ ├── autocast.h │ │ │ ├── bailout_graph.cpp │ │ │ ├── bailout_graph.h │ │ │ ├── batch_mm.cpp │ │ │ ├── batch_mm.h │ │ │ ├── canonicalize.cpp │ │ │ ├── canonicalize.h │ │ │ ├── canonicalize_graph_fuser_ops.cpp │ │ │ ├── canonicalize_graph_fuser_ops.h │ │ │ ├── check_strict_fusion.cpp │ │ │ ├── check_strict_fusion.h │ │ │ ├── clear_profiling.cpp │ │ │ ├── clear_profiling.h │ │ │ ├── clear_undefinedness.cpp │ │ │ ├── clear_undefinedness.h │ │ │ ├── common_subexpression_elimination.cpp │ │ │ ├── common_subexpression_elimination.h │ │ │ ├── concat_opt.cpp │ │ │ ├── concat_opt.h │ │ │ ├── constant_pooling.cpp │ │ │ ├── constant_pooling.h │ │ │ ├── constant_propagation.cpp │ │ │ ├── constant_propagation.h │ │ │ ├── create_autodiff_subgraphs.cpp │ │ │ ├── create_autodiff_subgraphs.h │ │ │ ├── create_functional_graphs.cpp │ │ │ ├── create_functional_graphs.h │ │ │ ├── dbr_quantization │ │ │ │ ├── remove_redundant_aliases.cpp │ │ │ │ └── remove_redundant_aliases.h │ │ │ ├── dead_code_elimination.cpp │ │ │ ├── dead_code_elimination.h │ │ │ ├── decompose_ops.cpp │ │ │ ├── decompose_ops.h │ │ │ ├── device_type_analysis.cpp │ │ │ ├── device_type_analysis.h │ │ │ ├── dtype_analysis.cpp │ │ │ ├── dtype_analysis.h │ │ │ ├── eliminate_no_ops.cpp │ │ │ ├── eliminate_no_ops.h │ │ │ ├── erase_number_types.cpp │ │ │ ├── erase_number_types.h │ │ │ ├── fixup_trace_scope_blocks.cpp │ │ │ ├── fixup_trace_scope_blocks.h │ │ │ ├── fold_conv_bn.cpp │ │ │ ├── fold_conv_bn.h │ │ │ ├── fold_linear_bn.cpp │ │ │ ├── fold_linear_bn.h │ │ │ ├── freeze_module.cpp │ │ │ ├── freeze_module.h │ │ │ ├── frozen_concat_linear.cpp │ │ │ ├── frozen_concat_linear.h │ │ │ ├── frozen_conv_add_relu_fusion.cpp │ │ │ ├── frozen_conv_add_relu_fusion.h │ │ │ ├── frozen_conv_add_relu_fusion_cuda.cpp │ │ │ ├── frozen_conv_folding.cpp │ │ │ ├── frozen_conv_folding.h │ │ │ ├── frozen_graph_optimizations.cpp │ │ │ ├── frozen_graph_optimizations.h │ │ │ ├── frozen_linear_folding.cpp │ │ │ ├── frozen_linear_folding.h │ │ │ ├── frozen_linear_transpose.cpp │ │ │ ├── frozen_linear_transpose.h │ │ │ ├── frozen_ops_to_mkldnn.cpp │ │ │ ├── frozen_ops_to_mkldnn.h │ │ │ ├── fuse_linear.cpp │ │ │ ├── fuse_linear.h │ │ │ ├── fuse_relu.cpp │ │ │ ├── fuse_relu.h │ │ │ ├── graph_fuser.cpp │ │ │ ├── graph_fuser.h │ │ │ ├── graph_rewrite_helper.cpp │ │ │ ├── graph_rewrite_helper.h │ │ │ ├── guard_elimination.cpp │ │ │ ├── guard_elimination.h │ │ │ ├── hoist_conv_packed_params.cpp │ │ │ ├── hoist_conv_packed_params.h │ │ │ ├── inline_autodiff_subgraphs.cpp │ │ │ ├── inline_autodiff_subgraphs.h │ │ │ ├── inline_fork_wait.cpp │ │ │ ├── inline_fork_wait.h │ │ │ ├── inline_forked_closures.cpp │ │ │ ├── inline_forked_closures.h │ │ │ ├── inliner.cpp │ │ │ ├── inliner.h │ │ │ ├── inplace_check.cpp │ │ │ ├── inplace_check.h │ │ │ ├── insert_guards.cpp │ │ │ ├── insert_guards.h │ │ │ ├── integer_value_refinement.cpp │ │ │ ├── integer_value_refinement.h │ │ │ ├── lift_closures.cpp │ │ │ ├── lift_closures.h │ │ │ ├── liveness.cpp │ │ │ ├── liveness.h │ │ │ ├── loop_unrolling.cpp │ │ │ ├── loop_unrolling.h │ │ │ ├── lower_grad_of.cpp │ │ │ ├── lower_grad_of.h │ │ │ ├── lower_graph.cpp │ │ │ ├── lower_graph.h │ │ │ ├── lower_tuples.cpp │ │ │ ├── lower_tuples.h │ │ │ ├── metal_rewrite.cpp │ │ │ ├── metal_rewrite.h │ │ │ ├── mkldnn_rewrite.cpp │ │ │ ├── mkldnn_rewrite.h │ │ │ ├── mobile_optimizer_type.h │ │ │ ├── normalize_ops.cpp │ │ │ ├── normalize_ops.h │ │ │ ├── onednn_graph_fuser.h │ │ │ ├── onnx.cpp │ │ │ ├── onnx.h │ │ │ ├── onnx │ │ │ │ ├── README.md │ │ │ │ ├── cast_all_constant_to_floating.cpp │ │ │ │ ├── cast_all_constant_to_floating.h │ │ │ │ ├── constant_fold.cpp │ │ │ │ ├── constant_fold.h │ │ │ │ ├── constant_map.cpp │ │ │ │ ├── constant_map.h │ │ │ │ ├── deduplicate_initializers.cpp │ │ │ │ ├── deduplicate_initializers.h │ │ │ │ ├── eliminate_unused_items.cpp │ │ │ │ ├── eliminate_unused_items.h │ │ │ │ ├── eval_peephole.cpp │ │ │ │ ├── eval_peephole.h │ │ │ │ ├── fixup_onnx_controlflow.cpp │ │ │ │ ├── fixup_onnx_controlflow.h │ │ │ │ ├── function_extraction.cpp │ │ │ │ ├── function_extraction.h │ │ │ │ ├── function_substitution.cpp │ │ │ │ ├── function_substitution.h │ │ │ │ ├── helper.cpp │ │ │ │ ├── helper.h │ │ │ │ ├── list_model_parameters.cpp │ │ │ │ ├── list_model_parameters.h │ │ │ │ ├── naming.cpp │ │ │ │ ├── naming.h │ │ │ │ ├── onnx_log.cpp │ │ │ │ ├── onnx_log.h │ │ │ │ ├── pattern_conversion │ │ │ │ │ ├── autograd_function_process.cpp │ │ │ │ │ ├── autograd_function_process.h │ │ │ │ │ ├── common.cpp │ │ │ │ │ ├── common.h │ │ │ │ │ ├── pattern_conversion.cpp │ │ │ │ │ ├── pattern_conversion.h │ │ │ │ │ ├── pattern_encapsulation.cpp │ │ │ │ │ └── pattern_encapsulation.h │ │ │ │ ├── peephole.cpp │ │ │ │ ├── peephole.h │ │ │ │ ├── prepare_division_for_onnx.cpp │ │ │ │ ├── prepare_division_for_onnx.h │ │ │ │ ├── preprocess_for_onnx.cpp │ │ │ │ ├── preprocess_for_onnx.h │ │ │ │ ├── remove_inplace_ops_for_onnx.cpp │ │ │ │ ├── remove_inplace_ops_for_onnx.h │ │ │ │ ├── scalar_type_analysis.cpp │ │ │ │ ├── scalar_type_analysis.h │ │ │ │ ├── shape_type_inference.cpp │ │ │ │ ├── shape_type_inference.h │ │ │ │ ├── unpack_quantized_weights.cpp │ │ │ │ └── unpack_quantized_weights.h │ │ │ ├── pass_manager.cpp │ │ │ ├── pass_manager.h │ │ │ ├── peephole.cpp │ │ │ ├── peephole.h │ │ │ ├── peephole_alias_sensitive.cpp │ │ │ ├── peephole_alias_sensitive.h │ │ │ ├── peephole_dict_idioms.cpp │ │ │ ├── peephole_dict_idioms.h │ │ │ ├── peephole_list_idioms.cpp │ │ │ ├── peephole_list_idioms.h │ │ │ ├── peephole_non_tensor.cpp │ │ │ ├── peephole_non_tensor.h │ │ │ ├── prepack_folding.cpp │ │ │ ├── prepack_folding.h │ │ │ ├── quantization │ │ │ │ ├── dedup_module_uses.cpp │ │ │ │ ├── dedup_module_uses.h │ │ │ │ ├── finalize.cpp │ │ │ │ ├── finalize.h │ │ │ │ ├── fusion_passes.cpp │ │ │ │ ├── fusion_passes.h │ │ │ │ ├── helper.cpp │ │ │ │ ├── helper.h │ │ │ │ ├── insert_observers.cpp │ │ │ │ ├── insert_observers.h │ │ │ │ ├── insert_quant_dequant.cpp │ │ │ │ ├── insert_quant_dequant.h │ │ │ │ ├── quantization_patterns.h │ │ │ │ ├── quantization_type.cpp │ │ │ │ ├── quantization_type.h │ │ │ │ ├── register_packed_params.cpp │ │ │ │ └── register_packed_params.h │ │ │ ├── refine_tuple_types.cpp │ │ │ ├── refine_tuple_types.h │ │ │ ├── remove_dropout.cpp │ │ │ ├── remove_dropout.h │ │ │ ├── remove_exceptions.cpp │ │ │ ├── remove_exceptions.h │ │ │ ├── remove_expands.cpp │ │ │ ├── remove_expands.h │ │ │ ├── remove_inplace_ops.cpp │ │ │ ├── remove_inplace_ops.h │ │ │ ├── remove_mutation.cpp │ │ │ ├── remove_mutation.h │ │ │ ├── remove_redundant_profiles.cpp │ │ │ ├── remove_redundant_profiles.h │ │ │ ├── replacement_of_old_operators.cpp │ │ │ ├── replacement_of_old_operators.h │ │ │ ├── requires_grad_analysis.cpp │ │ │ ├── requires_grad_analysis.h │ │ │ ├── restore_mutation.cpp │ │ │ ├── restore_mutation.h │ │ │ ├── shape_analysis.cpp │ │ │ ├── shape_analysis.h │ │ │ ├── specialize_autogradzero.cpp │ │ │ ├── specialize_autogradzero.h │ │ │ ├── subgraph_rewrite.cpp │ │ │ ├── subgraph_rewrite.h │ │ │ ├── symbolic_shape_analysis.cpp │ │ │ ├── symbolic_shape_analysis.h │ │ │ ├── symbolic_shape_cache.cpp │ │ │ ├── symbolic_shape_cache.h │ │ │ ├── symbolic_shape_runtime_fusion.cpp │ │ │ ├── symbolic_shape_runtime_fusion.h │ │ │ ├── tensorexpr_fuser.cpp │ │ │ ├── tensorexpr_fuser.h │ │ │ ├── update_differentiable_graph_requires_grad.cpp │ │ │ ├── update_differentiable_graph_requires_grad.h │ │ │ ├── utils │ │ │ │ ├── check_alias_annotation.cpp │ │ │ │ ├── check_alias_annotation.h │ │ │ │ ├── memory_dag.cpp │ │ │ │ ├── memory_dag.h │ │ │ │ ├── op_registry.cpp │ │ │ │ ├── op_registry.h │ │ │ │ ├── optimization_utils.cpp │ │ │ │ ├── optimization_utils.h │ │ │ │ ├── subgraph_utils.cpp │ │ │ │ └── subgraph_utils.h │ │ │ ├── value_refinement_utils.cpp │ │ │ ├── value_refinement_utils.h │ │ │ ├── variadic_ops.cpp │ │ │ ├── variadic_ops.h │ │ │ ├── vulkan_rewrite.cpp │ │ │ ├── vulkan_rewrite.h │ │ │ ├── xnnpack_rewrite.cpp │ │ │ └── xnnpack_rewrite.h │ │ ├── python │ │ │ ├── init.cpp │ │ │ ├── init.h │ │ │ ├── module_python.h │ │ │ ├── pybind.h │ │ │ ├── pybind_utils.cpp │ │ │ ├── pybind_utils.h │ │ │ ├── python_arg_flatten.cpp │ │ │ ├── python_arg_flatten.h │ │ │ ├── python_custom_class.cpp │ │ │ ├── python_custom_class.h │ │ │ ├── python_dict.cpp │ │ │ ├── python_dict.h │ │ │ ├── python_interpreter.cpp │ │ │ ├── python_ir.cpp │ │ │ ├── python_ir.h │ │ │ ├── python_ivalue.h │ │ │ ├── python_list.cpp │ │ │ ├── python_list.h │ │ │ ├── python_sugared_value.cpp │ │ │ ├── python_sugared_value.h │ │ │ ├── python_tracer.cpp │ │ │ ├── python_tracer.h │ │ │ ├── python_tree_views.cpp │ │ │ ├── python_tree_views.h │ │ │ ├── script_init.cpp │ │ │ ├── script_init.h │ │ │ ├── update_graph_executor_opt.cpp │ │ │ ├── update_graph_executor_opt.h │ │ │ ├── utf8_decoding_ignore.cpp │ │ │ └── utf8_decoding_ignore.h │ │ ├── resource_guard.h │ │ ├── runtime │ │ │ ├── argument_spec.cpp │ │ │ ├── argument_spec.h │ │ │ ├── autodiff.cpp │ │ │ ├── autodiff.h │ │ │ ├── calculate_necessary_args.h │ │ │ ├── custom_operator.h │ │ │ ├── decomposition_registry.cpp │ │ │ ├── decomposition_registry.h │ │ │ ├── decomposition_registry_util.cpp │ │ │ ├── decomposition_registry_util.h │ │ │ ├── exception_message.h │ │ │ ├── graph_executor.cpp │ │ │ ├── graph_executor.h │ │ │ ├── graph_executor_impl.h │ │ │ ├── graph_iterator.h │ │ │ ├── instruction.cpp │ │ │ ├── instruction.h │ │ │ ├── interpreter.cpp │ │ │ ├── interpreter.h │ │ │ ├── interpreter │ │ │ │ ├── can_emit_inline.h │ │ │ │ ├── code_impl.h │ │ │ │ ├── frame.cpp │ │ │ │ ├── frame.h │ │ │ │ ├── preprocess_graph.cpp │ │ │ │ └── preprocess_graph.h │ │ │ ├── jit_exception.cpp │ │ │ ├── jit_exception.h │ │ │ ├── jit_trace.cpp │ │ │ ├── jit_trace.h │ │ │ ├── logging.cpp │ │ │ ├── logging.h │ │ │ ├── operator.cpp │ │ │ ├── operator.h │ │ │ ├── operator_options.h │ │ │ ├── print_handler.cpp │ │ │ ├── print_handler.h │ │ │ ├── profiling_graph_executor_impl.cpp │ │ │ ├── profiling_graph_executor_impl.h │ │ │ ├── profiling_record.cpp │ │ │ ├── profiling_record.h │ │ │ ├── register_c10_ops.cpp │ │ │ ├── register_cuda_ops.cpp │ │ │ ├── register_distributed_ops.cpp │ │ │ ├── register_ops_utils.cpp │ │ │ ├── register_ops_utils.h │ │ │ ├── register_prim_ops.cpp │ │ │ ├── register_prim_ops_fulljit.cpp │ │ │ ├── register_special_ops.cpp │ │ │ ├── script_profile.cpp │ │ │ ├── script_profile.h │ │ │ ├── serialized_shape_function_registry.cpp │ │ │ ├── serialized_shape_function_registry.h │ │ │ ├── shape_function_registry.h │ │ │ ├── simple_graph_executor_impl.cpp │ │ │ ├── simple_graph_executor_impl.h │ │ │ ├── slice_indices_adjust.cpp │ │ │ ├── slice_indices_adjust.h │ │ │ ├── static │ │ │ │ ├── .clang-tidy │ │ │ │ ├── ProcessedNodeInputs.cpp │ │ │ │ ├── ProcessedNodeInputs.h │ │ │ │ ├── README.md │ │ │ │ ├── fusion.cpp │ │ │ │ ├── fusion.h │ │ │ │ ├── generated_ops.cpp │ │ │ │ ├── impl.cpp │ │ │ │ ├── impl.h │ │ │ │ ├── init.cpp │ │ │ │ ├── init.h │ │ │ │ ├── memory_planner.cpp │ │ │ │ ├── memory_planner.h │ │ │ │ ├── native_ops.cpp │ │ │ │ ├── ops.cpp │ │ │ │ ├── ops.h │ │ │ │ ├── passes.cpp │ │ │ │ ├── passes.h │ │ │ │ ├── processed_node_wrapper.h │ │ │ │ ├── static_method.h │ │ │ │ ├── te_wrapper.cpp │ │ │ │ └── te_wrapper.h │ │ │ ├── symbolic_script.cpp │ │ │ ├── symbolic_script.h │ │ │ ├── symbolic_shape_registry.cpp │ │ │ ├── symbolic_shape_registry.h │ │ │ ├── symbolic_shape_registry_util.cpp │ │ │ ├── symbolic_shape_registry_util.h │ │ │ ├── vararg_functions.cpp │ │ │ ├── vararg_functions.h │ │ │ └── variable_tensor_list.h │ │ ├── serialization │ │ │ ├── callstack_debug_info_serialization.cpp │ │ │ ├── callstack_debug_info_serialization.h │ │ │ ├── export.cpp │ │ │ ├── export.h │ │ │ ├── export_bytecode.cpp │ │ │ ├── export_bytecode.h │ │ │ ├── export_module.cpp │ │ │ ├── flatbuffer_serializer.cpp │ │ │ ├── flatbuffer_serializer.h │ │ │ ├── flatbuffer_serializer_jit.cpp │ │ │ ├── flatbuffer_serializer_jit.h │ │ │ ├── import.cpp │ │ │ ├── import.h │ │ │ ├── import_export_constants.h │ │ │ ├── import_export_functions.h │ │ │ ├── import_export_helpers.cpp │ │ │ ├── import_export_helpers.h │ │ │ ├── import_read.cpp │ │ │ ├── import_read.h │ │ │ ├── import_source.cpp │ │ │ ├── import_source.h │ │ │ ├── mobile_bytecode.fbs │ │ │ ├── mobile_bytecode_generated.h │ │ │ ├── onnx.cpp │ │ │ ├── onnx.h │ │ │ ├── pickle.cpp │ │ │ ├── pickle.h │ │ │ ├── pickler.cpp │ │ │ ├── pickler.h │ │ │ ├── python_print.cpp │ │ │ ├── python_print.h │ │ │ ├── source_range_serialization.cpp │ │ │ ├── source_range_serialization.h │ │ │ ├── source_range_serialization_impl.h │ │ │ ├── storage_context.h │ │ │ ├── type_name_uniquer.cpp │ │ │ ├── type_name_uniquer.h │ │ │ ├── unpickler.cpp │ │ │ └── unpickler.h │ │ ├── tensorexpr │ │ │ ├── ConditionalsInTE.md │ │ │ ├── IRSpecification.md │ │ │ ├── analysis.h │ │ │ ├── block_codegen.cpp │ │ │ ├── block_codegen.h │ │ │ ├── bounds_inference.cpp │ │ │ ├── bounds_inference.h │ │ │ ├── bounds_overlap.cpp │ │ │ ├── bounds_overlap.h │ │ │ ├── codegen.cpp │ │ │ ├── codegen.h │ │ │ ├── codegen_external.py │ │ │ ├── cpp_codegen.cpp │ │ │ ├── cpp_codegen.h │ │ │ ├── cpp_intrinsics.h │ │ │ ├── cuda_codegen.cpp │ │ │ ├── cuda_codegen.h │ │ │ ├── cuda_random.h │ │ │ ├── eval.cpp │ │ │ ├── eval.h │ │ │ ├── exceptions.h │ │ │ ├── expr.cpp │ │ │ ├── expr.h │ │ │ ├── external_functions.cpp │ │ │ ├── external_functions.h │ │ │ ├── external_functions_codegen.cpp │ │ │ ├── external_functions_core.cpp │ │ │ ├── external_functions_core.h │ │ │ ├── external_functions_registry.cpp │ │ │ ├── external_functions_registry.h │ │ │ ├── fwd_decls.h │ │ │ ├── graph_opt.cpp │ │ │ ├── graph_opt.h │ │ │ ├── half_support.h │ │ │ ├── hash_provider.cpp │ │ │ ├── hash_provider.h │ │ │ ├── intrinsic_symbols.cpp │ │ │ ├── intrinsic_symbols.h │ │ │ ├── ir.cpp │ │ │ ├── ir.h │ │ │ ├── ir_cloner.cpp │ │ │ ├── ir_cloner.h │ │ │ ├── ir_mutator.cpp │ │ │ ├── ir_mutator.h │ │ │ ├── ir_printer.cpp │ │ │ ├── ir_printer.h │ │ │ ├── ir_simplifier.cpp │ │ │ ├── ir_simplifier.h │ │ │ ├── ir_verifier.cpp │ │ │ ├── ir_verifier.h │ │ │ ├── ir_visitor.cpp │ │ │ ├── ir_visitor.h │ │ │ ├── kernel.cpp │ │ │ ├── kernel.h │ │ │ ├── llvm_codegen.cpp │ │ │ ├── llvm_codegen.h │ │ │ ├── llvm_jit.cpp │ │ │ ├── llvm_jit.h │ │ │ ├── loopnest.cpp │ │ │ ├── loopnest.h │ │ │ ├── loopnest_randomization.cpp │ │ │ ├── loopnest_randomization.h │ │ │ ├── lowerings.cpp │ │ │ ├── lowerings.h │ │ │ ├── mem_dependency_checker.cpp │ │ │ ├── mem_dependency_checker.h │ │ │ ├── operators │ │ │ │ ├── conv2d.cpp │ │ │ │ ├── conv2d.h │ │ │ │ ├── matmul.cpp │ │ │ │ ├── matmul.h │ │ │ │ ├── misc.cpp │ │ │ │ ├── misc.h │ │ │ │ ├── norm.cpp │ │ │ │ ├── norm.h │ │ │ │ ├── operators.h │ │ │ │ ├── pointwise.cpp │ │ │ │ ├── pointwise.h │ │ │ │ ├── quantization.cpp │ │ │ │ ├── quantization.h │ │ │ │ ├── reduction.cpp │ │ │ │ ├── reduction.h │ │ │ │ ├── softmax.cpp │ │ │ │ └── softmax.h │ │ │ ├── reduction.cpp │ │ │ ├── reduction.h │ │ │ ├── registerizer.cpp │ │ │ ├── registerizer.h │ │ │ ├── scripts │ │ │ │ └── bisect.py │ │ │ ├── stmt.h │ │ │ ├── tensor.cpp │ │ │ ├── tensor.h │ │ │ ├── tensorexpr_init.cpp │ │ │ ├── tensorexpr_init.h │ │ │ ├── types.cpp │ │ │ ├── types.h │ │ │ ├── unique_name_manager.cpp │ │ │ ├── unique_name_manager.h │ │ │ └── var_substitutor.h │ │ └── testing │ │ │ ├── catch_utils.hpp │ │ │ ├── file_check.cpp │ │ │ ├── file_check.h │ │ │ ├── hooks_for_testing.cpp │ │ │ └── hooks_for_testing.h │ ├── lazy │ │ ├── backend │ │ │ ├── backend_data.h │ │ │ ├── backend_device.cpp │ │ │ ├── backend_device.h │ │ │ ├── backend_interface.cpp │ │ │ ├── backend_interface.h │ │ │ ├── lowering_context.cpp │ │ │ └── lowering_context.h │ │ ├── core │ │ │ ├── cache.h │ │ │ ├── config.cpp │ │ │ ├── config.h │ │ │ ├── debug_util.cpp │ │ │ ├── debug_util.h │ │ │ ├── dynamic_ir.h │ │ │ ├── hash.cpp │ │ │ ├── hash.h │ │ │ ├── helpers.cpp │ │ │ ├── helpers.h │ │ │ ├── internal_ops │ │ │ │ └── ltc_ops.h │ │ │ ├── ir.cpp │ │ │ ├── ir.h │ │ │ ├── ir_builder.h │ │ │ ├── ir_dump_util.cpp │ │ │ ├── ir_dump_util.h │ │ │ ├── ir_metadata.cpp │ │ │ ├── ir_metadata.h │ │ │ ├── ir_util.cpp │ │ │ ├── ir_util.h │ │ │ ├── lazy_graph_executor.cpp │ │ │ ├── lazy_graph_executor.h │ │ │ ├── metrics.cpp │ │ │ ├── metrics.h │ │ │ ├── multi_wait.cpp │ │ │ ├── multi_wait.h │ │ │ ├── ops │ │ │ │ ├── arithmetic_ir_ops.cpp │ │ │ │ ├── arithmetic_ir_ops.h │ │ │ │ ├── utils.cpp │ │ │ │ └── utils.h │ │ │ ├── permutation_util.cpp │ │ │ ├── permutation_util.h │ │ │ ├── shape.cpp │ │ │ ├── shape.h │ │ │ ├── shape_inference.cpp │ │ │ ├── shape_inference.h │ │ │ ├── tensor.cpp │ │ │ ├── tensor.h │ │ │ ├── tensor_impl.cpp │ │ │ ├── tensor_impl.h │ │ │ ├── tensor_util.cpp │ │ │ ├── tensor_util.h │ │ │ ├── thread_pool.cpp │ │ │ ├── thread_pool.h │ │ │ ├── trie.cpp │ │ │ ├── trie.h │ │ │ ├── unique.h │ │ │ └── util.h │ │ ├── generated │ │ │ └── README.md │ │ ├── python │ │ │ ├── README.md │ │ │ ├── init.cpp │ │ │ ├── init.h │ │ │ ├── python_util.cpp │ │ │ └── python_util.h │ │ ├── test_mnist.py │ │ ├── ts_backend │ │ │ ├── config.cpp │ │ │ ├── config.h │ │ │ ├── dynamic_ir.cpp │ │ │ ├── dynamic_ir.h │ │ │ ├── ir_builder.h │ │ │ ├── ops │ │ │ │ ├── device_data.cpp │ │ │ │ ├── device_data.h │ │ │ │ ├── generic.cpp │ │ │ │ ├── generic.h │ │ │ │ └── to_copy.h │ │ │ ├── tensor_aten_ops.cpp │ │ │ ├── tensor_aten_ops.h │ │ │ ├── ts_autograd_functions.cpp │ │ │ ├── ts_autograd_functions.h │ │ │ ├── ts_backend_impl.cpp │ │ │ ├── ts_backend_impl.h │ │ │ ├── ts_eager_fallback.cpp │ │ │ ├── ts_eager_fallback.h │ │ │ ├── ts_lowering_context.cpp │ │ │ ├── ts_lowering_context.h │ │ │ ├── ts_native_functions.cpp │ │ │ ├── ts_node.cpp │ │ │ ├── ts_node.h │ │ │ ├── ts_node_lowering.cpp │ │ │ └── ts_node_lowering.h │ │ └── tutorial.md │ ├── monitor │ │ ├── counters.cpp │ │ ├── counters.h │ │ ├── events.cpp │ │ ├── events.h │ │ ├── python_init.cpp │ │ └── python_init.h │ ├── mps │ │ ├── Module.cpp │ │ └── Module.h │ ├── mtia │ │ ├── Module.cpp │ │ └── Module.h │ ├── multiprocessing │ │ ├── init.cpp │ │ └── init.h │ ├── onnx │ │ ├── back_compat.h │ │ ├── diagnostics │ │ │ ├── diagnostics.h │ │ │ └── generated │ │ │ │ └── rules.h │ │ ├── init.cpp │ │ ├── init.h │ │ └── onnx.h │ ├── profiler │ │ ├── README.md │ │ ├── api.h │ │ ├── collection.cpp │ │ ├── collection.h │ │ ├── combined_traceback.cpp │ │ ├── combined_traceback.h │ │ ├── containers.h │ │ ├── data_flow.cpp │ │ ├── data_flow.h │ │ ├── events.h │ │ ├── kineto_client_interface.cpp │ │ ├── kineto_client_interface.h │ │ ├── kineto_shim.cpp │ │ ├── kineto_shim.h │ │ ├── orchestration │ │ │ ├── observer.cpp │ │ │ ├── observer.h │ │ │ ├── python_tracer.cpp │ │ │ ├── python_tracer.h │ │ │ ├── vulkan.cpp │ │ │ └── vulkan.h │ │ ├── perf-inl.h │ │ ├── perf.cpp │ │ ├── perf.h │ │ ├── python │ │ │ ├── combined_traceback.cpp │ │ │ ├── combined_traceback.h │ │ │ ├── init.cpp │ │ │ ├── init.h │ │ │ └── pybind.h │ │ ├── standalone │ │ │ ├── execution_trace_observer.cpp │ │ │ ├── execution_trace_observer.h │ │ │ ├── itt_observer.cpp │ │ │ ├── itt_observer.h │ │ │ ├── nvtx_observer.cpp │ │ │ ├── nvtx_observer.h │ │ │ ├── privateuse1_observer.cpp │ │ │ └── privateuse1_observer.h │ │ ├── stubs │ │ │ ├── base.cpp │ │ │ ├── base.h │ │ │ ├── cuda.cpp │ │ │ └── itt.cpp │ │ ├── unwind │ │ │ ├── action.h │ │ │ ├── communicate.h │ │ │ ├── debug_info.h │ │ │ ├── dwarf_enums.h │ │ │ ├── dwarf_symbolize_enums.h │ │ │ ├── eh_frame_hdr.h │ │ │ ├── fast_symbolizer.h │ │ │ ├── fde.h │ │ │ ├── lexer.h │ │ │ ├── line_number_program.h │ │ │ ├── mem_file.h │ │ │ ├── range_table.h │ │ │ ├── sections.h │ │ │ ├── unwind.cpp │ │ │ ├── unwind.h │ │ │ ├── unwind_error.h │ │ │ ├── unwind_fb.cpp │ │ │ └── unwinder.h │ │ ├── util.cpp │ │ └── util.h │ ├── python_dimname.cpp │ ├── python_dimname.h │ ├── python_headers.h │ ├── serialization.cpp │ ├── serialization.h │ ├── stub.c │ ├── tensor │ │ ├── python_tensor.cpp │ │ └── python_tensor.h │ ├── utils.cpp │ ├── utils.h │ ├── utils │ │ ├── byte_order.cpp │ │ ├── byte_order.h │ │ ├── cpp_stacktraces.cpp │ │ ├── cpp_stacktraces.h │ │ ├── cuda_enabled.h │ │ ├── device_lazy_init.cpp │ │ ├── device_lazy_init.h │ │ ├── disable_torch_function.cpp │ │ ├── disable_torch_function.h │ │ ├── generated_serialization_types.h │ │ ├── init.cpp │ │ ├── init.h │ │ ├── invalid_arguments.cpp │ │ ├── invalid_arguments.h │ │ ├── nested.cpp │ │ ├── nested.h │ │ ├── numpy_stub.h │ │ ├── object_ptr.cpp │ │ ├── object_ptr.h │ │ ├── out_types.cpp │ │ ├── out_types.h │ │ ├── pybind.cpp │ │ ├── pybind.h │ │ ├── pycfunction_helpers.h │ │ ├── pyobject_preservation.cpp │ │ ├── pyobject_preservation.h │ │ ├── python_arg_parser.cpp │ │ ├── python_arg_parser.h │ │ ├── python_compat.h │ │ ├── python_dispatch.cpp │ │ ├── python_dispatch.h │ │ ├── python_numbers.h │ │ ├── python_raii.h │ │ ├── python_scalars.h │ │ ├── python_strings.h │ │ ├── python_stub.h │ │ ├── python_symnode.cpp │ │ ├── python_symnode.h │ │ ├── python_torch_function_mode.h │ │ ├── python_tuples.h │ │ ├── pythoncapi_compat.h │ │ ├── schema_info.cpp │ │ ├── schema_info.h │ │ ├── six.h │ │ ├── structseq.cpp │ │ ├── structseq.h │ │ ├── tensor_apply.cpp │ │ ├── tensor_apply.h │ │ ├── tensor_dtypes.cpp │ │ ├── tensor_dtypes.h │ │ ├── tensor_flatten.cpp │ │ ├── tensor_flatten.h │ │ ├── tensor_layouts.cpp │ │ ├── tensor_layouts.h │ │ ├── tensor_list.cpp │ │ ├── tensor_list.h │ │ ├── tensor_memoryformats.cpp │ │ ├── tensor_memoryformats.h │ │ ├── tensor_new.cpp │ │ ├── tensor_new.h │ │ ├── tensor_numpy.cpp │ │ ├── tensor_numpy.h │ │ ├── tensor_qschemes.cpp │ │ ├── tensor_qschemes.h │ │ ├── tensor_types.cpp │ │ ├── tensor_types.h │ │ ├── throughput_benchmark-inl.h │ │ ├── throughput_benchmark.cpp │ │ ├── throughput_benchmark.h │ │ ├── torch_dispatch_mode.h │ │ ├── variadic.cpp │ │ ├── variadic.h │ │ ├── verbose.cpp │ │ └── verbose.h │ └── xpu │ │ ├── Event.cpp │ │ ├── Event.h │ │ ├── Module.cpp │ │ ├── Module.h │ │ ├── Stream.cpp │ │ └── Stream.h ├── cuda │ ├── __init__.py │ ├── _gpu_trace.py │ ├── _memory_viz.py │ ├── _sanitizer.py │ ├── _utils.py │ ├── amp │ │ ├── __init__.py │ │ ├── autocast_mode.py │ │ ├── common.py │ │ └── grad_scaler.py │ ├── comm.py │ ├── error.py │ ├── gds.py │ ├── graphs.py │ ├── jiterator.py │ ├── memory.py │ ├── nccl.py │ ├── nvtx.py │ ├── profiler.py │ ├── random.py │ ├── sparse.py │ ├── streams.py │ └── tunable.py ├── custom_class.h ├── custom_class_detail.h ├── distributed │ ├── CONTRIBUTING.md │ ├── __init__.py │ ├── _checkpointable.py │ ├── _composable │ │ ├── __init__.py │ │ ├── checkpoint_activation.py │ │ ├── contract.py │ │ ├── fsdp │ │ │ ├── __init__.py │ │ │ └── fully_shard.py │ │ └── replicate.py │ ├── _composable_state.py │ ├── _functional_collectives.py │ ├── _functional_collectives_impl.py │ ├── _serialization.py │ ├── _shard │ │ ├── __init__.py │ │ ├── _utils.py │ │ ├── api.py │ │ ├── checkpoint │ │ │ └── __init__.py │ │ ├── common_op_utils.py │ │ ├── metadata.py │ │ ├── op_registry_utils.py │ │ ├── sharded_optim │ │ │ ├── __init__.py │ │ │ └── api.py │ │ ├── sharded_tensor │ │ │ ├── __init__.py │ │ │ ├── _ops │ │ │ │ ├── __init__.py │ │ │ │ ├── _common.py │ │ │ │ ├── binary_cmp.py │ │ │ │ ├── init.py │ │ │ │ ├── misc_ops.py │ │ │ │ └── tensor_ops.py │ │ │ ├── api.py │ │ │ ├── logger.py │ │ │ ├── logging_handlers.py │ │ │ ├── metadata.py │ │ │ ├── reshard.py │ │ │ ├── shard.py │ │ │ └── utils.py │ │ ├── sharder.py │ │ ├── sharding_plan │ │ │ ├── __init__.py │ │ │ └── api.py │ │ └── sharding_spec │ │ │ ├── __init__.py │ │ │ ├── _internals.py │ │ │ ├── api.py │ │ │ ├── chunk_sharding_spec.py │ │ │ └── chunk_sharding_spec_ops │ │ │ ├── __init__.py │ │ │ ├── _common.py │ │ │ ├── embedding.py │ │ │ └── embedding_bag.py │ ├── _sharded_tensor │ │ └── __init__.py │ ├── _sharding_spec │ │ └── __init__.py │ ├── _state_dict_utils.py │ ├── _symmetric_memory │ │ └── __init__.py │ ├── _tensor │ │ ├── __init__.py │ │ ├── api.py │ │ └── placement_types.py │ ├── _tools │ │ ├── __init__.py │ │ ├── fsdp2_mem_tracker.py │ │ ├── ilp_utils.py │ │ ├── mem_tracker.py │ │ ├── memory_tracker.py │ │ ├── mod_tracker.py │ │ ├── runtime_estimator.py │ │ ├── sac_estimator.py │ │ └── sac_ilp.py │ ├── algorithms │ │ ├── __init__.py │ │ ├── _checkpoint │ │ │ ├── __init__.py │ │ │ └── checkpoint_wrapper.py │ │ ├── _comm_hooks │ │ │ ├── __init__.py │ │ │ └── default_hooks.py │ │ ├── _optimizer_overlap │ │ │ ├── __init__.py │ │ │ └── optimizer_overlap.py │ │ ├── _quantization │ │ │ ├── __init__.py │ │ │ └── quantization.py │ │ ├── ddp_comm_hooks │ │ │ ├── __init__.py │ │ │ ├── ddp_zero_hook.py │ │ │ ├── debugging_hooks.py │ │ │ ├── default_hooks.py │ │ │ ├── mixed_precision_hooks.py │ │ │ ├── optimizer_overlap_hooks.py │ │ │ ├── post_localSGD_hook.py │ │ │ ├── powerSGD_hook.py │ │ │ └── quantization_hooks.py │ │ ├── join.py │ │ └── model_averaging │ │ │ ├── __init__.py │ │ │ ├── averagers.py │ │ │ ├── hierarchical_model_averager.py │ │ │ └── utils.py │ ├── argparse_util.py │ ├── autograd │ │ └── __init__.py │ ├── benchmarks │ │ ├── README.md │ │ └── benchmark_ddp_rpc.py │ ├── c10d_logger.py │ ├── checkpoint │ │ ├── __init__.py │ │ ├── _checkpointer.py │ │ ├── _dedup_save_plans.py │ │ ├── _dedup_tensors.py │ │ ├── _extension.py │ │ ├── _fsspec_filesystem.py │ │ ├── _hf_storage.py │ │ ├── _nested_dict.py │ │ ├── _sharded_tensor_utils.py │ │ ├── _storage_utils.py │ │ ├── _traverse.py │ │ ├── _version.py │ │ ├── api.py │ │ ├── default_planner.py │ │ ├── examples │ │ │ ├── async_checkpointing_example.py │ │ │ ├── fsdp_checkpoint_example.py │ │ │ └── stateful_example.py │ │ ├── filesystem.py │ │ ├── format_utils.py │ │ ├── logger.py │ │ ├── logging_handlers.py │ │ ├── metadata.py │ │ ├── optimizer.py │ │ ├── planner.py │ │ ├── planner_helpers.py │ │ ├── resharding.py │ │ ├── staging.py │ │ ├── state_dict.py │ │ ├── state_dict_loader.py │ │ ├── state_dict_saver.py │ │ ├── stateful.py │ │ ├── storage.py │ │ └── utils.py │ ├── collective_utils.py │ ├── constants.py │ ├── device_mesh.py │ ├── distributed_c10d.py │ ├── elastic │ │ ├── __init__.py │ │ ├── agent │ │ │ ├── __init__.py │ │ │ └── server │ │ │ │ ├── __init__.py │ │ │ │ ├── api.py │ │ │ │ ├── health_check_server.py │ │ │ │ └── local_elastic_agent.py │ │ ├── control_plane.py │ │ ├── events │ │ │ ├── __init__.py │ │ │ ├── api.py │ │ │ └── handlers.py │ │ ├── metrics │ │ │ ├── __init__.py │ │ │ └── api.py │ │ ├── multiprocessing │ │ │ ├── __init__.py │ │ │ ├── api.py │ │ │ ├── errors │ │ │ │ ├── __init__.py │ │ │ │ ├── error_handler.py │ │ │ │ └── handlers.py │ │ │ ├── redirects.py │ │ │ ├── subprocess_handler │ │ │ │ ├── __init__.py │ │ │ │ ├── handlers.py │ │ │ │ └── subprocess_handler.py │ │ │ └── tail_log.py │ │ ├── rendezvous │ │ │ ├── __init__.py │ │ │ ├── _etcd_stub.py │ │ │ ├── api.py │ │ │ ├── c10d_rendezvous_backend.py │ │ │ ├── dynamic_rendezvous.py │ │ │ ├── etcd_rendezvous.py │ │ │ ├── etcd_rendezvous_backend.py │ │ │ ├── etcd_server.py │ │ │ ├── etcd_store.py │ │ │ ├── registry.py │ │ │ ├── static_tcp_rendezvous.py │ │ │ └── utils.py │ │ ├── timer │ │ │ ├── __init__.py │ │ │ ├── api.py │ │ │ ├── debug_info_logging.py │ │ │ ├── file_based_local_timer.py │ │ │ └── local_timer.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── api.py │ │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── cycling_iterator.py │ │ │ └── elastic_distributed_sampler.py │ │ │ ├── distributed.py │ │ │ ├── log_level.py │ │ │ ├── logging.py │ │ │ └── store.py │ ├── examples │ │ └── memory_tracker_example.py │ ├── fsdp │ │ ├── __init__.py │ │ ├── _common_utils.py │ │ ├── _debug_utils.py │ │ ├── _dynamo_utils.py │ │ ├── _exec_order_utils.py │ │ ├── _flat_param.py │ │ ├── _fsdp_extensions.py │ │ ├── _fully_shard │ │ │ ├── __init__.py │ │ │ ├── _fsdp_api.py │ │ │ ├── _fsdp_collectives.py │ │ │ ├── _fsdp_common.py │ │ │ ├── _fsdp_init.py │ │ │ ├── _fsdp_param.py │ │ │ ├── _fsdp_param_group.py │ │ │ ├── _fsdp_state.py │ │ │ └── _fully_shard.py │ │ ├── _init_utils.py │ │ ├── _limiter_utils.py │ │ ├── _optim_utils.py │ │ ├── _runtime_utils.py │ │ ├── _shard_utils.py │ │ ├── _state_dict_utils.py │ │ ├── _trace_utils.py │ │ ├── _traversal_utils.py │ │ ├── _unshard_param_utils.py │ │ ├── _wrap_utils.py │ │ ├── api.py │ │ ├── fully_sharded_data_parallel.py │ │ ├── sharded_grad_scaler.py │ │ └── wrap.py │ ├── launch.py │ ├── launcher │ │ ├── __init__.py │ │ └── api.py │ ├── logging_handlers.py │ ├── nn │ │ ├── __init__.py │ │ ├── api │ │ │ ├── __init__.py │ │ │ └── remote_module.py │ │ ├── functional.py │ │ └── jit │ │ │ ├── __init__.py │ │ │ ├── instantiator.py │ │ │ └── templates │ │ │ ├── __init__.py │ │ │ └── remote_module_template.py │ ├── optim │ │ ├── __init__.py │ │ ├── _deprecation_warning.py │ │ ├── apply_optimizer_in_backward.py │ │ ├── functional_adadelta.py │ │ ├── functional_adagrad.py │ │ ├── functional_adam.py │ │ ├── functional_adamax.py │ │ ├── functional_adamw.py │ │ ├── functional_rmsprop.py │ │ ├── functional_rprop.py │ │ ├── functional_sgd.py │ │ ├── named_optimizer.py │ │ ├── optimizer.py │ │ ├── post_localSGD_optimizer.py │ │ ├── utils.py │ │ ├── zero_redundancy_optimizer.py │ │ └── zero_redundancy_optimizer.pyi │ ├── pipelining │ │ ├── README.md │ │ ├── _IR.py │ │ ├── __init__.py │ │ ├── _backward.py │ │ ├── _debug.py │ │ ├── _unflatten.py │ │ ├── _utils.py │ │ ├── microbatch.py │ │ ├── schedules.py │ │ └── stage.py │ ├── remote_device.py │ ├── rendezvous.py │ ├── rpc │ │ ├── __init__.py │ │ ├── _testing │ │ │ ├── __init__.py │ │ │ └── faulty_agent_backend_registry.py │ │ ├── _utils.py │ │ ├── api.py │ │ ├── backend_registry.py │ │ ├── constants.py │ │ ├── functions.py │ │ ├── internal.py │ │ ├── options.py │ │ ├── rref_proxy.py │ │ └── server_process_global_profiler.py │ ├── run.py │ ├── tensor │ │ ├── README.md │ │ ├── __init__.py │ │ ├── _api.py │ │ ├── _collective_utils.py │ │ ├── _dispatch.py │ │ ├── _dtensor_spec.py │ │ ├── _op_schema.py │ │ ├── _ops │ │ │ ├── __init__.py │ │ │ ├── _common_rules.py │ │ │ ├── _conv_ops.py │ │ │ ├── _einsum_strategy.py │ │ │ ├── _embedding_ops.py │ │ │ ├── _experimental_ops.py │ │ │ ├── _math_ops.py │ │ │ ├── _matrix_ops.py │ │ │ ├── _pointwise_ops.py │ │ │ ├── _random_ops.py │ │ │ ├── _tensor_ops.py │ │ │ ├── _view_ops.py │ │ │ └── utils.py │ │ ├── _random.py │ │ ├── _redistribute.py │ │ ├── _sharding_prop.py │ │ ├── _shards_wrapper.py │ │ ├── _tp_conv.py │ │ ├── _utils.py │ │ ├── debug │ │ │ ├── __init__.py │ │ │ ├── _comm_mode.py │ │ │ ├── _op_coverage.py │ │ │ ├── _visualize_sharding.py │ │ │ └── comm_mode_broswer_visual.js │ │ ├── device_mesh.py │ │ ├── examples │ │ │ ├── comm_mode_features_example.py │ │ │ ├── convnext_example.py │ │ │ ├── flex_attention_cp.py │ │ │ ├── torchrec_sharding_example.py │ │ │ └── visualize_sharding_example.py │ │ ├── experimental │ │ │ ├── __init__.py │ │ │ ├── _attention.py │ │ │ ├── _func_map.py │ │ │ ├── _register_sharding.py │ │ │ └── _tp_transform.py │ │ ├── parallel │ │ │ ├── __init__.py │ │ │ ├── _data_parallel_utils.py │ │ │ ├── _utils.py │ │ │ ├── api.py │ │ │ ├── ddp.py │ │ │ ├── fsdp.py │ │ │ ├── input_reshard.py │ │ │ ├── loss.py │ │ │ └── style.py │ │ └── placement_types.py │ └── utils.py ├── distributions │ ├── __init__.py │ ├── bernoulli.py │ ├── beta.py │ ├── binomial.py │ ├── categorical.py │ ├── cauchy.py │ ├── chi2.py │ ├── constraint_registry.py │ ├── constraints.py │ ├── continuous_bernoulli.py │ ├── dirichlet.py │ ├── distribution.py │ ├── exp_family.py │ ├── exponential.py │ ├── fishersnedecor.py │ ├── gamma.py │ ├── geometric.py │ ├── gumbel.py │ ├── half_cauchy.py │ ├── half_normal.py │ ├── independent.py │ ├── inverse_gamma.py │ ├── kl.py │ ├── kumaraswamy.py │ ├── laplace.py │ ├── lkj_cholesky.py │ ├── log_normal.py │ ├── logistic_normal.py │ ├── lowrank_multivariate_normal.py │ ├── mixture_same_family.py │ ├── multinomial.py │ ├── multivariate_normal.py │ ├── negative_binomial.py │ ├── normal.py │ ├── one_hot_categorical.py │ ├── pareto.py │ ├── poisson.py │ ├── relaxed_bernoulli.py │ ├── relaxed_categorical.py │ ├── studentT.py │ ├── transformed_distribution.py │ ├── transforms.py │ ├── uniform.py │ ├── utils.py │ ├── von_mises.py │ ├── weibull.py │ └── wishart.py ├── export │ ├── __init__.py │ ├── _draft_export.py │ ├── _remove_auto_functionalized_pass.py │ ├── _remove_effect_tokens_pass.py │ ├── _safeguard.py │ ├── _swap.py │ ├── _trace.py │ ├── _tree_utils.py │ ├── _unlift.py │ ├── custom_obj.py │ ├── custom_ops.py │ ├── decomp_utils.py │ ├── dynamic_shapes.py │ ├── experimental │ │ └── __init__.py │ ├── exported_program.py │ ├── graph_signature.py │ ├── passes │ │ └── __init__.py │ └── unflatten.py ├── extension.h ├── fft │ └── __init__.py ├── fhe │ ├── __init__.py │ ├── approx.py │ ├── bootstrapping.py │ ├── bs_context.py │ ├── ciphertext.py │ ├── client │ │ ├── __init__.py │ │ ├── bs_context.py │ │ ├── client.py │ │ ├── context.py │ │ └── gen_context.py │ ├── context.py │ ├── functional.py │ ├── homo_ops.py │ ├── hybrid_keyswitch.py │ └── utils.py ├── func │ └── __init__.py ├── functional.py ├── futures │ └── __init__.py ├── fx │ ├── README.md │ ├── __init__.py │ ├── _compatibility.py │ ├── _graph_pickler.py │ ├── _lazy_graph_module.py │ ├── _pytree.py │ ├── _symbolic_trace.py │ ├── _utils.py │ ├── annotate.py │ ├── config.py │ ├── experimental │ │ ├── __init__.py │ │ ├── _backward_state.py │ │ ├── _config.py │ │ ├── _constant_symnode.py │ │ ├── _dynamism.py │ │ ├── accelerator_partitioner.py │ │ ├── const_fold.py │ │ ├── debug.py │ │ ├── graph_gradual_typechecker.py │ │ ├── merge_matmul.py │ │ ├── meta_tracer.py │ │ ├── migrate_gradual_types │ │ │ ├── __init__.py │ │ │ ├── constraint.py │ │ │ ├── constraint_generator.py │ │ │ ├── constraint_transformation.py │ │ │ ├── operation.py │ │ │ ├── transform_to_z3.py │ │ │ ├── util.py │ │ │ └── z3_types.py │ │ ├── normalize.py │ │ ├── optimization.py │ │ ├── partitioner_utils.py │ │ ├── proxy_tensor.py │ │ ├── recording.py │ │ ├── refinement_types.py │ │ ├── rewriter.py │ │ ├── schema_type_annotation.py │ │ ├── shape_inference │ │ │ ├── infer_shape.py │ │ │ └── infer_symbol_values.py │ │ ├── sym_node.py │ │ ├── symbolic_shapes.py │ │ ├── unification │ │ │ ├── LICENSE.txt │ │ │ ├── __init__.py │ │ │ ├── core.py │ │ │ ├── dispatch.py │ │ │ ├── match.py │ │ │ ├── more.py │ │ │ ├── multipledispatch │ │ │ │ ├── __init__.py │ │ │ │ ├── conflict.py │ │ │ │ ├── core.py │ │ │ │ ├── dispatcher.py │ │ │ │ ├── utils.py │ │ │ │ └── variadic.py │ │ │ ├── unification_tools.py │ │ │ ├── utils.py │ │ │ └── variable.py │ │ ├── unify_refinements.py │ │ └── validator.py │ ├── graph.py │ ├── graph_module.py │ ├── immutable_collections.py │ ├── interpreter.py │ ├── node.py │ ├── operator_schemas.py │ ├── passes │ │ ├── README.md │ │ ├── __init__.py │ │ ├── _tensorify_python_scalars.py │ │ ├── annotate_getitem_nodes.py │ │ ├── backends │ │ │ ├── __init__.py │ │ │ └── cudagraphs.py │ │ ├── dialect │ │ │ ├── __init__.py │ │ │ └── common │ │ │ │ ├── __init__.py │ │ │ │ └── cse_pass.py │ │ ├── fake_tensor_prop.py │ │ ├── graph_drawer.py │ │ ├── graph_manipulation.py │ │ ├── graph_transform_observer.py │ │ ├── infra │ │ │ ├── __init__.py │ │ │ ├── partitioner.py │ │ │ ├── pass_base.py │ │ │ └── pass_manager.py │ │ ├── net_min_base.py │ │ ├── operator_support.py │ │ ├── param_fetch.py │ │ ├── pass_manager.py │ │ ├── reinplace.py │ │ ├── runtime_assert.py │ │ ├── shape_prop.py │ │ ├── split_module.py │ │ ├── split_utils.py │ │ ├── splitter_base.py │ │ ├── tests │ │ │ ├── __init__.py │ │ │ └── test_pass_manager.py │ │ ├── tools_common.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── common.py │ │ │ ├── fuser_utils.py │ │ │ ├── matcher_utils.py │ │ │ ├── matcher_with_name_node_map_utils.py │ │ │ └── source_matcher_utils.py │ ├── proxy.py │ ├── subgraph_rewriter.py │ ├── tensor_type.py │ └── traceback.py ├── hub.py ├── jit │ ├── __init__.py │ ├── _async.py │ ├── _await.py │ ├── _builtins.py │ ├── _check.py │ ├── _dataclass_impls.py │ ├── _decomposition_utils.py │ ├── _decompositions.py │ ├── _freeze.py │ ├── _fuser.py │ ├── _ir_utils.py │ ├── _logging.py │ ├── _monkeytype_config.py │ ├── _passes │ │ ├── __init__.py │ │ └── _property_propagation.py │ ├── _pickle.py │ ├── _recursive.py │ ├── _script.py │ ├── _script.pyi │ ├── _serialization.py │ ├── _shape_functions.py │ ├── _state.py │ ├── _trace.py │ ├── annotations.py │ ├── frontend.py │ ├── generate_bytecode.py │ ├── mobile │ │ └── __init__.py │ ├── quantized.py │ ├── supported_ops.py │ └── unsupported_tensor_ops.py ├── legacy │ └── README.md ├── lib │ ├── libshm │ │ ├── CMakeLists.txt │ │ ├── alloc_info.h │ │ ├── core.cpp │ │ ├── err.h │ │ ├── libshm.h │ │ ├── manager.cpp │ │ └── socket.h │ └── libshm_windows │ │ ├── CMakeLists.txt │ │ ├── core.cpp │ │ └── libshm.h ├── library.h ├── library.py ├── linalg │ └── __init__.py ├── masked │ ├── __init__.py │ ├── _docs.py │ ├── _ops.py │ └── maskedtensor │ │ ├── __init__.py │ │ ├── _ops_refs.py │ │ ├── binary.py │ │ ├── core.py │ │ ├── creation.py │ │ ├── passthrough.py │ │ ├── reductions.py │ │ └── unary.py ├── monitor │ └── __init__.py ├── mps │ ├── __init__.py │ ├── event.py │ └── profiler.py ├── mtia │ ├── __init__.py │ ├── _utils.py │ └── memory.py ├── multiprocessing │ ├── __init__.py │ ├── _atfork.py │ ├── cuda_multiprocessing.md │ ├── pool.py │ ├── queue.py │ ├── reductions.py │ └── spawn.py ├── nested │ ├── __init__.py │ └── _internal │ │ ├── __init__.py │ │ ├── nested_int.py │ │ ├── nested_tensor.py │ │ ├── ops.py │ │ └── sdpa.py ├── nn │ ├── __init__.py │ ├── _reduction.py │ ├── attention │ │ ├── __init__.py │ │ ├── _utils.py │ │ ├── bias.py │ │ ├── experimental │ │ │ ├── __init__.py │ │ │ └── _paged_attention.py │ │ └── flex_attention.py │ ├── backends │ │ ├── __init__.py │ │ └── thnn.py │ ├── common_types.py │ ├── cpp.py │ ├── functional.py │ ├── functional.pyi.in │ ├── grad.py │ ├── init.py │ ├── intrinsic │ │ ├── __init__.py │ │ ├── modules │ │ │ ├── __init__.py │ │ │ └── fused.py │ │ ├── qat │ │ │ ├── __init__.py │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── conv_fused.py │ │ │ │ ├── linear_fused.py │ │ │ │ └── linear_relu.py │ │ └── quantized │ │ │ ├── __init__.py │ │ │ ├── dynamic │ │ │ ├── __init__.py │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ └── linear_relu.py │ │ │ └── modules │ │ │ ├── __init__.py │ │ │ ├── bn_relu.py │ │ │ ├── conv_relu.py │ │ │ └── linear_relu.py │ ├── modules │ │ ├── __init__.py │ │ ├── _functions.py │ │ ├── activation.py │ │ ├── adaptive.py │ │ ├── batchnorm.py │ │ ├── channelshuffle.py │ │ ├── container.py │ │ ├── conv.py │ │ ├── distance.py │ │ ├── dropout.py │ │ ├── flatten.py │ │ ├── fold.py │ │ ├── instancenorm.py │ │ ├── lazy.py │ │ ├── linear.py │ │ ├── loss.py │ │ ├── module.py │ │ ├── normalization.py │ │ ├── padding.py │ │ ├── pixelshuffle.py │ │ ├── pooling.py │ │ ├── rnn.py │ │ ├── sparse.py │ │ ├── transformer.py │ │ ├── upsampling.py │ │ └── utils.py │ ├── parallel │ │ ├── __init__.py │ │ ├── _functions.py │ │ ├── comm.py │ │ ├── data_parallel.py │ │ ├── distributed.py │ │ ├── parallel_apply.py │ │ ├── replicate.py │ │ └── scatter_gather.py │ ├── parameter.py │ ├── parameter.pyi │ ├── qat │ │ ├── __init__.py │ │ ├── dynamic │ │ │ ├── __init__.py │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ └── linear.py │ │ └── modules │ │ │ ├── __init__.py │ │ │ ├── conv.py │ │ │ ├── embedding_ops.py │ │ │ └── linear.py │ ├── quantizable │ │ ├── __init__.py │ │ └── modules │ │ │ ├── __init__.py │ │ │ ├── activation.py │ │ │ └── rnn.py │ ├── quantized │ │ ├── __init__.py │ │ ├── _reference │ │ │ ├── __init__.py │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── conv.py │ │ │ │ ├── linear.py │ │ │ │ ├── rnn.py │ │ │ │ ├── sparse.py │ │ │ │ └── utils.py │ │ ├── dynamic │ │ │ ├── __init__.py │ │ │ └── modules │ │ │ │ ├── __init__.py │ │ │ │ ├── conv.py │ │ │ │ ├── linear.py │ │ │ │ └── rnn.py │ │ ├── functional.py │ │ └── modules │ │ │ ├── __init__.py │ │ │ ├── activation.py │ │ │ ├── batchnorm.py │ │ │ ├── conv.py │ │ │ ├── dropout.py │ │ │ ├── embedding_ops.py │ │ │ ├── functional_modules.py │ │ │ ├── linear.py │ │ │ ├── normalization.py │ │ │ ├── rnn.py │ │ │ └── utils.py │ └── utils │ │ ├── __init__.py │ │ ├── _deprecation_utils.py │ │ ├── _expanded_weights │ │ ├── __init__.py │ │ ├── conv_expanded_weights.py │ │ ├── conv_utils.py │ │ ├── embedding_expanded_weights.py │ │ ├── expanded_weights_impl.py │ │ ├── expanded_weights_utils.py │ │ ├── group_norm_expanded_weights.py │ │ ├── instance_norm_expanded_weights.py │ │ ├── layer_norm_expanded_weights.py │ │ └── linear_expanded_weights.py │ │ ├── _named_member_accessor.py │ │ ├── _per_sample_grad.py │ │ ├── clip_grad.py │ │ ├── convert_parameters.py │ │ ├── fusion.py │ │ ├── init.py │ │ ├── memory_format.py │ │ ├── parametrizations.py │ │ ├── parametrize.py │ │ ├── prune.py │ │ ├── rnn.py │ │ ├── spectral_norm.py │ │ ├── stateless.py │ │ └── weight_norm.py ├── onnx │ ├── README.md │ ├── __init__.py │ ├── _constants.py │ ├── _experimental.py │ ├── _flags.py │ ├── _globals.py │ ├── _internal │ │ ├── __init__.py │ │ ├── _exporter_legacy.py │ │ ├── _lazy_import.py │ │ ├── diagnostics │ │ │ ├── OVERVIEW.md │ │ │ ├── __init__.py │ │ │ ├── _diagnostic.py │ │ │ ├── _rules.py │ │ │ ├── infra │ │ │ │ ├── __init__.py │ │ │ │ ├── _infra.py │ │ │ │ ├── context.py │ │ │ │ ├── decorator.py │ │ │ │ ├── formatter.py │ │ │ │ ├── sarif │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── _address.py │ │ │ │ │ ├── _artifact.py │ │ │ │ │ ├── _artifact_change.py │ │ │ │ │ ├── _artifact_content.py │ │ │ │ │ ├── _artifact_location.py │ │ │ │ │ ├── _attachment.py │ │ │ │ │ ├── _code_flow.py │ │ │ │ │ ├── _configuration_override.py │ │ │ │ │ ├── _conversion.py │ │ │ │ │ ├── _edge.py │ │ │ │ │ ├── _edge_traversal.py │ │ │ │ │ ├── _exception.py │ │ │ │ │ ├── _external_properties.py │ │ │ │ │ ├── _external_property_file_reference.py │ │ │ │ │ ├── _external_property_file_references.py │ │ │ │ │ ├── _fix.py │ │ │ │ │ ├── _graph.py │ │ │ │ │ ├── _graph_traversal.py │ │ │ │ │ ├── _invocation.py │ │ │ │ │ ├── _location.py │ │ │ │ │ ├── _location_relationship.py │ │ │ │ │ ├── _logical_location.py │ │ │ │ │ ├── _message.py │ │ │ │ │ ├── _multiformat_message_string.py │ │ │ │ │ ├── _node.py │ │ │ │ │ ├── _notification.py │ │ │ │ │ ├── _physical_location.py │ │ │ │ │ ├── _property_bag.py │ │ │ │ │ ├── _rectangle.py │ │ │ │ │ ├── _region.py │ │ │ │ │ ├── _replacement.py │ │ │ │ │ ├── _reporting_configuration.py │ │ │ │ │ ├── _reporting_descriptor.py │ │ │ │ │ ├── _reporting_descriptor_reference.py │ │ │ │ │ ├── _reporting_descriptor_relationship.py │ │ │ │ │ ├── _result.py │ │ │ │ │ ├── _result_provenance.py │ │ │ │ │ ├── _run.py │ │ │ │ │ ├── _run_automation_details.py │ │ │ │ │ ├── _sarif_log.py │ │ │ │ │ ├── _special_locations.py │ │ │ │ │ ├── _stack.py │ │ │ │ │ ├── _stack_frame.py │ │ │ │ │ ├── _suppression.py │ │ │ │ │ ├── _thread_flow.py │ │ │ │ │ ├── _thread_flow_location.py │ │ │ │ │ ├── _tool.py │ │ │ │ │ ├── _tool_component.py │ │ │ │ │ ├── _tool_component_reference.py │ │ │ │ │ ├── _translation_metadata.py │ │ │ │ │ ├── _version_control_details.py │ │ │ │ │ ├── _web_request.py │ │ │ │ │ ├── _web_response.py │ │ │ │ │ └── version.py │ │ │ │ └── utils.py │ │ │ └── rules.yaml │ │ ├── exporter │ │ │ ├── __init__.py │ │ │ ├── _analysis.py │ │ │ ├── _building.py │ │ │ ├── _capture_strategies.py │ │ │ ├── _compat.py │ │ │ ├── _constants.py │ │ │ ├── _core.py │ │ │ ├── _decomp.py │ │ │ ├── _dispatching.py │ │ │ ├── _dynamic_shapes.py │ │ │ ├── _errors.py │ │ │ ├── _fx_passes.py │ │ │ ├── _ir_passes.py │ │ │ ├── _isolated.py │ │ │ ├── _onnx_program.py │ │ │ ├── _registration.py │ │ │ ├── _reporting.py │ │ │ ├── _schemas.py │ │ │ ├── _tensors.py │ │ │ ├── _testing.py │ │ │ ├── _torchlib │ │ │ │ ├── __init__.py │ │ │ │ ├── _tensor_typing.py │ │ │ │ ├── _torchlib_registry.py │ │ │ │ └── ops │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── core.py │ │ │ │ │ └── hop.py │ │ │ └── _verification.py │ │ ├── fx │ │ │ ├── __init__.py │ │ │ ├── _pass.py │ │ │ ├── analysis │ │ │ │ ├── __init__.py │ │ │ │ └── unsupported_nodes.py │ │ │ ├── decomposition_skip.py │ │ │ ├── decomposition_table.py │ │ │ ├── diagnostics.py │ │ │ ├── dynamo_graph_extractor.py │ │ │ ├── fx_onnx_interpreter.py │ │ │ ├── fx_symbolic_graph_extractor.py │ │ │ ├── onnxfunction_dispatcher.py │ │ │ ├── passes │ │ │ │ ├── __init__.py │ │ │ │ ├── _utils.py │ │ │ │ ├── decomp.py │ │ │ │ ├── functionalization.py │ │ │ │ ├── modularization.py │ │ │ │ ├── readability.py │ │ │ │ ├── type_promotion.py │ │ │ │ └── virtualization.py │ │ │ ├── patcher.py │ │ │ ├── registration.py │ │ │ ├── serialization.py │ │ │ └── type_utils.py │ │ ├── io_adapter.py │ │ ├── jit_utils.py │ │ ├── onnx_proto_utils.py │ │ ├── onnxruntime.py │ │ └── registration.py │ ├── _onnx_supported_ops.py │ ├── _type_utils.py │ ├── errors.py │ ├── operators.py │ ├── symbolic_caffe2.py │ ├── symbolic_helper.py │ ├── symbolic_opset10.py │ ├── symbolic_opset11.py │ ├── symbolic_opset12.py │ ├── symbolic_opset13.py │ ├── symbolic_opset14.py │ ├── symbolic_opset15.py │ ├── symbolic_opset16.py │ ├── symbolic_opset17.py │ ├── symbolic_opset18.py │ ├── symbolic_opset19.py │ ├── symbolic_opset20.py │ ├── symbolic_opset7.py │ ├── symbolic_opset8.py │ ├── symbolic_opset9.py │ ├── utils.py │ └── verification.py ├── optim │ ├── __init__.py │ ├── _adafactor.py │ ├── _functional.py │ ├── _multi_tensor │ │ ├── __init__.py │ │ └── __init__.pyi │ ├── adadelta.py │ ├── adagrad.py │ ├── adam.py │ ├── adamax.py │ ├── adamw.py │ ├── asgd.py │ ├── lbfgs.py │ ├── lr_scheduler.py │ ├── nadam.py │ ├── optimizer.py │ ├── radam.py │ ├── rmsprop.py │ ├── rprop.py │ ├── sgd.py │ ├── sparse_adam.py │ └── swa_utils.py ├── overrides.py ├── package │ ├── __init__.py │ ├── _digraph.py │ ├── _directory_reader.py │ ├── _importlib.py │ ├── _mangling.py │ ├── _mock.py │ ├── _package_pickler.py │ ├── _package_unpickler.py │ ├── _stdlib.py │ ├── analyze │ │ ├── __init__.py │ │ ├── find_first_use_of_broken_modules.py │ │ ├── is_from_package.py │ │ └── trace_dependencies.py │ ├── file_structure_representation.py │ ├── find_file_dependencies.py │ ├── glob_group.py │ ├── importer.py │ ├── mangling.md │ ├── package_exporter.py │ └── package_importer.py ├── profiler │ ├── __init__.py │ ├── _memory_profiler.py │ ├── _pattern_matcher.py │ ├── _utils.py │ ├── itt.py │ ├── profiler.py │ └── python_tracer.py ├── py.typed ├── quantization │ ├── __init__.py │ ├── _numeric_suite.py │ ├── _numeric_suite_fx.py │ ├── _quantized_conversions.py │ ├── fake_quantize.py │ ├── fuse_modules.py │ ├── fuser_method_mappings.py │ ├── fx │ │ ├── __init__.py │ │ ├── _equalize.py │ │ ├── convert.py │ │ ├── fuse.py │ │ ├── fusion_patterns.py │ │ ├── graph_module.py │ │ ├── match_utils.py │ │ ├── pattern_utils.py │ │ ├── prepare.py │ │ ├── quantization_patterns.py │ │ ├── quantization_types.py │ │ └── utils.py │ ├── observer.py │ ├── qconfig.py │ ├── quant_type.py │ ├── quantization_mappings.py │ ├── quantize.py │ ├── quantize_fx.py │ ├── quantize_jit.py │ ├── stubs.py │ └── utils.py ├── quasirandom.py ├── random.py ├── return_types.py ├── script.h ├── serialization.py ├── signal │ ├── __init__.py │ └── windows │ │ ├── __init__.py │ │ └── windows.py ├── sparse │ ├── __init__.py │ ├── _semi_structured_conversions.py │ ├── _semi_structured_ops.py │ ├── _triton_ops.py │ ├── _triton_ops_meta.py │ └── semi_structured.py ├── special │ └── __init__.py ├── storage.py ├── testing │ ├── __init__.py │ ├── _comparison.py │ ├── _creation.py │ ├── _internal │ │ ├── __init__.py │ │ ├── autocast_test_lists.py │ │ ├── autograd_function_db.py │ │ ├── check_kernel_launches.py │ │ ├── codegen │ │ │ └── __init__.py │ │ ├── common_cuda.py │ │ ├── common_device_type.py │ │ ├── common_dist_composable.py │ │ ├── common_distributed.py │ │ ├── common_dtype.py │ │ ├── common_fsdp.py │ │ ├── common_jit.py │ │ ├── common_methods_invocations.py │ │ ├── common_mkldnn.py │ │ ├── common_modules.py │ │ ├── common_nn.py │ │ ├── common_optimizers.py │ │ ├── common_pruning.py │ │ ├── common_quantization.py │ │ ├── common_quantized.py │ │ ├── common_subclass.py │ │ ├── common_utils.py │ │ ├── composite_compliance.py │ │ ├── custom_op_db.py │ │ ├── custom_tensor.py │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── network1.py │ │ │ └── network2.py │ │ ├── dist_utils.py │ │ ├── distributed │ │ │ ├── __init__.py │ │ │ ├── _shard │ │ │ │ ├── __init__.py │ │ │ │ ├── sharded_tensor │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── _test_ops_common.py │ │ │ │ │ └── _test_st_common.py │ │ │ │ └── test_common.py │ │ │ ├── _tensor │ │ │ │ ├── __init__.py │ │ │ │ └── common_dtensor.py │ │ │ ├── checkpoint_utils.py │ │ │ ├── common_state_dict.py │ │ │ ├── ddp_under_dist_autograd_test.py │ │ │ ├── distributed_test.py │ │ │ ├── distributed_utils.py │ │ │ ├── fake_pg.py │ │ │ ├── multi_threaded_pg.py │ │ │ ├── nn │ │ │ │ ├── __init__.py │ │ │ │ └── api │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── remote_module_test.py │ │ │ ├── rpc │ │ │ │ ├── __init__.py │ │ │ │ ├── dist_autograd_test.py │ │ │ │ ├── dist_optimizer_test.py │ │ │ │ ├── examples │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── parameter_server_test.py │ │ │ │ │ └── reinforcement_learning_rpc_test.py │ │ │ │ ├── faulty_agent_rpc_test.py │ │ │ │ ├── faulty_rpc_agent_test_fixture.py │ │ │ │ ├── jit │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── dist_autograd_test.py │ │ │ │ │ ├── rpc_test.py │ │ │ │ │ └── rpc_test_faulty.py │ │ │ │ ├── rpc_agent_test_fixture.py │ │ │ │ ├── rpc_test.py │ │ │ │ └── tensorpipe_rpc_agent_test_fixture.py │ │ │ └── rpc_utils.py │ │ ├── dynamo_test_failures.py │ │ ├── fake_config_module.py │ │ ├── fake_config_module2.py │ │ ├── fake_config_module3.py │ │ ├── generated │ │ │ └── __init__.py │ │ ├── hop_db.py │ │ ├── hypothesis_utils.py │ │ ├── inductor_utils.py │ │ ├── jit_metaprogramming_utils.py │ │ ├── jit_utils.py │ │ ├── logging_tensor.py │ │ ├── logging_utils.py │ │ ├── opinfo │ │ │ ├── __init__.py │ │ │ ├── core.py │ │ │ ├── definitions │ │ │ │ ├── __init__.py │ │ │ │ ├── _masked.py │ │ │ │ ├── fft.py │ │ │ │ ├── linalg.py │ │ │ │ ├── nested.py │ │ │ │ ├── signal.py │ │ │ │ ├── sparse.py │ │ │ │ └── special.py │ │ │ ├── refs.py │ │ │ └── utils.py │ │ ├── optests │ │ │ ├── __init__.py │ │ │ ├── aot_autograd.py │ │ │ ├── autograd_registration.py │ │ │ ├── fake_tensor.py │ │ │ ├── generate_tests.py │ │ │ └── make_fx.py │ │ ├── quantization_torch_package_models.py │ │ ├── static_module.py │ │ ├── subclasses.py │ │ ├── test_module │ │ │ ├── __init__.py │ │ │ ├── future_div.py │ │ │ └── no_future_div.py │ │ ├── torchbind_impls.py │ │ ├── triton_utils.py │ │ └── two_tensor.py │ └── _utils.py ├── torch_version.py ├── types.py ├── utils │ ├── __init__.py │ ├── _appending_byte_serializer.py │ ├── _backport_slots.py │ ├── _config_module.py │ ├── _config_typing.pyi │ ├── _content_store.py │ ├── _contextlib.py │ ├── _cpp_embed_headers.py │ ├── _cpp_extension_versioner.py │ ├── _cxx_pytree.py │ ├── _device.py │ ├── _exposed_in.py │ ├── _filelock.py │ ├── _foreach_utils.py │ ├── _freeze.py │ ├── _functools.py │ ├── _get_clean_triton.py │ ├── _import_utils.py │ ├── _mode_utils.py │ ├── _ordered_set.py │ ├── _python_dispatch.py │ ├── _pytree.py │ ├── _stats.py │ ├── _strobelight │ │ ├── __init__.py │ │ ├── cli_function_profiler.py │ │ └── examples │ │ │ └── cli_function_profiler_example.py │ ├── _sympy │ │ ├── __init__.py │ │ ├── functions.py │ │ ├── interp.py │ │ ├── numbers.py │ │ ├── printers.py │ │ ├── reference.py │ │ ├── singleton_int.py │ │ ├── solve.py │ │ ├── symbol.py │ │ └── value_ranges.py │ ├── _thunk.py │ ├── _traceback.py │ ├── _triton.py │ ├── _typing_utils.py │ ├── _zip.py │ ├── backcompat │ │ └── __init__.py │ ├── backend_registration.py │ ├── benchmark │ │ ├── README.md │ │ ├── __init__.py │ │ ├── examples │ │ │ ├── __init__.py │ │ │ ├── blas_compare_setup.py │ │ │ ├── compare.py │ │ │ ├── fuzzer.py │ │ │ ├── op_benchmark.py │ │ │ ├── prepare_e2e.sh │ │ │ ├── simple_timeit.py │ │ │ ├── sparse │ │ │ │ ├── compare.py │ │ │ │ ├── fuzzer.py │ │ │ │ └── op_benchmark.py │ │ │ └── spectral_ops_fuzz_test.py │ │ ├── op_fuzzers │ │ │ ├── __init__.py │ │ │ ├── binary.py │ │ │ ├── sparse_binary.py │ │ │ ├── sparse_unary.py │ │ │ ├── spectral.py │ │ │ └── unary.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── _stubs.py │ │ │ ├── common.py │ │ │ ├── compare.py │ │ │ ├── compile.py │ │ │ ├── cpp_jit.py │ │ │ ├── fuzzer.py │ │ │ ├── sparse_fuzzer.py │ │ │ ├── timeit_template.cpp │ │ │ ├── timer.py │ │ │ └── valgrind_wrapper │ │ │ ├── __init__.py │ │ │ ├── compat_bindings.cpp │ │ │ ├── timer_callgrind_template.cpp │ │ │ └── timer_interface.py │ ├── bottleneck │ │ ├── __init__.py │ │ └── __main__.py │ ├── bundled_inputs.py │ ├── checkpoint.py │ ├── collect_env.py │ ├── cpp_backtrace.py │ ├── cpp_extension.py │ ├── data │ │ ├── __init__.py │ │ ├── _utils │ │ │ ├── __init__.py │ │ │ ├── collate.py │ │ │ ├── fetch.py │ │ │ ├── pin_memory.py │ │ │ ├── signal_handling.py │ │ │ └── worker.py │ │ ├── backward_compatibility.py │ │ ├── dataframes_pipes.ipynb │ │ ├── dataloader.py │ │ ├── datapipes │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── _decorator.py │ │ │ ├── _hook_iterator.py │ │ │ ├── _typing.py │ │ │ ├── dataframe │ │ │ │ ├── __init__.py │ │ │ │ ├── dataframe_wrapper.py │ │ │ │ ├── dataframes.py │ │ │ │ ├── datapipes.py │ │ │ │ └── structures.py │ │ │ ├── datapipe.py │ │ │ ├── datapipe.pyi.in │ │ │ ├── gen_pyi.py │ │ │ ├── iter │ │ │ │ ├── __init__.py │ │ │ │ ├── callable.py │ │ │ │ ├── combinatorics.py │ │ │ │ ├── combining.py │ │ │ │ ├── filelister.py │ │ │ │ ├── fileopener.py │ │ │ │ ├── grouping.py │ │ │ │ ├── routeddecoder.py │ │ │ │ ├── selecting.py │ │ │ │ ├── sharding.py │ │ │ │ ├── streamreader.py │ │ │ │ └── utils.py │ │ │ ├── map │ │ │ │ ├── __init__.py │ │ │ │ ├── callable.py │ │ │ │ ├── combinatorics.py │ │ │ │ ├── combining.py │ │ │ │ ├── grouping.py │ │ │ │ └── utils.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── common.py │ │ │ │ ├── decoder.py │ │ │ │ └── snapshot.py │ │ ├── dataset.py │ │ ├── distributed.py │ │ ├── graph.py │ │ ├── graph_settings.py │ │ ├── sampler.py │ │ ├── standard_pipes.ipynb │ │ └── typing.ipynb │ ├── deterministic.py │ ├── dlpack.py │ ├── file_baton.py │ ├── flop_counter.py │ ├── hipify │ │ ├── __init__.py │ │ ├── constants.py │ │ ├── cuda_to_hip_mappings.py │ │ ├── hipify_python.py │ │ └── version.py │ ├── hooks.py │ ├── jit │ │ ├── __init__.py │ │ └── log_extract.py │ ├── mkldnn.py │ ├── mobile_optimizer.py │ ├── model_dump │ │ ├── __init__.py │ │ ├── __main__.py │ │ ├── code.js │ │ ├── htm.mjs │ │ ├── preact.mjs │ │ └── skeleton.html │ ├── model_zoo.py │ ├── module_tracker.py │ ├── serialization │ │ ├── __init__.py │ │ └── config.py │ ├── show_pickle.py │ ├── tensorboard │ │ ├── __init__.py │ │ ├── _convert_np.py │ │ ├── _embedding.py │ │ ├── _onnx_graph.py │ │ ├── _proto_graph.py │ │ ├── _pytorch_graph.py │ │ ├── _utils.py │ │ ├── summary.py │ │ └── writer.py │ ├── throughput_benchmark.py │ ├── viz │ │ ├── MemoryViz.js │ │ ├── __init__.py │ │ └── _cycles.py │ └── weak.py ├── version.py.tpl └── xpu │ ├── __init__.py │ ├── _gpu_trace.py │ ├── _utils.py │ ├── memory.py │ ├── random.py │ └── streams.py ├── torchgen ├── BUCK.oss ├── BUILD.bazel ├── __init__.py ├── _autoheuristic │ ├── README.md │ ├── ah_tree.py │ ├── benchmark_runner.py │ ├── benchmark_utils.py │ ├── collect_data.sh │ ├── generate_heuristic.sh │ ├── merge_data.py │ ├── mixed_mm │ │ ├── README.md │ │ ├── gen_data_mixed_mm.py │ │ ├── gen_mixedmm_heuristic_a100.sh │ │ ├── gen_mixedmm_heuristic_h100.sh │ │ ├── generate_heuristic_mixedmm.sh │ │ ├── get_mixedmm_dataset.sh │ │ ├── test_mixed_mm.py │ │ └── train_decision_mixedmm.py │ ├── mm │ │ ├── README.md │ │ ├── gen_data_mm.py │ │ ├── gen_heuristic_a100.sh │ │ ├── gen_heuristic_h100.sh │ │ ├── get_mm_dataset.sh │ │ └── train_decision_mm.py │ ├── pad_mm │ │ ├── README.md │ │ ├── gen_data_pad_mm.py │ │ ├── gen_pad_mm_a100.sh │ │ ├── gen_pad_mm_h100.sh │ │ ├── generate_heuristic_pad_mm.sh │ │ ├── get_padmm_dataset.sh │ │ ├── test_pad_mm.py │ │ ├── train_decision_pad_mm.py │ │ ├── train_pad_mm.py │ │ └── train_regression_pad_mm.py │ ├── requirements.txt │ ├── test.sh │ ├── test_utils.py │ ├── train.py │ ├── train_decision.py │ └── train_regression.py ├── aoti │ ├── __init__.py │ └── fallback_ops.py ├── api │ ├── __init__.py │ ├── autograd.py │ ├── cpp.py │ ├── dispatcher.py │ ├── functionalization.py │ ├── lazy.py │ ├── meta.py │ ├── native.py │ ├── python.py │ ├── structured.py │ ├── translate.py │ ├── types │ │ ├── __init__.py │ │ ├── signatures.py │ │ ├── types.py │ │ └── types_base.py │ ├── ufunc.py │ └── unboxing.py ├── build.bzl ├── code_template.py ├── context.py ├── decompositions │ └── gen_jit_decompositions.py ├── dest │ ├── __init__.py │ ├── lazy_ir.py │ ├── lazy_ts_lowering.py │ ├── native_functions.py │ ├── register_dispatch_key.py │ └── ufunc.py ├── executorch │ ├── __init__.py │ ├── api │ │ ├── __init__.py │ │ ├── custom_ops.py │ │ ├── et_cpp.py │ │ ├── types │ │ │ ├── __init__.py │ │ │ ├── signatures.py │ │ │ └── types.py │ │ └── unboxing.py │ ├── model.py │ └── parse.py ├── fuse │ └── gen_patterns.py ├── gen.py ├── gen_aoti_c_shim.py ├── gen_backend_stubs.py ├── gen_executorch.py ├── gen_functionalization_type.py ├── gen_lazy_tensor.py ├── gen_schema_utils.py ├── gen_vmap_plumbing.py ├── local.py ├── model.py ├── native_function_generation.py ├── operator_versions │ ├── __init__.py │ ├── gen_mobile_upgraders.py │ └── gen_mobile_upgraders_constant.py ├── selective_build │ ├── __init__.py │ ├── operator.py │ └── selector.py ├── shape_functions │ └── gen_jit_shape_functions.py ├── static_runtime │ ├── __init__.py │ ├── config.py │ ├── gen_static_runtime_ops.py │ └── generator.py ├── utils.py └── yaml_utils.py ├── ubsan.supp ├── ufunc_defs.bzl └── version.txt /.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/.bazelrc -------------------------------------------------------------------------------- /.bazelversion: -------------------------------------------------------------------------------- 1 | 6.5.0 2 | -------------------------------------------------------------------------------- /.ci/docker/ci_commit_pins/nccl-cu11.txt: -------------------------------------------------------------------------------- 1 | v2.21.5-1 2 | -------------------------------------------------------------------------------- /.ci/docker/ci_commit_pins/nccl-cu12.txt: -------------------------------------------------------------------------------- 1 | v2.25.1-1 2 | -------------------------------------------------------------------------------- /.ci/docker/triton_version.txt: -------------------------------------------------------------------------------- 1 | 3.2.0 2 | -------------------------------------------------------------------------------- /.ci/docker/ubuntu-rocm/.gitignore: -------------------------------------------------------------------------------- 1 | *.sh 2 | -------------------------------------------------------------------------------- /.ci/magma/.gitignore: -------------------------------------------------------------------------------- 1 | output/ 2 | magma-cuda*/ 3 | -------------------------------------------------------------------------------- /.circleci/.gitignore: -------------------------------------------------------------------------------- 1 | *.svg 2 | *.png 3 | -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/.flake8 -------------------------------------------------------------------------------- /.gdbinit: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/.gdbinit -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- 1 | Fixes #ISSUE_NUMBER 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/.gitignore -------------------------------------------------------------------------------- /.lldbinit: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/.lldbinit -------------------------------------------------------------------------------- /BUCK.oss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/BUCK.oss -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/Makefile -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/README.md -------------------------------------------------------------------------------- /RELEASE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/RELEASE.md -------------------------------------------------------------------------------- /WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/WORKSPACE -------------------------------------------------------------------------------- /aten.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/aten.bzl -------------------------------------------------------------------------------- /aten/src/ATen/.gitignore: -------------------------------------------------------------------------------- 1 | Config.h 2 | -------------------------------------------------------------------------------- /aten/src/ATen/Device.h: -------------------------------------------------------------------------------- 1 | #pragma once 2 | #include 3 | -------------------------------------------------------------------------------- /aten/src/ATen/Dimname.h: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /aten/src/ATen/Formatting.h: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /aten/src/ATen/Layout.h: -------------------------------------------------------------------------------- 1 | #pragma once 2 | #include 3 | -------------------------------------------------------------------------------- /aten/src/ATen/NamedTensor.h: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /aten/src/ATen/core/Scalar.h: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /aten/src/ATen/core/ScalarType.h: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /aten/src/ATen/core/blob.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /aten/src/ATen/core/typeid.h: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /benchmarks/dynamo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/dynamo/microbenchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/dynamo/pr_time_benchmarks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/instruction_counts/applications/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/instruction_counts/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/instruction_counts/definitions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/instruction_counts/execution/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/instruction_counts/worker/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/operator_benchmark/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/operator_benchmark/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/operator_benchmark/pt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmarks/sparse/__init__.py: -------------------------------------------------------------------------------- 1 | if __name__ == "__main__": 2 | pass 3 | -------------------------------------------------------------------------------- /c10/util/C++17.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /c10/util/LeftRight.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /c10/util/Metaprogramming.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /c10/util/Optional.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /c10/util/TypeList.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /c10/util/TypeTraits.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /c10/util/intrusive_ptr.cpp: -------------------------------------------------------------------------------- 1 | #include 2 | -------------------------------------------------------------------------------- /caffe2/unexported_symbols.lds: -------------------------------------------------------------------------------- 1 | *4llvm* 2 | -------------------------------------------------------------------------------- /cmake/IncludeSource.cpp.in: -------------------------------------------------------------------------------- 1 | #include "@IMPL@" 2 | -------------------------------------------------------------------------------- /defs.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/defs.bzl -------------------------------------------------------------------------------- /example/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /example/bootstrap/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /example/resnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /functorch/_src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /functorch/examples/.gitignore: -------------------------------------------------------------------------------- 1 | cifar10/ 2 | -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/mypy.ini -------------------------------------------------------------------------------- /pt_ops.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/pt_ops.bzl -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/pytest.ini -------------------------------------------------------------------------------- /scripts/release_notes/.gitignore: -------------------------------------------------------------------------------- 1 | results/* 2 | -------------------------------------------------------------------------------- /scripts/release_notes/requirements.txt: -------------------------------------------------------------------------------- 1 | PyGithub 2 | tqdm -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jizhuoran/EasyFHE/HEAD/setup.py -------------------------------------------------------------------------------- /test/cpp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/cpp/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/cpp_api_parity/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/delete.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/distributed/nn/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/distributed/pipelining/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/distributed/tensor/__init__.py: -------------------------------------------------------------------------------- 1 | # shut up pylint 2 | -------------------------------------------------------------------------------- /test/distributed/tensor/parallel/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo/mock_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/AutogradFunctionTests.test_print_in_bwd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/AutogradFunctionTests.test_stride_in_bwd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ComptimeTests.test_graph_break: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/DefaultsTests.test_cast_tensor_single_elem: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/DefaultsTests.test_func_default_torch_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/DefaultsTests.test_in_set_inplace: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/End2EndTests.test_init_group: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExcTests.test_backend_suppress_line: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExcTests.test_graph_break_log: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExcTests.test_internal_error_no_suppress: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExcTests.test_not_implemented_error: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExcTests.test_trigger_on_error: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExportTests.test_export_decomp: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExportTests.test_export_dynamic_dim_cleanup: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExportTests.test_export_no_raise: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExportTests.test_export_with_map_cond: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ExportTests.test_torch_inference_mode_ctx: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/FuncTorchHigherOrderOpTests.test_grad: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/FuncTorchHigherOrderOpTests.test_vmap: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/FunctionTests.test_default_dict: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/FunctionTests.test_default_dict_closure: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/FunctionTests.test_default_dict_lambda: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/FunctionTests.test_math_radians: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HigherOrderOpTests.test_map_lowers_to_graph: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HigherOrderOpTests.test_map_multi_return: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HigherOrderOpTests.test_map_pytree_return: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HigherOrderOpTests.test_map_source_fn_stack: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HigherOrderOpTests.test_map_symint_input: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HigherOrderOpTests.test_modules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HooksTests.test_functools_arg_vary: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/HooksTests.test_post_acc_grad_hook: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/InteropTests.test_vmap_in_graph: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_aot: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_aot_graphs: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_aot_joint_graph: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_bytecode: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_custom_format: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_custom_format_exc: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_dynamo_debug: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_dynamo_error: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_dynamo_info: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_graph: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_graph_breaks: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_graph_code: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_graph_sizes: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_graph_sizes_dynamic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_guards_recompiles: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_inductor_debug: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_inductor_error: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_inductor_info: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_multiline_format: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_output_code: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_recompiles: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_trace_source_cond: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/LoggingTests.test_trace_source_nested: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_add_to_set: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_any_all_symnode: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_backend_match_guard: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_boolarg: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_callpacked: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_cond_nested: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_cond_side_effects: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_dataclass_fields: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_deque_append_left: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_deque_input: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_dict_order_keys: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_dict_order_keys_modules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_dict_order_keys_tensors: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_dictcomp: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_dtypes_no_graphbreaks: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_get_cache_entry: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_guard_failure_fn: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_guard_failure_fn2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_inference_mode: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_inplace_param_update: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_is_compiling: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_iter_set: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_itertools_repeat: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_listcomp: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_mandelbrot_numpy: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_mark_static: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_namedtuple1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_namedtuple2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_nan: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_nested_optimize: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_nested_optimize_run: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_nn_module_getattr: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_array_of_arrays: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_force: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_int_constant: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_iter: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_non_torch_dtype: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_readonly: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_size_attr: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_subdtype: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_tolist: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_torch_operators: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_numpy_with_builtin_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_pure_python_accumulate: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_py_guards_mark_dynamic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_raise_on_backend_error: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_release_input_memory: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_release_module_memory: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_release_scope_memory: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_return_nested_function: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_set_aliasing_recompiles: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_setattr_mutation1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_simple_set_usage: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_size_input: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_str_format_assert2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tensor_dict1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tensor_dict2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tensor_dict3: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tensor_item_capture: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tolist_0d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tolist_1d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tolist_kd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tolist_kd_dynamic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tolist_scalar: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_torch_seed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tracing_nested_py_tree: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tracing_py_tree: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_tracing_tree_map_only: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_type_copy: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_typing_typevar: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_user_getattribute: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_with_builtin_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_yield_from: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/MiscTests.test_yield_gen_and_from: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_lazy_module1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_lazy_module2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_lazy_module4: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_lazy_module5: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_lazy_module6: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_self_mutating1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_unsupportedmethod: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NNModuleTests.test_unsupportedmodule: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/NumpyTestsCPU.test_index_no_floats_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizedModuleTest.test_hooks_inner: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizedModuleTest.test_hooks_outer: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizedModuleTest.test_nn_module: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizedModuleTest.test_to: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_adagrad: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_adam: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_adamax: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_adamw: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_asgd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_nadam: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_rmsprop: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_rprop: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/OptimizerTests.test_sgd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/PackedSequenceTest.test_pack_sequence: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/PackedSequenceTest.test_to: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/PackedSequenceTest.test_total_length: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/PackedSequenceTest.test_type_casts: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/RecompileUxTests.test_mismatched_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_add_sub_alpha_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_addr_alpha_beta_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_boxes_len: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_chunk_reformer_ff: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_do_paste_mask: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_function_in_skipfiles: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_functools_wraps: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_hf_t5_forward: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_hf_xsoftmax_training: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_issue175: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_jit_trace_errors: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_list_aliasing: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_list_self_reference: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_longformer_chunk: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_module_in_skipfiles: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_multi_import: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_negative_shape_guard: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_recursive_map: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_reformer_eval: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_reformer_min_chunk_len: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_reformer_sorting: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_relative_import: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_rewrite_assert_noop: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_seq_append_list: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_size_typematch: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_tensor_data_kwarg: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_threading_local: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/ReproTests.test_validate_model_kwargs: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/SkipNonTensorTests.test_add_tensor1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/SkipNonTensorTests.test_add_tensor2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/SkipNonTensorTests.test_recursive_list: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/SubGraphTests.test_dynamic_duck_size: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/SubGraphTests.test_dynamic_getitem: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/SubGraphTests.test_dynamic_kwarg: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAnalyze.test_trace_dependencies: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestApplyAlongAxis.test_0d_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestApplyAlongAxis.test_axis_insertion: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestApplyAlongAxis.test_scalar_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArange.test_infinite: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArange.test_nan_step: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArange.test_require_range: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArange.test_zero_step: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArgmax.test_combinations_data62: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArgmax.test_combinations_data63: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArgmax.test_combinations_data65: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArgmax.test_combinations_data66: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestArraySplit.test_integer_0_split: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAttributes.test_fill_max_uint64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAttributes.test_fill_readonly: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAttributes.test_fill_struct_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAutograd.test_backward_with_inputs: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAutograd.test_gradcheck_forward_ad: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAutograd.test_record_function: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAutograd.test_return_duplicate: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestAutograd.test_set_grad_coroutines: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestBaseMath.test_lower_align: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestBaseSparsifier.test_state_dict: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestBitsCPU.test_cat_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestBool.test_sum_2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestClip.test_clip_func_takes_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestClip.test_simple_complex: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestColumnStack.test_non_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestConcatenate.test_bad_out_shape: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestConcatenate.test_concatenate: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestConcatenate.test_exceptions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestCond.test_empty_sq_cases: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestCond.test_sq_cases: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestControlFlow.test_map_dict_in_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestControlFlow.test_map_list_in_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestConversion.test_to_int_scalar: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestCov.test_aweights: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestCov.test_fweights: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestCross.test_broadcasting_shapes: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestCustomOp.test_impl_device_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestCustomOp.test_legacy_define: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_dlpack_device: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_dunder_dlpack_refcount: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_from_dlpack_refcount: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_from_torch: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_10: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_11: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_12: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_13: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_14: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_15: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_17: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_18: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_19: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_20: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_21: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_22: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_23: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_24: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_25: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_26: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_27: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_28: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_29: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_3: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_30: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_31: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_4: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_5: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_6: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_7: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_higher_dims_ndim_9: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDLPack.test_ndim0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDelete.test_slices: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDiag.test_failure: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDiff.test_append: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDiff.test_axis: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDiff.test_n: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDiff.test_prepend: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDropoutNN.test_invalid_dropout_p: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDsplit.test_0D_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDsplit.test_1D_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDsplit.test_2D_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDsplit.test_non_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDstack.test_non_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestDynamicQuantizedOps.test_qlinear: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestEigh.test_invalid: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestEigvalsh.test_invalid: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestExpandDims.test_axis_out_of_range: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestExpandDims.test_repeated_axis: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestExport.test_nn_module_stack: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestFlag.test_writeable_from_readonly: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestFlip.test_axes: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestFliplr.test_basic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestFromBuffer.test_empty: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestGradNewOnesOverride.test_newones: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestGradient.test_badargs: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestGradient.test_specific_axes: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestGradient.test_values: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogram.test_error_binnum_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogram.test_finite_range: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogram.test_one_bin: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogramdd.test_bins_error_2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogramdd.test_bins_errors: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogramdd.test_equal_edges: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogramdd.test_inf_edges: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHistogramdd.test_weights: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHsplit.test_0D_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHsplit.test_non_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHstack.test_empty_input: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestHstack.test_non_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestIndexing.test_index_no_floats: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestIndexing.test_slicing_no_floats: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestIndexingCPU.test_byte_mask_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestIndexingCPU.test_index_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestIndexingCPU.test_index_limits_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestIsScalar.test_is_not_scalar_value6: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestIterator.test_iterator: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestJacCPU.test_aux_pytree_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestJacCPU.test_dimensionality_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestJacCPU.test_empty_output_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestJacCPU.test_inplace_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestJacCPU.test_unrelated_input_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestJacCPU.test_unrelated_output_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestJit.test_batchnorm: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLexsort.test_datetime: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLexsort.test_mixed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_geqrf_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_geqrf_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_geqrf_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_pinv_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_pinv_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_pinv_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_pinv_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_solve_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_solve_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestLinalgCPU.test_solve_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMatmul.test_exceptions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMedian.test_extended_axis_invalid: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMeshgrid.test_indexing: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_adapt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_attn: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_big_split: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_c: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_compare_dims: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_diag: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_dim_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_dims_with_size: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_dir: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_doc: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_embed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_eq: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_expand: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_functorch: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_hello: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_index: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_index_placement: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_inplace: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_manual_stuff: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_mask: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_max: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_mm: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_mm_fuse: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_monkey: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_network: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_order: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_order_keyword: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_parse: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_permute_orig: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_seg: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_simple: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_softmax_split: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_stack: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_time_mm_fuse: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMin.test_with_dims_split: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_adapt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_attn: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_big_split: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_c: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_compare_dims: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_diag: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_dim_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_dir: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_doc: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_embed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_eq: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_expand: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_functorch: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_hello: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_index: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_inplace: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_manual_stuff: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_mask: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_max: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_mm: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_mm_fuse: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_monkey: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_network: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_order: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_parse: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_permute_orig: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_seg: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_simple: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinFunctorchOnly.test_stack: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinMax.test_axis: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMinMax.test_scalar: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMisc.test_byteorder_check: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMkldnnCPU.test_add_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMkldnnCPU.test_copy_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMkldnnCPU.test_mul_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMkldnnCPU.test_sigmoid_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMkldnnCPU.test_tanh_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestModuleHookNN.test_hook_inplace: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMoveaxis.test_errors: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMultiDot.test_too_few_input_arrays: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMultiprocessing.test_empty_shared: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMultiprocessing.test_fs_is_shared: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestMultiprocessing.test_is_shared: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNN.test_fb_fc_packed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNamedTensor.test_flatten: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNamedTensor.test_reduction_fns: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNamedTensor.test_unflatten: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNamedTuple.test_max: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNegative.test_exceptions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNestedTensor.test_unbind: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormDouble.test_axis: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormDouble.test_bad_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormDouble.test_matrix_2x2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormInt64.test_axis: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormInt64.test_bad_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormInt64.test_matrix_2x2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormSingle.test_axis: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormSingle.test_bad_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestNormSingle.test_matrix_2x2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestObserver.test_per_tensor_observers: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestOptimizations.test_example_inputs: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPadding.test_constant_padNd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPercentile.test_scalar_q: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPickle.test_pickle: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestProfiler.test_flops: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestProfiler.test_high_level_trace: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestProfiler.test_is_profiler_enabled: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestProfiler.test_source: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPutAlongAxis.test_broadcast: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPutmask.test_record_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPythonAutograd.test_backwards2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPythonAutograd.test_forwards1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPythonAutograd.test_forwards2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPythonAutograd.test_split: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestPythonRegistration.test_finalizer: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestQR.test_mode_raw: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestQR.test_qr_empty_m_0_n_0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestQR.test_qr_empty_m_0_n_3: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestQR.test_qr_empty_m_3_n_0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestQuantizedOps.test_max_pool2d_cudnn: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestQuantizedOps.test_qtanh: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestQuantizedTensor.test_qtensor_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestRequire.test_C_and_F_simul: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestRequire.test_require_each: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestRequire.test_unknown_requirement: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestRollaxis.test_exceptions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestRot90.test_basic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestScript.test_conv_error: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestScript.test_infer_size: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestScript.test_is_scripting: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestScript.test_method_overloading: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestScript.test_parse_nested_names: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestScript.test_parse_tensor_constants: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestScript.test_python_call: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestSplit.test_unequal_split: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestStackMisc.test_stack: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestStats.test_dtype_from_dtype: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestStats.test_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTEFuserDynamic.test_to_dtype: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTEFuserDynamic.test_torch_to: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTEFuserStatic.test_to_dtype: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTEFuserStatic.test_torch_to: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTake.test_raise: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTakeAlongAxis.test_invalid: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTorch.test_map: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTorch.test_new: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTorch.test_parsing_intlist: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTorch.test_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTorch.test_upsample_nearest2d_meta: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTorchbind.test_torchbind_inductor: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTrilIndicesFrom.test_exceptions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTriuIndicesFrom.test_exceptions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestTypeHints.test_doc_examples: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestUnique.test_unique_axis: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestUnique.test_unique_axis_errors: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestVerifyCorrectness.test_torchscript: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestVsplit.test_0D_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestVsplit.test_1D_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestVsplit.test_non_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestVstack.test_empty_input: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestVstack.test_non_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/TestWhere.test_error: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_builtin_max_min: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_conv1d_symint_padding: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_isinstance_symint: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_mark_01_dynamic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_no_recompilations: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_no_recompiles: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_propagate_dynamic_dim: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_expected_failures/UnspecTests.test_use_and_specialize: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/DecoratorTests.test_allow_in_graph: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/DynamicShapesMiscTests.test_boolarg_dynamic_shapes: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/DynamicShapesMiscTests.test_nan_dynamic_shapes: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/ExportTests.test_predispatch_with_for_out_dtype: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/FrameInitTests.test_frame_init: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/HigherOrderOpTests.test_access_module_attr: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/InPlaceCompilationTests.test_compilation: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/InteropTests.test_fx_fn: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/LoggingTests.test_distributed_rank_logging: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/LoggingTests.test_logs_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/LoggingTests.test_trace_call: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/LoggingTests.test_trace_call_graph_break: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/LoggingTests.test_trace_call_inline_call: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_auto_functionalize_on_view: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_auto_functionalize_optional: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_exception_table_e2e: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_exception_table_e2e_2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_exception_table_parsing: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_generate_trivial_abstract_impl: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_itertools_accumulate_tensors_kwargs: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_itertools_infinite_count: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_linetable_311_writer1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_linetable_311_writer2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/MiscTests.test_py311_jump_offset: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/NopTests.test_extended_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/OptimizerTests.test_adadelta: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/RecompileTests.test_aliasing_guard_failures: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/RecompileUxTests.test_drop_cache_on_skip: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/ReproTests.test_optim_state_references_cleared: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/ReproTests.test_reformer_train: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestAOTAutograd.test_view_detach: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestArgmax.test_combinations_data58: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestArgmax.test_combinations_data61: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestAwait.test_await_python: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestBackends.test_execution: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestBackends.test_save_load: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestBackendsWithCompiler.test_execution: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestBasicsCPU.test_invalid_sparse_coo_values_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestBasicsCPU.test_invalid_sparse_csr_values_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestBinaryUfuncsCPU.test_add_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestBinaryUfuncsCPU.test_shift_limits_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestCheckpoint.test_checkpoint_trigger: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestConfigModule.test_env_name_semantics: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestConstant.test_check_constant: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestConvolutionNN.test_Conv1d_module_same_padding: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestConvolutionNN.test_Conv2d_backward_twice: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestConvolutionNN.test_Conv2d_module_same_padding: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestConvolutionNN.test_Conv3d_module_same_padding: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestCustomOp.test_impl_device_function: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestDLPack.test_non_contiguous: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestDtypeAnalysis.test_custom_rules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestDtypeAnalysis.test_unary: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestFFTShift.test_fft_n: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestFX.test_annotations_empty_tuple: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestFX.test_assert: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestFXExperimental.test_optimize_for_inference_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestFreezing.test_freeze_module_with_fork2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestFrozenOptimizations.test_conv_add_folding: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestFrozenOptimizations.test_conv_bn_folding: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestGenericProxyTensorFake.test_amp_cache: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestGenericProxyTensorFake.test_inplace_metadata: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestGenericProxyTensorReal.test_inplace_metadata: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestGenericProxyTensorSymbolic.test_amp_cache: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestHistogramdd.test_bins_array: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestIndexingCPU.test_invalid_index_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_dilated: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_groups: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_pad1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_pad1size1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_pad2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_pad2size1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_pad_same: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_pad_same2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_pad_valid: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_stride: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv1d_zero_batch: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_depthwise: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_dilated: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_groups: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_groups_thnn: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_pad_same: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_pad_valid: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_padding: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_strided: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv2d_zero_batch: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv3d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv3d_dilated: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv3d_groups: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv3d_pad_same: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv3d_pad_valid: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv3d_stride: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_Conv3d_zero_batch: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_ConvTranspose1d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_ConvTranspose2d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestJitGeneratedModule.test_nn_ConvTranspose3d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLazyDynamicOps.test_nonzero_dynamic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLazyReuseIr.testAdd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLazyReuseIr.testAddSub: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLazyReuseIr.testAddSubFallback: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLazyReuseIr.testBatchNorm: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLazyTensor.test_tensor_ctr: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLazyTensor.test_view_mark_step_preserved: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLinalgCPU.test_lobpcg_torchscript_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestList.test_comprehension_iterable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestLogging.test_trace_numeric_counter: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMKLDNNReinplacing.test_always_alive_values: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMatmulOperator.test_exceptions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMatmulOperator.test_matmul_raises: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_compress: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_dot_out_mem_overlap: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_matmul_out: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_partition_iterative: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_round: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_searchsorted_complex: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_searchsorted_type_specific_2: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_searchsorted_with_invalid_sorter: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMethods.test_transpose: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMisc.test_broadcasting_list: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMisc.test_parse_ir_annotate: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestMkldnnFusion.test_single_conv: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestModels.test_snli_quantized: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestModels.test_time_sequence_prediction: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestModels.test_vae_quantized: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNN.test_padding_list: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNN.test_vector_to_parameters: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_avg_pool2d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_cat: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_compile_spec_santiy: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_conv2d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_conv2d_transpose: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_dequantize: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_detach: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_flatten: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_hardtanh: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_linear: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_log_softmax: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_max_pool2d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_mean: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_multi_output: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_pointwise_binary: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_pointwise_binary_const: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_pointwise_unary: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_prelu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_qadd: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_qlinear: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_quantize: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_reshape: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_seblock_mul: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_slice: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_softmax: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_tensor_input: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_to: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_unsqueeze: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestNnapiBackend.test_upsample_nearest2d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestOpenMP_ParallelFor.test_one_thread: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestParametrization.test_traceable: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPeephole.test_peephole_int: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPeephole.test_peephole_optional_refine: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestProfiler.test_profiler_metadata: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestProfiler.test_tensorboard_trace_handler: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestProfilerTree.test_profiler_experimental_tree: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPythonAutograd.test_backwards1: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPythonBindings.test_cu_create_function: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPythonBuiltinOP.test_stepped_tuple_slicing: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPythonDispatch.test_list_ret: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPythonRegistration.test_fallback: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestPythonRegistration.test_fallback_keyset: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestQuantizedEmbeddingOps.test_embedding_bag_byte: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestRecursiveScript.test_inner_traced_module: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestReductionsCPU.test_all_any_vs_numpy_cpu_bool: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestReductionsCPU.test_all_any_vs_numpy_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestReductionsCPU.test_histogram_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestReductionsCPU.test_histogramdd_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestReductionsCPU.test_tensor_reduce_ops_empty_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_function_overloading_isinstance: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_function_overloads: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_ignored_as_value: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_namedtuple_python: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_no_self_arg_ignore_function: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_python_call_non_tensor_wrong: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_python_op_builtins: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_type_annotation_module: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_unused_decorator: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScript.test_wrong_return_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScriptProfile.test_script: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestScriptProfile.test_section: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSerializeCustomClass.test_custom_class: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSetOps.test_in1d_table_timedelta_fails: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSetOps.test_in1d_timedelta_kind0: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSetOps.test_in1d_timedelta_kind_sort: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSetOps.test_setdiff1d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSlice.test_tuple_slicing: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_add_zeros_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_add_zeros_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_any_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_asin_arcsin_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_asin_arcsin_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_asin_arcsin_cpu_int16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_asin_arcsin_cpu_int32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_asin_arcsin_cpu_int64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_asin_arcsin_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_asin_arcsin_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_assign_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_basic_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_basic_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_basic_ops_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_bmm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_cat_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_cat_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_clone_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_clone_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_contig_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_contig_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_contig_hybrid_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_contig_hybrid_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_div_rounding_mode_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_div_rounding_mode_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_dsmm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_empty_like_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_empty_like_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_factory_copy_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_factory_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_factory_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_factory_cpu_float16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_factory_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_factory_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_hsmm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_index_select_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_index_select_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_is_nonzero_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_isnan_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_legacy_new_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_legacy_new_device_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_log1p_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_log1p_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_log1p_cpu_int16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_log1p_cpu_int32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_log1p_cpu_int64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_log1p_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_log1p_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_mm_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_mm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_mv_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_neg_negative_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_neg_negative_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_new_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_new_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_norm_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_norm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_permute_masked_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_permute_masked_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_permute_sparse_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_permute_sparse_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_pickle_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_print_coalesced_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_print_uncoalesced_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_resize_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_resize_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_saddmm_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_saddmm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_select_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_select_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_shared_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_shared_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_small_nnz_coalesced_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_spadd_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_add_coalesce_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_add_coalesce_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_addmm_cpu_bfloat16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_addmm_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_addmm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_broadcast_to_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_bool: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_int16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_int32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_int64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_dense_mul_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_mask_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_mask_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_mask_hybrid_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_mm_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_int16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_int32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_int64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sparse_mul_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_bool: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_int16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_int32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_int64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_spdiags_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_sum_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sparse_to_numpy_cpu: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_bool: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_complex64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_int16: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_int32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_int64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_sum_cpu_uint8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_transpose_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_transpose_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_unsqueeze_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_unsqueeze_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_zeros_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_zeros_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_zeros_like_cpu_complex128: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCPU.test_zeros_like_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCSRCPU.test_addmm_errors_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCSRCPU.test_mm_errors_cpu_float32: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_bool: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCSRCPU.test_sparse_csc_to_dense_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_bool: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseCSRCPU.test_sparse_csr_to_dense_cpu_int8: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSparseMeta.test_basic: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSqueeze.test_squeeze_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSubscripting.test_test_zero_rank: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestSymbolicTracing.test_cpu_scalar_cuda: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTEFuserDynamic.test_matmul: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTEFuserDynamic.test_unary_ops: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTEFuserStatic.test_skip_grad_in_check: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTEFuserStatic.test_unary_ops: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTensorBuiltins.test_scalar_to_num_conversions: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchTidyProfiler.test_optimizer: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_default_args: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_lambda_as_constructor: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_profiler_custom_op: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_staticmethod: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_attr_exception: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_class_attr_recursive: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_class_attribute: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_deepcopy: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_getattr: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_getstate: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_lambda_method: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_no_init: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_pass_wrong_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_pickle_serialization: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_python_deepcopy: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_return_instance: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_return_tuple: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_save_load: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_take_as_arg: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_tracing: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTorchbind.test_torchbind_tracing_nested: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTypesAndAnnotation.test_pep585_type: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestTyping.test_optional_conversion: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestUnaryUfuncsCPU.test_sinc_cpu_float64: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestVerifyCorrectness.test_example_inputs: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestVmapOperators.test_conv2d: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TestVstack.test_generator: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/dynamo_skips/TraceRuleTests.test_torch_name_rule_map_updated: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/expect/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/export/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/hi.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/inductor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/inductor_skips/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/jit/_imported_class_test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/jit/_imported_class_test/very/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/jit/_imported_class_test/very/very/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/jit/fixtures_srcs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/lazy/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/package/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/package/module_a.py: -------------------------------------------------------------------------------- 1 | result = "module_a" 2 | -------------------------------------------------------------------------------- /test/package/package_d/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/profiler/test_cpp_thread_lib.pyi: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/quantization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/quantization/ao_migration/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/quantization/bc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/quantization/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/quantization/eager/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/quantization/fx/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/quantization/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/torch_np/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/xpu.txt: -------------------------------------------------------------------------------- 1 | 98c808dea6de7330c415aa777d6921944cf79887 2 | -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/alerts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/autograd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/code_coverage/package/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/code_coverage/package/oss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/code_coverage/package/tool/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/code_coverage/package/tool/parser/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/code_coverage/package/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/coverage_plugins_package/src/coverage_plugins/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/github/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/jit/test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/linter/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/linter/clang_tidy/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/lite_interpreter/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/pyi/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/rules/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/stats/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/test/heuristics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/testing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_C/_cusparselt.pyi: -------------------------------------------------------------------------------- 1 | def getVersionInt() -> int: ... 2 | -------------------------------------------------------------------------------- /torch/_custom_op/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_dispatch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_dynamo/backends/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_dynamo/repro/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_export/pass_infra/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_export/serde/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/autoheuristic/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/autoheuristic/artifacts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/codegen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/codegen/cuda/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/codegen/cuda/cutlass_lib_extensions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/codegen/rocm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/codegen/xpu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/compile_worker/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/fx_passes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/fx_passes/serialized_patterns/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_inductor/runtime/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_refs/nn/__init__.py: -------------------------------------------------------------------------------- 1 | __all__: list[str] = [] 2 | -------------------------------------------------------------------------------- /torch/_strobelight/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_thread_safe_fork.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/_vendor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/nn/qat/__init__.py: -------------------------------------------------------------------------------- 1 | from .modules import * # noqa: F403 2 | -------------------------------------------------------------------------------- /torch/ao/nn/sparse/__init__.py: -------------------------------------------------------------------------------- 1 | from . import quantized 2 | -------------------------------------------------------------------------------- /torch/ao/ns/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/ns/fx/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/pruning/_experimental/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/pruning/_experimental/activation_sparsifier/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/pruning/_experimental/data_sparsifier/lightning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/pruning/scheduler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/pruning/sparsifier/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/quantization/backend_config/observation_type.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/quantization/fx/_model_report/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/ao/quantization/pt2e/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/backends/_coreml/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/backends/_nnapi/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/backends/xeon/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/contrib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/csrc/empty.c: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/cuda/error.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/distributed/algorithms/_checkpoint/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/distributed/algorithms/_quantization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/distributed/algorithms/model_averaging/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/distributed/elastic/agent/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/distributed/nn/api/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/distributed/nn/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/distributed/nn/jit/templates/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/fx/experimental/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/fx/experimental/migrate_gradual_types/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/fx/passes/backends/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/fx/passes/dialect/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/fx/passes/dialect/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/fx/passes/infra/__init__.py: -------------------------------------------------------------------------------- 1 | from . import pass_manager 2 | -------------------------------------------------------------------------------- /torch/fx/passes/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/jit/_passes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/nested/_internal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/nn/backends/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/onnx/_internal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/onnx/_internal/exporter/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/onnx/_internal/exporter/_torchlib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/data/__init__.py: -------------------------------------------------------------------------------- 1 | # mypy: ignore-errors 2 | -------------------------------------------------------------------------------- /torch/testing/_internal/distributed/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/distributed/_tensor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/distributed/nn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/distributed/nn/api/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/distributed/rpc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/distributed/rpc/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/distributed/rpc/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/generated/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/testing/_internal/test_module/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/_strobelight/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/_sympy/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/benchmark/examples/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/benchmark/op_fuzzers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/benchmark/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/benchmark/utils/valgrind_wrapper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/bottleneck/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/data/datapipes/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/hipify/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '1.0.0' 2 | -------------------------------------------------------------------------------- /torch/utils/jit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch/utils/serialization/__init__.py: -------------------------------------------------------------------------------- 1 | from . import config 2 | -------------------------------------------------------------------------------- /torch/utils/viz/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchgen/_autoheuristic/requirements.txt: -------------------------------------------------------------------------------- 1 | pandas 2 | scikit-learn 3 | -------------------------------------------------------------------------------- /torchgen/aoti/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchgen/api/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchgen/executorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchgen/executorch/api/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchgen/operator_versions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchgen/selective_build/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchgen/static_runtime/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ubsan.supp: -------------------------------------------------------------------------------- 1 | vptr:pybind11::detail::translate_exception -------------------------------------------------------------------------------- /version.txt: -------------------------------------------------------------------------------- 1 | 2.7.0a0 2 | --------------------------------------------------------------------------------