forked from flexflow/flexflow-train
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Makefile
57 lines (51 loc) · 2.48 KB
/
Makefile
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
# Copyright 2017 Stanford University
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
ifndef LG_RT_DIR
#$(error LG_RT_DIR variable is not defined, aborting build)
LG_RT_DIR ?= legion/runtime
endif
# Flags for directing the runtime makefile what to include
DEBUG ?= 0 # Include debugging symbols
MAX_DIM ?= 4 # Maximum number of dimensions
OUTPUT_LEVEL ?= LEVEL_DEBUG # Compile time logging level
USE_CUDA ?= 1 # Include CUDA support (requires CUDA)
USE_GASNET ?= 0 # Include GASNet support (requires GASNet)
USE_HDF ?= 0 # Include HDF5 support (requires HDF5)
ALT_MAPPERS ?= 0 # Include alternative mappers (not recommended)
# Put the binary file name here
OUTFILE ?= $(app)
# List all the application source files here
GEN_SRC ?= src/runtime/model.cc src/mapper/mapper.cc src/runtime/initializer.cc src/runtime/optimizer.cc\
src/runtime/strategy.pb.cc src/runtime/strategy.cc $(app).cc
GEN_GPU_SRC ?= src/ops/conv_2d.cu src/runtime/model.cu src/ops/pool_2d.cu src/ops/batch_norm.cu src/ops/linear.cu \
src/ops/softmax.cu src/ops/concat.cu src/ops/flat.cu src/ops/embedding.cu src/ops/mse_loss.cu\
src/runtime/initializer_kernel.cu src/runtime/optimizer_kernel.cu src/runtime/accessor_kernel.cu\
src/runtime/cuda_helper.cu $(app).cu# .cu files
# You can modify these variables, some will be appended to by the runtime makefile
INC_FLAGS ?= -Iinclude/ -Iprotobuf/src -I/public/apps/cudnn/v7.6/cuda/include -I/usr/include/hdf5/serial/
CC_FLAGS ?=
NVCC_FLAGS ?=
GASNET_FLAGS ?=
LD_FLAGS ?= -L/usr/lib/x86_64-linux-gnu/hdf5/serial/ -L/usr/local/lib -Lprotobuf/src/.libs -L/public/apps/cudnn/v7.6/cuda/lib64
LD_FLAGS += -lcudnn -lcublas -lcurand -lprotobuf -lhdf5
# For Point and Rect typedefs
CC_FLAGS += -std=c++11
NVCC_FLAGS += -std=c++11
###########################################################################
#
# Don't change anything below here
#
###########################################################################
include $(LG_RT_DIR)/runtime.mk