OpenACC与cublas交互-c版本

最新推荐文章于 2021-01-02 14:53:02 发布

原创最新推荐文章于 2021-01-02 14:53:02 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

OpenACC 专栏收录该内容

9 篇文章

订阅专栏

本文通过一个简单的示例介绍了如何结合使用OpenACC和cuBLAS进行并行计算。示例中演示了如何利用GPU加速向量运算，包括初始化向量、调用cuBLAS库函数以及数据的内存管理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

比较简单，通俗易懂，就直接上代码了：

/*openacc_c_cublas.c*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

extern void cublasSaxpy(int,float,float*,int,float*,int);

int main(int argc, char **argv)
{
  float *x, *y, tmp;
  int n = 1<<20, i;

  x = (float*)malloc(n*sizeof(float));
  y = (float*)malloc(n*sizeof(float));

  #pragma acc data create(x[0:n]) copyout(y[0:n])
  {
    #pragma acc kernels
    {
      #pragma acc loop independent
      for( i = 0; i < n; i++)
      {
        x[i] = 1.0f;
        y[i] = 0.0f;
      }
    }
      
    #pragma acc host_data use_device(x,y)
    {
      cublasSaxpy(n, 2.0, x, 1, y, 1);
    }
  }

  fprintf(stdout, "y[0] = %f\n",y[0]);
  return 0;
}

编译：