add image support for gpu version

Signed-off-by: Jade Abraham <jade.abraham@hpe.com>

Author: jabraham17

06-heat-1D-gpu.chpl

@@ -1,3 +1,6 @@
+use ImageUtils;
+
+
 config const nx = 100000;
 config const N = 10000;
 config const alpha = 0.1;
@@ -10,10 +13,13 @@ on here.gpus[0] {
   u[nx/4..3*nx/4] = 2.0;
   var un = u;
 
+  writeImage(u);
+
   for 1..N {
     un <=> u;
     forall i in omegaHat do
       u[i] = un[i] + alpha *
   	   (un[i-1] - 2*un[i] + un[i+1]);
+    writeImage(u);
   }
 }

ImageUtils.chpl

@@ -43,9 +43,11 @@ module ImageUtils {
 
     checkDims(u);
 
-    @functionStatic
-    ref pipe = try! new mediaPipe("heat.mp4", imageType.bmp, framerate);
+    on Locales[0] {
+      @functionStatic
+      ref pipe = try! new mediaPipe("heat.mp4", imageType.bmp, framerate);
 
-    pipe.writeFrame(processFrameData(u));
+      pipe.writeFrame(processFrameData(u));
+    }
   }
 }