mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
edd4e2c5d1
* Expose proto utils and ONNX from PyTorch libcaffe2.so * Try to use protobuf from _C.so * Fix ONNX proto header include * Adjust order of imports for ONNX until nanopb goes away * Set and use ONNX_NAMESPACE for PyTorch builds * Show protobuf summary for all builds * Add ONNX_NAMESPACE for cpp_build * Statically link libprotobuf.a into libtorch.so * Set ONNX_NAMESPACE on Windows build * Move core/dispatch up as well * Add /MD flag for Windows build of _C * Potential Windows fix for ONNX and protobuf * Add direct linkage from _C to ONNX on Windows * Only include protobuf wrapper for PyTorch * Pass extra_compile_args to _nvrtc ext build * Remove installation of .a files
47 lines
1.2 KiB
Bash
Executable file
47 lines
1.2 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
|
|
BUILD_PATH="${1:-$SCRIPTPATH/build}"
|
|
INSTALL_PREFIX="$BUILD_PATH/install"
|
|
PYTORCHPATH="$SCRIPTPATH/../.."
|
|
|
|
USE_CUDA=0
|
|
if [ -x "$(command -v nvcc)" ]; then
|
|
USE_CUDA=1
|
|
fi
|
|
|
|
CAFFE2_BUILDPATH="$BUILD_PATH/caffe2"
|
|
NANOPB_BUILDPATH="$BUILD_PATH/nanopb"
|
|
LIBTORCH_BUILDPATH="$BUILD_PATH/libtorch"
|
|
|
|
# Build with Ninja if available. It has much cleaner output.
|
|
GENERATE="Unix Makefiles"
|
|
MAKE=make
|
|
if [ -x "$(command -v ninja)" ]; then
|
|
GENERATE=Ninja
|
|
MAKE=ninja
|
|
fi
|
|
|
|
# Code is developed a lot more than released, so default to Debug.
|
|
BUILD_TYPE=${BUILD_TYPE:-Debug}
|
|
|
|
# Try to build with as many threads as we have cores, default to 4 if the
|
|
# command fails.
|
|
set +e
|
|
if [ -n "$MAX_JOBS" ]; then # Use MAX_JOBS if it is set
|
|
JOBS=$MAX_JOBS
|
|
elif [[ "$(uname)" == "Linux" ]]; then
|
|
# https://stackoverflow.com/questions/6481005/how-to-obtain-the-number-of-cpus-cores-in-linux-from-the-command-line
|
|
JOBS="$(grep -c '^processor' /proc/cpuinfo)"
|
|
else # if [[ "$(uname)" == "Darwin"]]
|
|
# https://stackoverflow.com/questions/1715580/how-to-discover-number-of-logical-cores-on-mac-os-x
|
|
JOBS="$(sysctl -n hw.ncpu)"
|
|
fi
|
|
set -e
|
|
if [[ $? -ne 0 ]]; then
|
|
JOBS=4
|
|
fi
|
|
|
|
# Make sure an ONNX namespace is set
|
|
if [ -z "$ONNX_NAMESPACE" ]; then
|
|
ONNX_NAMESPACE="onnx_torch"
|
|
fi
|