SAXPY implementation with cublas library

saxpy_cublas.cu

#include <cublas.h>

int main()
{
  int N = 1 << 16;
  
  int size = N * sizeof(float);
  float *h_x = (float*)malloc(size);
  float *h_y = (float*)malloc(size);
  
  for(int i = 0; i < N; i++)
  {
    h_x[i] = 2.0;
    h_y[i] = 2.0;
  }
  
  float *d_x;
  float *d_y;
  
  cublasInit();
  cublasAlloc(N, sizeof(float), (void**)&d_x);
  cublasAlloc(N, sizeof(float), (void**)&d_y);
  
  cublasSetVector(N, sizeof(float), h_x, 1, d_x, 1);
  cublasSetVector(N, sizeof(float), h_y, 1, d_y, 1);
  
  cublasSaxpy(N, 2.0, d_x, 1, d_y, 1);
  
  cublasGetVector(N, sizeof(float), d_y, 1, h_y, 1);
  cublasShutdown();
  
  cublasFree(d_x);
  cublasFree(d_y);
  
  free(h_x);
  free(h_y);
  
  return 0;
}