saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/tools
History
Kurt Mohler b915877deb Support numpy array in Tensor.__eq__ (#122249) 
When the `other` arg of `Tensor.__eq__` is a numpy array, it is converted to a PyTorch tensor view of the numpy array, which is then given as the `other` arg to a `Tensor.eq` call

Fixes #119965
Pull Request resolved: https://github.com/pytorch/pytorch/pull/122249
Approved by: https://github.com/ezyang
2024-03-21 04:55:01 +00:00
..
alerts
amd_build [aot_inductor] move CppWrapperCodeGen into a separate file (#119871) 2024-02-16 08:14:20 +00:00
autograd Support numpy array in Tensor.__eq__ (#122249) 2024-03-21 04:55:01 +00:00
bazel_tools
build/bazel
build_defs
code_analyzer
code_coverage
config
coverage_plugins_package
dynamo
gdb
github
iwyu
jit
linter [CI] Enables support for pytorch ci build in ARC + introduces _linux-build-rg.yml. (#121930) 2024-03-20 19:06:10 +00:00
lite_interpreter
lldb
onnx
pyi Fix torch.return_types init signature (#119284) 2024-02-23 21:52:34 +00:00
rules
rules_cc
setup_helpers
shared
stats TD outside of test job (#118250) 2024-03-01 23:08:10 +00:00
test Fix round robin sharding (#121022) 2024-03-11 17:30:12 +00:00
testing [TD] Add LLM retrieval + heuristic (#121836) 2024-03-19 17:31:47 +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 [EZ] Fix argument parsing in build_with_debinfo (#120088) 2024-02-16 20:06:52 +00:00
download_mnist.py
extract_scripts.py
gen_flatbuffers.sh
gen_vulkan_spv.py [ET-VK] Enable Dynamic shape support via tensor virtual and physical resizing (#121598) 2024-03-12 14:32:00 +00:00
generate_torch_version.py
generated_dirs.txt
git_add_generated_dirs.sh
git_reset_generated_dirs.sh
nightly.py
nvcc_fix_deps.py
pytorch.version
README.md
render_junit.py
substitute.py
update_masked_docs.py
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: