forked from huawei/mindspore2022
49 lines
1.5 KiB
Makefile
49 lines
1.5 KiB
Makefile
BASE_DIR=$(realpath ../../../../)
|
|
APP:=bin/net_runner
|
|
LMSLIB:=-lmindspore-lite-train -lmindspore-lite
|
|
LMDLIB:=-lminddata-lite
|
|
MSDIR:=$(realpath package-$(TARGET)/lib)
|
|
ifneq ("$(wildcard $(MSDIR)/libhiai.so)","")
|
|
LHIAILIB:=-lhiai_ir_build -lhiai_ir -lhiai
|
|
else
|
|
LHIAILIB:=
|
|
endif
|
|
|
|
SRC:=src/net_runner.cc src/dataset.cc
|
|
OBJ:=$(SRC:.cc=.o)
|
|
|
|
CFLAGS := -Ofast -std=c++17 \
|
|
-I . \
|
|
-I ./msl/train \
|
|
-I ./msl/train/minddata \
|
|
-I ./msl/tools/third_party/flatbuffers/include
|
|
|
|
|
|
ifeq ($(TARGET),arm64)
|
|
CXX := ${ANDROID_NDK}/toolchains/llvm/prebuilt/linux-x86_64/bin/clang++
|
|
CFLAGS += --target=aarch64-none-linux-android21 --gcc-toolchain=${ANDROID_NDK}/toolchains/llvm/prebuilt/linux-x86_64 --sysroot=${ANDROID_NDK}/toolchains/llvm/prebuilt/linux-x86_64/sysroot -fdata-sections -ffunction-sections
|
|
LDFLAGS := --target=aarch64-none-linux-android21 --gcc-toolchain=${ANDROID_NDK}/toolchains/llvm/prebuilt/linux-x86_64 --sysroot=${ANDROID_NDK}/toolchains/llvm/prebuilt/linux-x86_64/sysroot -Wl,--gc-sections
|
|
LDFLAGS += -L$(MSDIR) $(LMSLIB) $(LMDLIB) $(LHIAILIB) -pthread -llog -latomic -lm -Wl,-rpath,$(MSDIR)
|
|
else
|
|
CFLAGS += -g
|
|
LDFLAGS := -L$(MSDIR) $(LMSLIB) $(LMDLIB) $(LHIAILIB) -lpthread -Wl,-rpath,$(MSDIR)
|
|
endif
|
|
LD := ${CXX}
|
|
|
|
|
|
all:$(APP)
|
|
|
|
$(APP): $(OBJ)
|
|
@mkdir -p bin
|
|
$(LD) $(OBJ) $(LDFLAGS) -o $@
|
|
|
|
clean:
|
|
rm -rf src/*.o bin/
|
|
|
|
|
|
mrproper:
|
|
rm -rf dataset package* msl src/*.o bin/ model/*.mindir model/*.ms model/*.so model/converter_lite
|
|
|
|
%.o:%.cc
|
|
$(CXX) $(CFLAGS) -c $< -o $@
|