You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

Makefile 1.1 kB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. # -*- coding: utf-8 -*-
  2. # MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
  3. #
  4. # Copyright (c) 2014-2020 Megvii Inc. All rights reserved.
  5. #
  6. # Unless required by applicable law or agreed to in writing,
  7. # software distributed under the License is distributed on an
  8. # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  9. ifeq ($(SDKPATH),)
  10. CURDIR := $(shell dirname $(realpath $(firstword $(MAKEFILE_LIST))))
  11. SDKPATH = $(CURDIR)/arm64-v8a
  12. endif
  13. TARGET := libmace_loader.so
  14. all: $(TARGET)
  15. CXX = aarch64-linux-android-g++
  16. CXXFLAGS = -std=c++14 -I$(SDKPATH)/include \
  17. -Dmgb_mace_loader_init=mgb_c_opr_init
  18. DEBUG ?= 0
  19. ifeq ($(DEBUG), 1)
  20. CXXFLAGS += -O0 -g
  21. else
  22. CXXFLAGS += -O2
  23. endif
  24. LDFLAGS = -L$(SDKPATH)/lib \
  25. -lmace_static -lcore -lgenerated_version \
  26. -lport_android -lport_linux_base -lport_posix -lport_base \
  27. -lproto -lutils -lgenerated_opencl_kernel \
  28. -lops -lcore -lprotobuf-lite -llog \
  29. -lpthread -l:rpcmem.a
  30. LDLIBS = -shared -fPIC
  31. $(TARGET): mace_loader.cpp extern_c_opr.h
  32. $(CXX) $(CXXFLAGS) $(LDLIBS) -o $@ $< $(LDFLAGS)
  33. clean:
  34. @rm -f $(TARGET)

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境,不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序,请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉,欢迎访问 MegStudio 平台