-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy patharch.make.docker
More file actions
27 lines (18 loc) · 1.07 KB
/
arch.make.docker
File metadata and controls
27 lines (18 loc) · 1.07 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
CXX = g++
CC = g++
FC = gfortran --free-form
FC_SERIAL = gfortran --free-form
LDFLAGS = -lgfortran
LIBS= $(LD_FLAGS) -lstdc++
# Libtorch location (required)
TORCH=/lib/libtorch
# Google test location (optional)
GTEST= -L/lib/googletest/lib/ -lgtest
GTEST_INCLUDE=/lib/googletest/googletest/include/
# Libxc (optional)
LIBXC_INCLUDE=
LIBXC_LD=
LIBXCDIR=
# The following parts should remain largely unchanged
CXXFLAGS=-DAT_PARALLEL_OPENMP=1 -isystem $(TORCH)/include -isystem $(TORCH)/include/torch/csrc/api/include -DBOOST_MATH_DISABLE_FLOAT128 -fPIC -D_GLIBCXX_USE_CXX11_ABI=1 -Wall -Wextra -Wno-unused-parameter -Wno-missing-field-initializers -Wno-write-strings -Wno-unknown-pragmas -Wno-missing-braces -fopenmp -std=gnu++17 -I/usr/include
TORCH_LD= -DBOOST_MATH_DISABLE_FLOAT128 -fPIC -Wl,-rpath,/lib -L/lib -Wl,-rpath,$(TORCH)/lib/ $(TORCH)/lib/libtorch.so $(TORCH)/lib/libc10.so -Wl,--no-as-needed,$(TORCH)/lib/libtorch_cpu.so -Wl,--as-needed $(TORCH)/lib/libc10.so -lpthread -Wl,--no-as-needed,$(TORCH)/lib/libtorch.so -Wl,--as-needed -lgfortran -lquadmath -lstdc++ -shared