如何使用英特尔 MKL (cblas) 的“双”功能和“单”功能

Question

blas 的 C 接口（我使用的是英特尔 MKL）具有不同的函数，用于对单数据类型和双数据类型的输入数据进行操作。例如，cblas_dgemm 与 cblas_sgemm。

我编写了一个广泛使用这些功能的程序。我希望用户可以选择使用单精度或双精度进行计算。这仅仅意味着它说 dgemm 的每个地方都应该是 sgemm （对于其他功能等等）。我不想复制所有代码并将所有 sgemm 替换为 dgemm。我怎样才能最优雅地向用户提供这两种功能，而不需要两个相同的代码副本？

一种可能的解决方案是将 cblas_dgemm 的所有实例替换为 cblas_gemm，并使用宏将 cblas_gemm #define cblas_gemm 定义为 cblas_dgemm 或 cblas_sgemm，具体取决于 g++ 编译器标志 -DDATATYPE_SINGLE 或 -DDATATYPE_DOUBLE。然后，将有两个不同的可执行文件，具有相同的代码库。这是最好的解决方案吗？

Answer 1

因为你不能只用单精度函数替换函数调用（参数类型也必须改变，我建议为浮点和双精度定义一个模板类，比如说template <typename T> class processing，并且，编写模板类的两个特化，用于差异很重要的地方（即您调用 cblas 例程的地方）。