mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
4ba28deb6e
Summary: This PR is an intermediate step toward the ultimate goal of eliminating "caffe2" in favor of "torch". This PR moves all of the files that had constituted "libtorch.so" into the "libcaffe2.so" library, and wraps "libcaffe2.so" with a shell library named "libtorch.so". This means that, for now, `caffe2/CMakeLists.txt` becomes a lot bigger, and `torch/CMakeLists.txt` becomes smaller. The torch Python bindings (`torch_python.so`) still remain in `torch/CMakeLists.txt`. The follow-up to this PR will rename references to `caffe2` to `torch`, and flatten the shell into one library. Pull Request resolved: https://github.com/pytorch/pytorch/pull/17783 Differential Revision: D15284178 Pulled By: kostmo fbshipit-source-id: a08387d735ae20652527ced4e69fd75b8ff88b05
51 lines
1.5 KiB
Bash
Executable File
51 lines
1.5 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -ex
|
|
|
|
BASE_BRANCH=master
|
|
# From https://docs.travis-ci.com/user/environment-variables
|
|
if [[ $TRAVIS ]]; then
|
|
git remote add upstream https://github.com/pytorch/pytorch
|
|
git fetch upstream "$TRAVIS_BRANCH"
|
|
BASE_BRANCH="upstream/$TRAVIS_BRANCH"
|
|
fi
|
|
|
|
if [[ ! -d build ]]; then
|
|
git submodule update --init --recursive
|
|
|
|
mkdir build
|
|
pushd build
|
|
# We really only need compile_commands.json, so no need to build!
|
|
time cmake ..
|
|
popd
|
|
|
|
# Generate ATen files.
|
|
time python aten/src/ATen/gen.py \
|
|
-s aten/src/ATen \
|
|
-d build/aten/src/ATen \
|
|
aten/src/ATen/Declarations.cwrap \
|
|
aten/src/THNN/generic/THNN.h \
|
|
aten/src/THCUNN/generic/THCUNN.h \
|
|
aten/src/ATen/nn.yaml \
|
|
aten/src/ATen/native/native_functions.yaml
|
|
|
|
# Generate PyTorch files.
|
|
time python tools/setup_helpers/generate_code.py \
|
|
--declarations-path build/aten/src/ATen/Declarations.yaml \
|
|
--nn-path aten/src
|
|
fi
|
|
|
|
# Run Clang-Tidy
|
|
# The negative filters below are to exclude files that include onnx_pb.h or
|
|
# caffe2_pb.h, otherwise we'd have to build protos as part of this CI job.
|
|
time python tools/clang_tidy.py \
|
|
--verbose \
|
|
--paths torch/csrc/ \
|
|
--diff "$BASE_BRANCH" \
|
|
-g"-torch/csrc/distributed/Module.cpp" \
|
|
-g"-torch/csrc/jit/export.cpp" \
|
|
-g"-torch/csrc/jit/import.cpp" \
|
|
-g"-torch/csrc/jit/netdef_converter.cpp" \
|
|
-g"-torch/csrc/jit/register_quantized_ops.cpp" \
|
|
"$@"
|