Skip to content

Instantly share code, notes, and snippets.

@ezyang
Last active April 11, 2019 16:12
Show Gist options
  • Select an option

  • Save ezyang/8e19c3b49cce4ec825def84c1982a522 to your computer and use it in GitHub Desktop.

Select an option

Save ezyang/8e19c3b49cce4ec825def84c1982a522 to your computer and use it in GitHub Desktop.
Classification of every file in PyTorch into a module
# autograd
/test/test_autograd.py
/torch/autograd/
/torch/csrc/autograd/
/torch/utils/hooks.py
# benchmarks
/benchmarks/
# binaries
/aten/conda/
/tools/pytorch.version
# bottleneck
/test/bottleneck/
# bottleneck
torch/utils/bottleneck/
# build
CMakeLists.txt
/cmake/
Makefile
requirements.txt
/scripts/
/setup.py
/tools/build_*.py
/tools/setup_helpers/
# caffe2
/caffe2/
/binaries/
/docs/caffe2/
/modules/
# checkpoint
torch/utils/checkpoint.py
# ci
/.circleci/
/.jenkins/
/.travis.aten.yml
/.travis.yml
/aten/tools/
/docker/caffe2/jenkins/
# collect_env.py
torch/utils/collect_env.py
# contrib
/torch/contrib/
# cpp
/docs/cpp/
/test/cpp/
/torch/abi-check.cpp
/torch/csrc/api/
# cpp-extensions
/test/test_cpp_extensions.py
/test/cpp_extensions/
/torch/extension.h
/torch/utils/cpp_extension.py
/torch/utils/_cpp_extension_versioner.py
/torch/utils/ffi/
# cpu
/aten/src/ATen/cpu/
/aten/src/TH/
/aten/src/THNN/
/aten/src/ATen/native/cpu/
# cuda
/aten/src/ATen/cuda/
/c10/cuda/
/test/*cuda*.py
/torch/backends/cuda/
/aten/src/THC/
/aten/src/THCUNN/
/aten/src/ATen/native/cuda/
/torch/csrc/cuda/
/torch/cuda/
# cudnn
/torch/backends/cudnn/
/aten/src/ATen/native/cudnn/
/aten/src/ATen/cudnn/
# custom-operators
/torch/_ops.py
/test/custom_operator/
# dataloader
/test/test_dataloader.py
/torch/utils/data/
# diagnostics
/aten/src/ATen/Version.*
/torch/__config__.py
# distributed
/test/test_c10d.py
/test/test_distributed.py
/test/test_nccl.py
/test/test_thd_distributed.py
/torch/csrc/distributed/
/torch/distributed/
/torch/lib/THD/
/torch/lib/c10d/
# distributions
/test/test_distributions.py
/torch/distributions/
# dlpack
torch/utils/dlpack.py
# docker
/docker/
!/docker/caffe2/jenkins/
/tools/docker/
# documentation
/CONTRIBUTING.md
README*
/docs/
/test/test_docs_coverage.py
*_docs.py
# hub
/torch/hub.py
/torch/utils/model_zoo.py
# infra
/.github/
/.ctags.d/
/.dockerignore
/.gitattributes
.gitignore
/CITATION
/CODEOWNERS
/LICENSE
/NOTICE
/tools/__init__.py
/tools/aten_mirror.sh
/tools/download_mnist.py
/tools/generated_dirs.txt
/tools/git*
/tools/shared/
# internals
/aten/src/ATen/*
/aten/src/ATen/core/
/aten/src/ATen/detail/
/aten/src/ATen/stub/
/aten/src/ATen/templates/
/aten/src/ATen/test/
/c10
/tools/autograd/
/tools/cwrap/
/tools/nnwrap/
/torch/_six.py
/torch/_utils*
/torch/__init__.py
/torch/backends/__init__.py
/torch/csrc/*
/torch/csrc/generic/
/torch/csrc/nn/
/torch/csrc/tensor/
/torch/csrc/utils/
/torch/utils/__init__.py
/torch/utils/backcompat/
# jit
/torch/csrc/jit/
/test/*jit*
/test/test_module/
/tools/jit/
/torch/_jit_internal.py
/torch/jit
/torch/script.h
# lint
/.clang-format
/.clang-tidy
/.flake8
/tools/clang_*.py
/tools/flake8_*.py
/tools/run-clang-tidy-in-ci.sh
/ubsan.supp
# mkl
/aten/src/ATen/mkl/
/aten/src/ATen/native/mkl/
/torch/backends/mkl/
# mkldnn
/test/test_mkldnn.py
/aten/src/ATen/mkldnn/
/aten/src/ATen/native/mkldnn/
/torch/backends/mkldnn/
# multiprocessing
/test/test_multiprocessing*.py
/torch/csrc/multiprocessing/
/torch/lib/libshm*/
/torch/multiprocessing/
/torch/utils/file_baton.py
# nn
/test/test_nn.py
/test/common_nn.py
/torch/_thnn/
/torch/nn/
# numba
/test/test_numba_integration.py
# onnx
/test/onnx/
/torch/csrc/onnx/
/torch/for_onnx/
/torch/onnx/
# openmp
/torch/backends/openmp/
# operators
/aten/src/THC/
/aten/src/THCUNN/
/aten/src/TH/
/aten/src/THNN/
/aten/src/ATen/native/
/test/test_indexing.py
/test/test_namedtuple_return_api.py
/test/test_torch.py
/test/test_type_info.py
/torch/functional.py
/torch/storage.py
/torch/tensor.py
# optimizer
/test/optim/
/test/test_optim.py
/torch/optim/
# printing
/torch/_tensor_str.py
# quantization
/test/test_quantized.py
/test/test_quantizer.py
/aten/src/ATen/quantized/
/aten/src/ATen/native/quantized/
# random
/torch/quasirandom.py
/torch/random.py
# rocm
/c10/hip/
/tools/amd_build/
/aten/src/ATen/hip/
/aten/src/ATen/miopen/
/aten/src/ATen/native/miopen/
# serialization
/torch/serialization.py
# sparse
/test/test_sparse.py
/aten/src/ATen/native/sparse/
/torch/sparse/
# tests
/test/*.md
/test/common_methods_invocations.py
/test/common_utils.py
/test/data/
/test/error_messages/storage.py
/test/expect/
/test/expecttest.py
/test/test_expecttest.py
/test/run_test.py
/test/test_utils.py
/torch/testing/
# third_party
/.gitmodules
/submodules/
/third_party/
# typing
*.pyi
*.pyi.in
mypy*
/test/test_type_hints.py
/tools/pyi/
/torch/py.typed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment