chapel/test/gpu/native/multiGPU/multiGPU.chpl at main · chapel-lang/chapel · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
use GpuDiagnostics;

config const n = 10;

config const alpha = 10;

config const writeArrays = false;


startGpuDiagnostics();
for gpu in here.gpus do on gpu {
  var A: [1..n] int;
  var B: [1..n] int;
  var C: [1..n] int;

  / each of these are one launch
  A = 1;                            assertElemVal(A, 1);
  B = 2;                            assertElemVal(B, 2);
  C = 3;                            assertElemVal(C, 3);
  A = B + alpha * C;                assertElemVal(A, 2+alpha*3);
}
stopGpuDiagnostics();

const nLaunch = getGpuDiagnostics().kernel_launch;

assert(nLaunch == here.gpus.size*4);

proc assertElemVal(X, val) {
  if writeArrays {
    writeln(X);
  }

  for x in X do assert(x == val, "Expected: ", val, " Actual: ", x);
}