OpenACC与cublas交互-fortran版本

最新推荐文章于 2022-07-28 13:35:53 发布

wcj0626

最新推荐文章于 2022-07-28 13:35:53 发布

阅读量1.7k

点赞数

分类专栏： OpenACC

本文链接：https://blog.csdn.net/wcj0626/article/details/12612155

版权

OpenACC 专栏收录该内容

9 篇文章 3 订阅

订阅专栏

本文提供了一个使用 Fortran 调用 CUBLAS 的示例程序，展示了如何通过 OpenACC 指令进行数据传输，并利用 CUBLAS 库函数实现向量运算。该示例有助于理解 CUDA 在 Fortran 中的应用。

摘要由CSDN通过智能技术生成

目前，cuda有许多库函数可以直接调用，不多都是基于C开发，对于Fortran用户，稍显麻烦一点的是需要自己定义调用这些库函数的接口。据我的了解，目前cubals，cula已经做好接口，可以直接使用。下面就以cubals为例：

! openacc_cublas.f90
program main
  use cublas
  integer, parameter :: N = 2**20
  real, dimension(N) :: X, Y

  X(:) = 1.0
  Y(:) = 0.0

  !$acc data copy(y) copyin(x)
  !$acc host_data use_device(x,y)
  call cublassaxpy(N, 2.0, x, 1, y, 1)
  !$acc end host_data
  !$acc end data

  print *, y(1)
end program

编译：