您的位置:首页 > 运维架构

OpenACC与cublas交互-fortran版本

2013-10-11 10:16 246 查看
目前,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


编译:

pgf90 -o cuda_fortran_opencc_cublas openacc_cublas.f90 -acc -Mcuda=cc3.5 -Minfo -lcublas
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐