HPC Magazine septembre 2013 - L'Atelier CUDA.

Listing 1 - axpy, "le Hello World" parallèle, en version Copperhead.


from copperhead import *
import numpy as np
@cu
def axpy(a, x, y):
  return [a * xi + yi for xi, yi in zip(x, y)]

n = 1000000
a = 2.0
x = np.random.rand(n)
y = np.random.rand(n)
with places.gpu0:
  gpu_result = axpy(a, x, y)
with places.openmp:
  cpu_result = axpy(a, x, y)