MFEM 有限元库安装与构建指南

概述

MFEM 是一个开源的 C++ 库，用于实现高性能有限元方法 (FEM) 计算。它支持串行和并行计算，并提供了多种先进功能，包括对 GPU 加速的支持。本文将详细介绍如何构建和安装 MFEM 库。

构建前准备

基本要求

• C++ 编译器：从 4.0 版本开始，MFEM 需要支持 C++11 的编译器。推荐使用较新的编译器，如 GCC 4.9 或更高版本。

并行版本额外要求

构建并行版本需要以下外部库：

1. hypre：高性能预处理器库
2. METIS：多级分区算法库（推荐使用 metis-4.0.3/5.1.0）

注意：METIS 依赖可以禁用，但不推荐这样做（通过 MFEM_USE_METIS 选项控制）。

设备支持

MFEM 支持多种设备和编程模型：

• CUDA：需要 NVIDIA GPU 和 CUDA Toolkit
• HIP：需要 AMD GPU 和 ROCm 软件栈
• OCCA：需要 OCCA 库
• OpenMP：需要支持 OpenMP API 的编译器
• RAJA：需要安装 RAJA 性能可移植层

构建系统选择

MFEM 支持两种构建系统：

1. GNU make：传统的基于 makefile 的构建系统
2. CMake：跨平台的构建系统

快速开始

使用 GNU make

串行构建

make serial -j 4

并行构建

# 先下载并构建 hypre 和 METIS
make parallel -j 4

CUDA 构建

make cuda -j 4
# 指定计算能力
make cuda -j 4 CUDA_ARCH=sm_70

HIP 构建

make hip -j 4
# 指定 AMD GPU 架构
make hip -j 4 HIP_ARCH=gfx900

构建示例代码

cd examples
make -j 4

完整构建

make all -j 4

快速测试

make check

使用 CMake

串行构建

mkdir build && cd build
cmake ..
make -j 4

并行构建

mkdir build && cd build
cmake .. -DMFEM_USE_MPI=YES
make -j 4

CUDA 构建

mkdir build && cd build
cmake .. -DMFEM_USE_CUDA=YES -DCUDA_ARCH=sm_70
make -j 4

HIP 构建

mkdir build && cd build
cmake .. -DMFEM_USE_HIP=YES -DHIP_ARCH=gfx942 \
         -DCMAKE_CXX_COMPILER=amdclang++ \
         -DCMAKE_HIP_COMPILER=amdclang++
make -j 4

构建示例代码

make examples -j 4

完整构建

make exec -j 4

快速测试

make check

详细构建说明

GNU make 构建流程

MFEM 的构建分为两个步骤：配置和编译。

配置步骤

make config [OPTIONS] ...

或使用配置文件：

cp config/defaults.mk config/user.mk
# 编辑 config/user.mk
make config

构建步骤

make

快捷目标

• make serial：构建串行优化版本
• make parallel：构建并行优化版本
• make debug：构建串行调试版本
• make pdebug：构建并行调试版本
• make cuda：构建串行 CUDA 优化版本
• make pcuda：构建并行 CUDA 优化版本
• make hip：构建串行 HIP 优化版本
• make phip：构建并行 HIP 优化版本

测试

make check  # 快速测试
make test   # 完整测试

安装

make install [PREFIX=<dir>]

配置选项详解

编译器选项

• CXX：C++ 编译器（串行构建）
• MPICXX：MPI C++ 编译器（并行构建）
• CUDA_CXX：CUDA 编译器（'nvcc' 或 'clang++'）

构建选项

• STATIC：是否构建静态库（默认 YES）
• SHARED：是否构建共享库（默认 NO）

MFEM 功能选项

• MFEM_USE_MPI：启用/禁用并行构建
• MFEM_USE_METIS：启用/禁用 METIS 库
• MFEM_PRECISION：设置浮点精度（double 或 single）
• MFEM_DEBUG：启用/禁用调试构建
• MFEM_USE_EXCEPTIONS：启用异常处理
• MFEM_USE_LIBUNWIND：使用 libunwind 打印堆栈跟踪
• MFEM_USE_OPENMP：启用 OpenMP 后端
• MFEM_USE_SUNDIALS：启用 SUNDIALS 库支持
• MFEM_USE_SUITESPARSE：启用 SuiteSparse 支持
• MFEM_USE_CUDA：启用 CUDA 支持
• MFEM_USE_HIP：启用 HIP 支持

常见问题

1. 如何指定 MPI 启动器？ 默认使用 "mpirun -np"，可以通过修改配置来更改。

2. 如何构建外部库？

   • hypre 和 METIS 需要先下载并构建
   • 确保路径设置正确

3. 如何启用特定功能？ 通过设置相应的 MFEM_USE_* 选项为 YES

总结

MFEM 提供了灵活的构建选项，可以满足从简单串行计算到复杂并行 GPU 加速的各种需求。通过合理配置构建选项，用户可以根据自己的硬件环境和计算需求定制 MFEM 库。建议初次使用的用户从串行构建开始，逐步尝试并行和 GPU 加速功能。