中文

Squeeze:面向张量核心 GPU 的高效紧凑分形处理方案

分布式、并行与集群计算 2022-01-04 v1 计算几何 离散数学

摘要

本文提出 Squeeze,一种面向张量核心 GPU 的高效紧凑分形处理方案。通过结合紧凑形式与展开形式之间的离散空间变换,人们可以在不将分形在内存中展开的情况下,对其执行具有邻域访问能力的数据并行计算。这些空间变换被表述为两个由 GPU 张量核心加速的线程映射 λ(ω)\lambda(\omega)ν(ω)\nu(\omega),分别充当紧凑到展开空间以及展开到紧凑空间的函数。映射的时间开销为 O(log2logs(n))\mathcal{O}(\log_2 \log_s(n)),其中 nn 为分形展开形式下 n×nn \times n 嵌入的边长,ss 为线性缩放因子。所提方法适用于属于非重叠边界框(NBB)类离散分形的任意分形,并可扩展至三维。以离散 Sierpinski 三角形为案例研究的实验结果表明,相较于基于 GPU 的展开空间边界框方法,最高可实现约 12×\sim12\times 的加速比以及高达约 315×\sim 315\times 的内存缩减因子。这些结果表明,所提紧凑方法将使科学界能够高效处理此前无法放入 GPU 内存的问题。

关键词

引用

@article{arxiv.2201.00613,
  title  = {Squeeze: Efficient Compact Fractals for Tensor Core GPUs},
  author = {Felipe A. Quezada and Cristóbal A. Navarro and Nancy Hitschfeld and Benjamin Bustos},
  journal= {arXiv preprint arXiv:2201.00613},
  year   = {2022}
}