OpenACC與cublas交互-c版本

比較簡單，通俗易懂，就直接上代碼了：

/*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;
}

編譯：

1. pgcc -o openacc_c_cublas openacc_c_cublas.c  -acc -Mcuda=cc3.5 -Minfo -lcublas