隐式时间离散化的cuda和数值方法

我希望移植一些通过IMPLICIT形式的有限体积方法（用于时间离散化）解析一组偏微分方程（PDE）的代码。

结果，存在由ADI / TDMA方案处理的x，y，z方向上的三对角方程组。

我似乎找不到关于使用CUDA进行PDE的隐式解决方案的任何信息。

ADI / TDMA方案是否可以在CUDA中实现？某处是否有类似2D热扩散方程的示例？

我所能找到的是一个二维热扩散方程的CUDA示例代码，它具有有限的差异，但形式为EXPLICIT（剑桥大学）。

任何提示/参考将不胜感激。

此问题使其本身具有高度矢量化的形式。如您所述，ADI方法给出了三对角线系统的几个步骤。由于采用线性方程式，因此您可以使用CUsolver和CUblas调用标准线性代数例程的并行GPU版本。使用这些，您应该能够获取显式代码，只需将内部循环更改为适当的CUsolver调用，并以某种方式使它看起来几乎与CPU实现完全相同，但是可以通过在GPU上进行矩阵运算来解决它。图书馆电话。