OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
78c9b2948a
pytorch/tools
History
Yang Chen 78c9b2948a [aot_inductor] move CudaWrapperCodeGen into a separate file (#119870) 
This reverts commit 3ab08946d5.

Differential Revision: [D53817852](https://our.internmc.facebook.com/intern/diff/D53817852)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/119870
Approved by: https://github.com/khabinov
2024-02-16 08:10:51 +00:00
..
alerts
amd_build [aot_inductor] move CudaWrapperCodeGen into a separate file (#119870) 2024-02-16 08:10:51 +00:00
autograd [2/N] Replace std::tie with structural binding (#119879) 2024-02-15 02:56:34 +00:00
bazel_tools
build/bazel
build_defs Fix buck OSS build after #115570 (#115804) 2023-12-14 08:33:07 +00:00
code_analyzer [BE]: Enable RUFF PERF402 and apply fixes (#115505) 2023-12-20 18:01:24 +00:00
code_coverage
config
coverage_plugins_package
dynamo Replaced deprecated pkg_resources.packaging with packaging module (#113023) 2023-11-10 15:06:03 +00:00
gdb
github
iwyu [2/N] Cleanup header inclusions in torch_cpu by iwyu (#109964) 2023-11-19 20:56:32 +00:00
jit
linter Use dmypy instead of mypy in lintrunner (#118475) 2024-01-28 13:42:06 +00:00
lite_interpreter
lldb
onnx [torch/csrc/onnx] Use nested namespaces (3/N) (#113993) 2023-11-18 00:20:19 +00:00
pyi Fix signatures of torch.{add, sub, mul} (#118398) 2024-01-30 22:18:15 +00:00
rules
rules_cc
setup_helpers [BE][Ez]: FURB129: remove unneeded readlines() (#119796) 2024-02-13 21:21:22 +00:00
shared
stats [td] Consistent pytest cache (#113804) 2023-11-17 23:45:47 +00:00
test Revert "Alternate sharding (#119078)" 2024-02-15 16:59:50 +00:00
testing Revert "Alternate sharding (#119078)" 2024-02-15 16:59:50 +00:00
__init__.py
bazel.bzl
BUCK.bzl Reify view_func() closures as ViewFuncs (#118404) 2024-02-14 22:00:43 +00:00
BUCK.oss
build_libtorch.py
build_pytorch_libs.py
build_with_debinfo.py [DevX] Add tool and doc on partial debug builds (#116521) 2023-12-29 05:15:35 +00:00
download_mnist.py
extract_scripts.py
gen_flatbuffers.sh
gen_vulkan_spv.py [pt-vulkan][ez] Replace ska::flat_hash_map, c10::get_hash with std::unordered_map, std::hash (#117177) 2024-01-11 21:43:15 +00:00
generate_torch_version.py Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 2023-11-17 18:24:21 +00:00
generated_dirs.txt
git_add_generated_dirs.sh
git_reset_generated_dirs.sh
nightly.py
nvcc_fix_deps.py [BE][Ez]: FURB129: remove unneeded readlines() (#119796) 2024-02-13 21:21:22 +00:00
pytorch.version
README.md
render_junit.py
substitute.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
update_masked_docs.py Use strict to toggle strict options in MYPYSTRICT (#118479) 2024-01-28 19:22:22 +00:00
vscode_settings.py

README.md

This folder contains a number of scripts which are used as part of the PyTorch build process. This directory also doubles as a Python module hierarchy (thus the __init__.py).

Overview

Modern infrastructure:

  • autograd - Code generation for autograd. This includes definitions of all our derivatives.
  • jit - Code generation for JIT
  • shared - Generic infrastructure that scripts in tools may find useful.
    • module_loader.py - Makes it easier to import arbitrary Python files in a script, without having to add them to the PYTHONPATH first.

Build system pieces:

  • setup_helpers - Helper code for searching for third-party dependencies on the user system.
  • build_pytorch_libs.py - cross-platform script that builds all of the constituent libraries of PyTorch, but not the PyTorch Python extension itself.
  • build_libtorch.py - Script for building libtorch, a standalone C++ library without Python support. This build script is tested in CI.

Developer tools which you might find useful:

Important if you want to run on AMD GPU:

  • amd_build - HIPify scripts, for transpiling CUDA into AMD HIP. Right now, PyTorch and Caffe2 share logic for how to do this transpilation, but have separate entry-points for transpiling either PyTorch or Caffe2 code.
    • build_amd.py - Top-level entry point for HIPifying our codebase.

Tools which are only situationally useful: