ADBenchSolids/Makefile at main · laylagi/ADBenchSolids · GitHub

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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# Variables for Rust library
RUST_PROFILE ?= release # or debug
RUST_LIB_DIR ?= $(abspath target/$(RUST_PROFILE))
RUST_LIB = $(RUST_LIB_DIR)/libenzyme_rust.a

# Variables for Enzyme, ADOL-C, and Tapenade paths
ENZYME_LIB ?=
ADOLC_INCLUDE ?=
ADOLC_LIB ?=

RUSTC = $(realpath $(shell rustup +enzyme which rustc))
RUSTC_DIR = $(abspath $(dir $(RUSTC))../..)
RUSTC_LLVM = $(RUSTC_DIR)/llvm/bin
ENZYME_LIB = $(wildcard $(RUSTC_DIR)/enzyme/lib/ClangEnzyme-*.so)

# Compilers
CC = $(RUSTC_LLVM)/clang
CXX = g++
FC = gfortran

# Flags
CFLAGS = $(OPT) -Wall -Wextra -Wunused-variable -Wunused-function -Iinclude
CXXFLAGS = $(OPT) -std=c++11 -Wall -Wextra -Wunused-variable -Wunused-function \
            -Wno-unused-parameter $(patsubst %,-I%,$(INCDIR) $(ADOLC_INCLUDE))

ifneq ($(ADOLC_LIB),)
    LDFLAGS += -L$(ADOLC_LIB) -Wl,-rpath,$(ADOLC_LIB)
endif
LDFLAGS += -L$(RUST_LIB_DIR)
LDLIBS = -ladolc -lm -lenzyme_rust

# Add Enzyme-specific flags if ENZYME_LIB is defined
ifneq ($(ENZYME_LIB),)
    CFLAGS += -Xclang -load -Xclang $(ENZYME_LIB)
endif

# Directories
SRCDIR = src
ADTOOLSDIR = $(SRCDIR)/ad-tools
INCDIR = include
BUILDDIR = build
BUILDTOOLSDIR = $(BUILDDIR)/ad-tools

# Source files
SOURCES_CXX = $(wildcard $(SRCDIR)/*.cpp) $(wildcard $(ADTOOLSDIR)/*.cpp)
SOURCES_C = $(wildcard $(ADTOOLSDIR)/*.c)

# Object files
OBJ_CXX = $(SOURCES_CXX:$(SRCDIR)/%.cpp=$(BUILDDIR)/%.o)
OBJ_C = $(SOURCES_C:$(ADTOOLSDIR)/%.c=$(BUILDTOOLSDIR)/%.o)

# All object files
OBJ = $(OBJ_CXX) $(OBJ_C)

# Executable name
TARGET = $(BUILDDIR)/elasticity-exec

# Default target
all: $(TARGET)

# Build the Rust library
$(RUST_LIB): Cargo.toml src/ad-tools/enzyme-rust/Cargo.toml $(wildcard src/ad-tools/enzyme-rust/src/*.rs)
	RUSTFLAGS='-C target-cpu=native -Z autodiff=Enable' cargo +enzyme build --profile $(RUST_PROFILE)

# Link object files to create the single executable
$(TARGET): $(OBJ) $(RUST_LIB) | $(BUILDDIR)
	$(CXX) $(CXXFLAGS) $(OBJ) $(LDFLAGS) $(LDLIBS) -o $@

# Compile C++ source files
$(BUILDDIR)/%.o: $(SRCDIR)/%.cpp | $(BUILDDIR)
	$(CXX) $(CXXFLAGS) -c $< -o $@

$(BUILDTOOLSDIR)/%.o: $(ADTOOLSDIR)/%.cpp | $(BUILDTOOLSDIR)
	$(CXX) $(CXXFLAGS) -c $< -o $@

# Compile C source files
$(BUILDTOOLSDIR)/%.o: $(ADTOOLSDIR)/%.c | $(BUILDTOOLSDIR)
	$(CC) $(CFLAGS) -c $< -o $@

# Ensure necessary directories exist
$(BUILDDIR):
	mkdir -p $(BUILDDIR)

$(BUILDTOOLSDIR):
	mkdir -p $(BUILDTOOLSDIR)

# Clean up build artifacts
clean:
	rm -f $(BUILDDIR)/*.o $(BUILDTOOLSDIR)/*.o $(TARGET)
	cargo +enzyme clean

print-% :
	$(info [ variable name]: $*)
	$(info [        origin]: $(origin $*))
	$(info [        flavor]: $(flavor $*))
	$(info [         value]: $(value $*))
	$(info [expanded value]: $($*))
	$(info )
	@true

.PHONY: all clean