mirror of https://github.com/alibaba/MNN.git
9.3 KiB
9.3 KiB
编译宏介绍
MNN使用CMake构建项目,CMake中的宏定义列表如下:
| 宏名 | 宏说明 |
|---|---|
| MNN_USE_SYSTEM_LIB | 在使用opencl和vulkan时,使用系统库(ON)还是通过dlopen引入动态库(OFF),默认为OFF |
| MNN_BUILD_HARD | 是否使用-mfloat-abi=hard,默认为OFF |
| MNN_BUILD_SHARED_LIBS | 是否构建为动态库,默认为ON |
| MNN_WIN_RUNTIME_MT | 在Windows上构建dll时是否使用/MT,默认为OFF |
| MNN_FORBID_MULTI_THREAD | 是否禁止多线程,默认为OFF |
| MNN_OPENMP | 是否使用OpenMP的线程池,该选项在Mac/iOS平台无效,默认为OFF |
| MNN_USE_THREAD_POOL | 是否使用MNN内部线程池,默认为ON |
| MNN_BUILD_TRAIN | 是否构建MNN的训练框架,默认为OFF |
| MNN_BUILD_DEMO | 是否构建MNN的demo,默认为OFF |
| MNN_BUILD_TOOLS | 是否构建MNN的测试工具,默认为ON |
| MNN_BUILD_QUANTOOLS | 是否构建MNN的量化工具,默认为OFF |
| MNN_EVALUATION | 是否构建MNN的评估工具,默认为OFF |
| MNN_BUILD_CONVERTER | 是否构建MNN的转换工具,默认为OFF |
| MNN_SUPPORT_QUNAT_EXTEND | 是否编译非核心算子的量化版本,默认为ON |
| MNN_SUPPORT_DEPRECATED_OP | 是否支持Tflite的量化算子等已经废弃的算子,用于兼容历史模型(1.1.0版本之前),默认为OFF |
| MNN_SUPPORT_DEPRECATED_OPV2 | 是否编译MNN更新到3.0之后已经废弃的算子,用于兼容历史模型(3.0.0版本之前),比如 Convolution3D 和 ConvTranspose3D在3.0.0 版本之后改由模型转换器转化为对应2D算子,不再需要运行时支持,默认为ON |
| MNN_REDUCE_SIZE | 是否裁剪MNN库大小,去除求导相关算子,减少优化策略,默认为OFF ,开启时,MNN_SUPPORT_QUNAT_EXTEND / MNN_SUPPORT_DEPRECATED_OP / MNN_SUPPORT_DEPRECATED_OPV2 都会设成 OFF |
| MNN_DEBUG_MEMORY | 是否开启MNN内存调试,默认为OFF |
| MNN_DEBUG_TENSOR_SIZE | 是否开启MNN tensor size调试,默认为OFF |
| MNN_GPU_TRACE | 是否开启MNN GPU调试,默认为OFF |
| MNN_SEP_BUILD | 是否构建MNN的后端和表达式分离版本,只在MNN_BUILD_SHARED_LIBS=ON时生效,默认为ON |
| NATIVE_LIBRARY_OUTPUT | 如果构建为动态库,则指定动态库的输出路径,默认为OFF |
| NATIVE_INCLUDE_OUTPUT | 如果构建为动态库,则指定动态库的头文件路径,默认为OFF |
| MNN_AAPL_FMWK | 是否构建MNN.framework替代*.dylib,默认为OFF |
| MNN_WITH_PLUGIN | 是否支持Plugin算子,默认为OFF |
| MNN_SKIPBUILD_GEOMETRY | 是否跳过MNN的几何计算编译,若是,MNN引擎仅支持在模型转换工具时加上 --saveStaticModel 转换出来的固定输入形状的模型,默认为OFF |
| MNN_BUILD_MINI | 是否构建MNN的最小化版本,若是,开启 MNN_SKIPBUILD_GEOMETRY 和 MNN_REDUCE_SIZE,默认为OFF |
| MNN_USE_SSE | 在x86上是否使用SSE指令集,默认为OFF |
| MNN_BUILD_CODEGEN | 是否构建MNN的代码生成部分,该功能提供了算子融合与代码生成能力,为实验性功能,默认为OFF |
| MNN_ENABLE_COVERAGE | 是否开启MNN的代码覆盖率,默认为OFF |
| MNN_BUILD_PROTOBUFFER | 是否使用MNN中的protobuffer,默认为ON |
| MNN_BUILD_OPENCV | 是否构建MNN的OpenCV功能,默认为OFF |
| MNN_BUILD_AUDIO | 是否构建MNN的Audio功能,默认为OFF |
| MNN_INTERNAL | 是否构建MNN的一些内部功能,如:日志;默认为OFF |
| MNN_JNI | 是否构建MNN的JNI支持,默认为OFF |
| MNN_METAL | 是否构建Metal后端,默认为OFF |
| MNN_OPENCL | 是否构建OpenCL后端,默认为OFF |
| MNN_OPENGL | 是否构建OpenGL后端,默认为OFF |
| MNN_VULKAN | 是否构建Vulkan后端,默认为OFF |
| MNN_ARM82 | 编译ARM架构时,是否构建Armv8.2后端,以支持FP16计算,默认为ON |
| MNN_SUPPORT_FP16_ARMV7 | 编译armeabi-v7a架构时,是否构建Armv8.2后端,以支持FP16计算,默认为OFF |
| MNN_ONEDNN | 是否使用oneDNN,默认为OFF |
| MNN_AVX2 | 在MNN_USE_SSE开启的基础上,是否增加AVX2指令的支持,默认为ON |
| MNN_AVX512 | 在MNN_USE_SSE和MNN_AVX2开启的基础上,是否增加avx512指令集的支持,默认为OFF |
| MNN_CUDA | 是否构建Cuda后端,默认为OFF |
| MNN_CUDA_PROFILE | 是否打开CUDA profile工具,默认为OFF |
| MNN_CUDA_QUANT | 是否打开CUDA 量化文件编译,默认为OFF |
| MNN_CUDA_BF16 | 是否打开CUDA Bf16文件编译,默认为OFF |
| MNN_CUDA_TUNE_PARAM | 是否打开CUDA TUNE相关文件编译,目前仅支持安培及以上架构,默认为OFF |
| MNN_TENSORRT | 是否构建TensorRT后端,默认为OFF |
| MNN_COREML | 是否构建CoreML后端,默认为OFF |
| MNN_NNAPI | 是否构建NNAPI后端,默认为OFF |
| MNN_BUILD_BENCHMARK | 是否构建MNN的性能测试,默认为OFF |
| MNN_BUILD_TEST | 是否构建MNN的单元测试,默认为OFF |
| MNN_BUILD_FOR_ANDROID_COMMAND | 是否使用命令行构建Android,默认为OFF |
| MNN_USE_LOGCAT | 是否使用logcat代替printf输出日志,默认为OFF |
| MNN_USE_CPP11 | 是否使用C++11编译MNN,默认为ON |
| MNN_SUPPORT_BF16 | 是否支持BF16,默认为OFF |
| MNN_SSE_USE_FP16_INSTEAD | 在X86平台是否使用FP16替代BF16,默认为OFF |
| MNN_AVX512_VNNI | 是否使用avx512_vnni指令,该宏仅在MNN_AVX512=ON时生效,默认为OFF |
| MNN_OPENCL_SIZE_CUT | 是否为了降低OpenCL大小而关闭OpenCL Buffer实现,该宏仅在MNN_OPENCL=ON时生效,默认为OFF |
| MNN_GPU_TIME_PROFILE | 是否打开OpenCL后端及Vulkan后端的Kernel性能Profile,该宏仅在MNN_OPENCL=ON或MNN_VULKAN=ON时生效,默认为OFF |
| MNN_METALLIB_SOURCE | 使用Metal时是否直接使用Metal源码,该宏仅在MNN_METAL=ON时生效,默认为ON |
| MNN_VULKAN_DEBUG | 是否打开Vulkan的DEBUG模式,该宏仅在MNN_VULKAN=ON时生效,默认为OFF |
| MNN_OPENGL_REGEN | 是否重新生成OpenGL Kenel,该宏仅在MNN_OPENGL=ON时生效,默认为OFF |
| MNN_TRT_DYNAMIC | 是否通过dlopen的方式引入TRT的动态库,该宏仅在MNN_TENSORRT=ON时生效,默认为`OFF |
| MNN_BUILD_TORCH | 构建的MNNConvert是否支持TorchScript,该宏仅在MNN_BUILD_CONVERTER=ON时生效,默认为OFF |
| MNN_TRAIN_DEBUG | 构建的训练模块是否支持调试,该宏仅在MNN_BUILD_TRAIN=ON时生效,默认为OFF |
| MNN_USE_OPENCV | 构建的训练Demo是否使用OpenCV依赖,该宏仅在MNN_BUILD_TRAIN=ON时生效,默认为OFF |
| MNN_IMGPROC_COLOR | 构建MNN的OpenCV功能是否开启颜色空间转换,默认为ON |
| MNN_IMGPROC_GEOMETRIC | 构建MNN的OpenCV功能是否开启形变,默认为ON |
| MNN_IMGPROC_DRAW | 构建MNN的OpenCV功能是否开启画图,默认为ON |
| MNN_IMGPROC_FILTER | 构建MNN的OpenCV功能是否开启滤波,默认为ON |
| MNN_IMGPROC_MISCELLANEOUS | 构建MNN的OpenCV功能是否开启混合,默认为ON |
| MNN_IMGPROC_STRUCTRAL | 构建MNN的OpenCV功能是否开启结构,默认为ON |
| MNN_IMGPROC_HISTOGRAMS | 构建MNN的OpenCV功能是否开启直方图,默认为ON |
| MNN_CALIB3D | 构建MNN的OpenCV功能是否开启3d,默认为ON |
| MNN_IMGCODECS | 构建MNN的OpenCV功能是否开启图像编解码,默认为OFF |
| MNN_CVCORE | 构建MNN的OpenCV功能是否开启core功能,默认为ON |
| MNN_OPENCV_TEST | 构建MNN的OpenCV功能是否开启单元测试,默认为OFF |
| MNN_OPENCV_BENCH | 构建MNN的OpenCV功能是否开启性能benchmark,默认为OFF |
| MNN_AUDIO_TEST | 构建MNN的Audio功能是否开启单元测试,默认为OFF |
| MNN_VULKAN_IMAGE | 构建MNN的Vulkan后端时采用Image内存模式,以便支持FP16和部分移动端上GPU的加速,默认为ON |
| MNN_LOW_MEMORY | 是否支持低内存模式,支持低内存模式使用权值量化模型并设置low_memory则会使用计算时反量化,默认为OFF |
| MNN_CPU_WEIGHT_DEQUANT_GEMM | 是否编译CPU权重反量化的矩阵乘Kernel, 如果打开该编译宏并且在CPU推理时设置MNN::BackendConfig::MemoryMode=Memory_Normal,就会使用权重反量化算子进行权重量化模型的推理,默认为OFF |
| MNN_SUPPORT_RENDER | 是否支持图形渲染相关算子实现,默认为 OFF |
| MNN_SUPPORT_TRANSFORMER_FUSE | 是否支持Fuse Transformer相关OP实现,默认为 OFF |
| MNN_BUILD_LLM | 是否构建基于MNN的llm库和demo,默认为OFF |
| MNN_BUILD_DIFFUSION | 是否构建基于MNN的diffusion demo,需要打开MNN_BUILD_OPENCV和MNN_IMGCODECS宏使用 默认为OFF |
| MNN_KLEIDIAI | 是否集成ARM的klediAI加速库【目前处于实验状态,只能跑对称量化的LLM模型】,默认为OFF |