TiledArray：大规模并行块稀疏张量框架

TiledArray：大规模并行块稀疏张量框架

tiledarray A massively-parallel, block-sparse tensor framework written in C++ 项目地址: https://gitcode.com/gh_mirrors/ti/tiledarray

项目介绍

TiledArray 是一个用 C++ 编写的高效、可扩展的大规模并行块稀疏张量计算框架，特别适用于高能量子力学等领域的复杂数学运算。它支持用户通过类似于标准数学符号的原生 C++ 代码构建任意复杂的张量表达式。该框架旨在从单个多核计算机到大型分布式内存多处理器系统提供无缝的平行性能，并基于 MADNESS（MADWorld）并发运行时环境构建。

该项目得到了美国国家科学基金会、Alfred P. Sloan 基金会、Camille and Henry Dreyfus 基金会以及能源部的支持，其设计目标包括通用的稠密和块稀疏张量算术、高级和低级访问控制，以及深度可重用性，允许用户自定义从瓷砖类型到稀疏张量结构描述的各个方面。

项目快速启动

安装 TiledArray

首先，你需要克隆仓库并设置好编译环境：

$ git clone https://github.com/ValeevGroup/TiledArray.git tiledarray
$ cd tiledarray
$ mkdir build && cd build
$ cmake -DCMAKE_INSTALL_PREFIX=/your/install/path \
        -DCMAKE_TOOLCHAIN_FILE=../cmake/vg/toolchains/<your-toolchain-file> ..
$ make
$ make check  # 可选步骤，用于测试
$ make install

确保你的系统已配置好 MPI 编译器包裹器及全局安装的 BLAS 和 LAPACK 库。

快速示例

以下是一个简单的矩阵乘法示例，展示 TiledArray 的基本用法：

#include <tiledarray.h>
int main(int argc, char** argv) {
    // 初始化并行运行时
    TA::World& world = TA::initialize(argc, argv);
    
    // 定义一个二维平铺范围结构，用于数组的分块
    auto trange = TA::TiledRange<TA::TiledRange1<0, 10, 20, ..., 100>,
                                TA::TiledRange1<0, 10, 20, ..., 100>>();
    
    // 构造并填充数据
    TA::TArrayD A(world, trange), B(world, trange);
    A.fill_local(3.0);  // 设置本地部分的值
    B.fill_local(2.0);
    
    // 创建结果数组
    TA::TArrayD C;
    
    // 执行分布式矩阵乘法
    C("i", "j") = A("i", "k") * B("k", "j");
    
    // 结束并行运行时
    TA::finalize();
    return 0;
}

应用案例和最佳实践

在实际应用中，TiledArray 被用来解决高度复杂的并行计算问题，比如量子化学中的耦合簇单双激发（CCSD）波函数求解。这种情况下，它展示了即使对于比简单矩阵乘法更为复杂的表达式，也能实现卓越的并行加速效果。

最佳实践提示：

• 在编写张量表达式时，利用 Einstein 求和约定自动处理循环。
• 利用预定义的 TiledRange 来优化内存布局，提高计算效率。
• 对于大规模并行计算，正确初始化和最终化并行环境至关重要。

典型生态项目

虽然项目本身即为其生态的核心，TiledArray 的设计目的是为了广泛地融入科学计算领域，特别是在需要处理大规模、高维度且结构复杂的稀疏数据集的应用中。尽管没有特定列出“典型生态项目”，TiledArray 直接应用于物理、化学、材料科学等领域内的高精度数值模拟软件，是那些要求高性能计算基础架构的科学家们的工具箱中不可或缺的一部分。

开发者们在使用 TiledArray 时，通常结合其它库和技术（如 MPI、BLAS/LAPACK）以构建完整的科学计算解决方案，而具体生态项目往往依赖于各学科的具体研究需求。

---

此文档概览了 TiledArray 的基本使用，提供了快速入门的指引，并简要介绍了其在科学计算领域的潜在应用价值。深入学习和应用 TiledArray 需要进一步参考其官方文档和源码细节。

tiledarray A massively-parallel, block-sparse tensor framework written in C++ 项目地址: https://gitcode.com/gh_mirrors/ti/tiledarray